From 8c147d6e87cc3387cb1accd3c92322241dacec03 Mon Sep 17 00:00:00 2001 From: "databricks-ci-ghec-1[bot]" <184311507+databricks-ci-ghec-1[bot]@users.noreply.github.com> Date: Thu, 22 Oct 2026 04:33:43 +0000 Subject: [PATCH 1/4] Update Go SDK to v0.190.0 --- .codegen/_openapi_sha | 2 +- .codegen/cli.json | 285 +++- .../dependency-updates/go-sdk-v0.190.0.md | 1 + acceptance/bundle/refschema/out.fields.txt | 12 + .../dresources/configs/features.generated.yml | 2 + .../configs/sql_warehouses.generated.yml | 5 + .../validation/generated/enum_fields.go | 1 + bundle/schema/jsonschema.json | 155 +- cmd/workspace/ai-gateway/ai-gateway.go | 27 +- .../bundle-deployments/bundle-deployments.go | 4 +- cmd/workspace/shares/shares.go | 1 - cmd/workspace/warehouses/warehouses.go | 4 + go.mod | 2 +- go.sum | 4 +- python/codegen/uv.lock | 36 +- python/databricks/bundles/alerts/__init__.py | 195 +-- .../bundles/alerts/_models/aggregation.py | 36 +- .../bundles/alerts/_models/alert.py | 67 +- .../alerts/_models/alert_evaluation_state.py | 36 +- .../_models/alert_statement_parameter.py | 43 +- .../alerts/_models/alert_v2_evaluation.py | 62 +- .../alerts/_models/alert_v2_notification.py | 44 +- .../alerts/_models/alert_v2_operand.py | 50 +- .../alerts/_models/alert_v2_operand_column.py | 41 +- .../alerts/_models/alert_v2_operand_value.py | 40 +- .../bundles/alerts/_models/alert_v2_run_as.py | 40 +- .../alerts/_models/alert_v2_subscription.py | 40 +- .../alerts/_models/comparison_operator.py | 45 +- .../bundles/alerts/_models/cron_schedule.py | 44 +- .../alerts/_models/custom_template_format.py | 34 +- .../bundles/alerts/_models/lifecycle.py | 40 +- .../bundles/alerts/_models/permission.py | 44 +- .../alerts/_models/permission_level.py | 58 +- .../alerts/_models/schedule_pause_status.py | 31 +- python/databricks/bundles/apps/__init__.py | 359 ++--- python/databricks/bundles/apps/_models/app.py | 94 +- .../bundles/apps/_models/app_config.py | 63 +- .../bundles/apps/_models/app_env_var.py | 62 +- .../bundles/apps/_models/app_permission.py | 66 +- .../apps/_models/app_permission_level.py | 54 +- .../bundles/apps/_models/app_resource.py | 104 +- .../bundles/apps/_models/app_resource_app.py | 66 +- .../app_resource_app_app_permission.py | 53 +- .../apps/_models/app_resource_database.py | 66 +- ...p_resource_database_database_permission.py | 57 +- .../apps/_models/app_resource_experiment.py | 66 +- ...source_experiment_experiment_permission.py | 58 +- .../apps/_models/app_resource_genie_space.py | 66 +- ...urce_genie_space_genie_space_permission.py | 58 +- .../bundles/apps/_models/app_resource_job.py | 66 +- .../app_resource_job_job_permission.py | 58 +- .../apps/_models/app_resource_postgres.py | 66 +- ...p_resource_postgres_postgres_permission.py | 57 +- .../apps/_models/app_resource_secret.py | 66 +- .../app_resource_secret_secret_permission.py | 58 +- .../_models/app_resource_serving_endpoint.py | 70 +- ...ng_endpoint_serving_endpoint_permission.py | 58 +- .../_models/app_resource_sql_warehouse.py | 66 +- ..._sql_warehouse_sql_warehouse_permission.py | 58 +- .../apps/_models/app_resource_uc_securable.py | 72 +- ...ce_uc_securable_uc_securable_permission.py | 60 +- ...resource_uc_securable_uc_securable_type.py | 58 +- .../bundles/apps/_models/compute_size.py | 53 +- .../bundles/apps/_models/git_repository.py | 63 +- .../bundles/apps/_models/git_source.py | 63 +- .../apps/_models/lifecycle_with_started.py | 62 +- .../_models/telemetry_export_destination.py | 71 +- .../bundles/apps/_models/unity_catalog.py | 63 +- .../databricks/bundles/catalogs/__init__.py | 75 +- .../_models/azure_encryption_settings.py | 23 +- .../bundles/catalogs/_models/catalog.py | 38 +- .../catalogs/_models/encryption_settings.py | 28 +- .../bundles/catalogs/_models/lifecycle.py | 23 +- .../bundles/catalogs/_models/privilege.py | 72 +- .../catalogs/_models/privilege_assignment.py | 24 +- .../bundles/cluster_policies/__init__.py | 94 +- .../_models/cluster_policy.py | 35 +- .../_models/cluster_policy_permission.py | 29 +- .../cluster_policy_permission_level.py | 17 +- .../cluster_policies/_models/library.py | 39 +- .../cluster_policies/_models/lifecycle.py | 25 +- .../cluster_policies/_models/maven_library.py | 29 +- .../_models/python_py_pi_library.py | 25 +- .../_models/r_cran_library.py | 25 +- .../databricks/bundles/clusters/__init__.py | 407 ++--- .../bundles/clusters/_models/adlsgen2_info.py | 63 +- .../bundles/clusters/_models/auto_scale.py | 62 +- .../clusters/_models/aws_attributes.py | 73 +- .../clusters/_models/aws_availability.py | 60 +- .../clusters/_models/azure_attributes.py | 73 +- .../clusters/_models/azure_availability.py | 59 +- .../bundles/clusters/_models/clients_types.py | 62 +- .../bundles/clusters/_models/cluster.py | 126 +- .../clusters/_models/cluster_log_conf.py | 77 +- .../clusters/_models/cluster_permission.py | 66 +- .../_models/cluster_permission_level.py | 58 +- .../_models/confidential_compute_type.py | 60 +- .../clusters/_models/data_security_mode.py | 80 +- .../clusters/_models/dbfs_storage_info.py | 63 +- .../clusters/_models/dependency_mode.py | 65 +- .../clusters/_models/docker_basic_auth.py | 62 +- .../bundles/clusters/_models/docker_image.py | 66 +- .../clusters/_models/ebs_volume_type.py | 58 +- .../clusters/_models/gcp_attributes.py | 73 +- .../clusters/_models/gcp_availability.py | 59 +- .../clusters/_models/gcs_storage_info.py | 63 +- .../clusters/_models/init_script_info.py | 93 +- .../bundles/clusters/_models/kind.py | 60 +- .../bundles/clusters/_models/library.py | 76 +- .../_models/lifecycle_with_started.py | 62 +- .../clusters/_models/local_file_info.py | 62 +- .../clusters/_models/log_analytics_info.py | 62 +- .../bundles/clusters/_models/maven_library.py | 66 +- .../clusters/_models/node_type_flexibility.py | 63 +- .../clusters/_models/python_py_pi_library.py | 62 +- .../clusters/_models/r_cran_library.py | 62 +- .../clusters/_models/runtime_engine.py | 53 +- .../clusters/_models/s3_storage_info.py | 63 +- .../clusters/_models/volumes_storage_info.py | 63 +- .../bundles/clusters/_models/workload_type.py | 67 +- .../_models/workspace_storage_info.py | 63 +- python/databricks/bundles/core/__init__.py | 36 +- .../bundles/core/_generated/__init__.py | 125 +- .../bundles/core/_generated/alerts.py | 4 +- .../bundles/core/_generated/apps.py | 4 +- .../bundles/core/_generated/catalogs.py | 4 +- .../core/_generated/cluster_policies.py | 11 +- .../bundles/core/_generated/clusters.py | 4 +- .../bundles/core/_generated/dashboards.py | 5 +- .../core/_generated/database_catalogs.py | 17 +- .../core/_generated/database_instances.py | 21 +- .../bundles/core/_generated/experiments.py | 21 +- .../core/_generated/external_locations.py | 21 +- .../bundles/core/_generated/genie_spaces.py | 5 +- .../bundles/core/_generated/instance_pools.py | 5 +- .../bundles/core/_generated/mcp_services.py | 5 +- .../_generated/model_provider_services.py | 29 +- .../bundles/core/_generated/model_services.py | 5 +- .../_generated/model_serving_endpoints.py | 29 +- .../bundles/core/_generated/models.py | 5 +- .../core/_generated/quality_monitors.py | 17 +- .../core/_generated/registered_models.py | 17 +- .../bundles/core/_generated/secret_scopes.py | 5 +- .../bundles/core/_generated/sql_warehouses.py | 5 +- .../core/_generated/synced_database_tables.py | 25 +- .../_generated/vector_search_endpoints.py | 29 +- .../core/_generated/vector_search_indexes.py | 25 +- .../databricks/bundles/dashboards/__init__.py | 56 +- .../bundles/dashboards/_models/dashboard.py | 28 +- .../bundles/dashboards/_models/lifecycle.py | 23 +- .../dashboards/_models/lifecycle_state.py | 14 +- .../bundles/dashboards/_models/permission.py | 27 +- .../dashboards/_models/permission_level.py | 41 +- .../bundles/database_catalogs/__init__.py | 24 +- .../_models/database_catalog.py | 21 +- .../database_catalogs/_models/lifecycle.py | 18 +- .../bundles/database_instances/__init__.py | 70 +- .../database_instances/_models/custom_tag.py | 23 +- .../_models/database_instance.py | 43 +- .../_models/database_instance_ref.py | 24 +- .../database_instances/_models/lifecycle.py | 23 +- .../database_instances/_models/permission.py | 27 +- .../_models/permission_level.py | 41 +- .../bundles/experiments/__init__.py | 82 +- .../_models/experiment_permission_level.py | 20 +- .../experiments/_models/experiment_tag.py | 25 +- .../_models/experiment_trace_location.py | 31 +- .../bundles/experiments/_models/lifecycle.py | 24 +- .../experiments/_models/mlflow_experiment.py | 47 +- .../_models/mlflow_experiment_permission.py | 32 +- .../experiments/_models/uc_trace_location.py | 27 +- .../bundles/external_locations/__init__.py | 128 +- .../_models/aws_sqs_queue.py | 29 +- .../_models/azure_queue_storage.py | 29 +- .../_models/encryption_details.py | 34 +- .../_models/external_location.py | 48 +- .../_models/file_event_queue.py | 41 +- .../external_locations/_models/gcp_pubsub.py | 29 +- .../external_locations/_models/lifecycle.py | 29 +- .../external_locations/_models/privilege.py | 78 +- .../_models/privilege_assignment.py | 33 +- .../_models/sse_encryption_details.py | 34 +- .../sse_encryption_details_algorithm.py | 25 +- .../databricks/bundles/features/__init__.py | 530 +++--- .../features/_models/aggregation_function.py | 132 +- .../_models/approx_count_distinct_function.py | 67 +- .../_models/approx_percentile_function.py | 63 +- .../bundles/features/_models/avg_function.py | 63 +- .../features/_models/column_selection.py | 63 +- .../features/_models/count_function.py | 63 +- .../bundles/features/_models/custom_udf.py | 69 +- .../bundles/features/_models/data_source.py | 101 +- .../features/_models/delta_table_source.py | 62 +- .../bundles/features/_models/entity_column.py | 62 +- .../bundles/features/_models/feature.py | 89 +- .../features/_models/feature_reference.py | 65 +- .../features/_models/feature_view_source.py | 69 +- .../features/_models/field_definition.py | 67 +- .../_models/first_distinct_function.py | 65 +- .../features/_models/first_function.py | 63 +- .../features/_models/first_n_function.py | 65 +- .../bundles/features/_models/flat_schema.py | 67 +- .../bundles/features/_models/function.py | 74 +- .../bundles/features/_models/input_binding.py | 65 +- .../bundles/features/_models/job_context.py | 63 +- .../bundles/features/_models/kafka_source.py | 62 +- .../_models/last_distinct_function.py | 65 +- .../bundles/features/_models/last_function.py | 63 +- .../features/_models/last_n_function.py | 65 +- .../bundles/features/_models/lifecycle.py | 62 +- .../features/_models/lineage_context.py | 69 +- .../bundles/features/_models/max_function.py | 63 +- .../bundles/features/_models/min_function.py | 63 +- .../features/_models/request_source.py | 64 +- .../features/_models/rolling_window.py | 63 +- .../features/_models/sawtooth_window.py | 65 +- .../features/_models/scalar_data_type.py | 71 +- .../bundles/features/_models/sdp_source.py | 189 +++ .../features/_models/sliding_window.py | 62 +- .../features/_models/source_lateness.py | 65 +- .../features/_models/stddev_pop_function.py | 63 +- .../features/_models/stddev_samp_function.py | 63 +- .../bundles/features/_models/stream_source.py | 63 +- .../bundles/features/_models/sum_function.py | 63 +- .../bundles/features/_models/time_window.py | 78 +- .../features/_models/timeseries_column.py | 62 +- .../features/_models/tumbling_window.py | 62 +- .../features/_models/var_pop_function.py | 63 +- .../features/_models/var_samp_function.py | 63 +- .../bundles/genie_spaces/__init__.py | 46 +- .../genie_spaces/_models/genie_space.py | 25 +- .../bundles/genie_spaces/_models/lifecycle.py | 21 +- .../genie_spaces/_models/permission.py | 25 +- .../genie_spaces/_models/permission_level.py | 39 +- .../bundles/instance_pools/__init__.py | 192 +-- .../instance_pools/_models/disk_spec.py | 41 +- .../instance_pools/_models/disk_type.py | 48 +- .../disk_type_azure_disk_volume_type.py | 35 +- .../_models/disk_type_ebs_volume_type.py | 35 +- .../_models/docker_basic_auth.py | 39 +- .../instance_pools/_models/docker_image.py | 43 +- .../_models/gcp_availability.py | 36 +- .../instance_pools/_models/instance_pool.py | 72 +- .../_models/instance_pool_aws_attributes.py | 48 +- ...stance_pool_aws_attributes_availability.py | 35 +- .../_models/instance_pool_azure_attributes.py | 48 +- ...ance_pool_azure_attributes_availability.py | 35 +- .../_models/instance_pool_gcp_attributes.py | 48 +- .../_models/instance_pool_permission.py | 43 +- .../_models/instance_pool_permission_level.py | 35 +- .../instance_pools/_models/lifecycle.py | 39 +- .../_models/node_type_flexibility.py | 40 +- .../databricks/bundles/job_runs/__init__.py | 64 +- .../bundles/job_runs/_models/job_run.py | 43 +- .../job_runs/_models/job_run_lifecycle.py | 27 +- .../job_runs/_models/job_run_trigger.py | 23 +- .../job_runs/_models/performance_target.py | 19 +- .../job_runs/_models/pipeline_params.py | 23 +- .../job_runs/_models/queue_settings.py | 23 +- python/databricks/bundles/jobs/__init__.py | 1418 ++++++----------- .../bundles/jobs/_models/adlsgen2_info.py | 177 +- .../bundles/jobs/_models/ai_runtime_task.py | 191 ++- .../_models/ai_runtime_task_priority_class.py | 174 +- .../bundles/jobs/_models/alert_task.py | 184 ++- .../jobs/_models/alert_task_subscriber.py | 177 +- .../jobs/_models/authentication_method.py | 167 +- .../bundles/jobs/_models/auto_scale.py | 176 +- .../bundles/jobs/_models/aws_attributes.py | 185 ++- .../bundles/jobs/_models/aws_availability.py | 174 +- .../bundles/jobs/_models/azure_attributes.py | 185 ++- .../jobs/_models/azure_availability.py | 173 +- .../jobs/_models/clean_rooms_notebook_task.py | 181 ++- .../bundles/jobs/_models/clients_types.py | 176 +- .../bundles/jobs/_models/cluster_log_conf.py | 189 ++- .../bundles/jobs/_models/cluster_spec.py | 225 ++- .../bundles/jobs/_models/compute.py | 180 ++- .../bundles/jobs/_models/compute_config.py | 177 +- .../bundles/jobs/_models/compute_spec.py | 181 ++- .../_models/compute_spec_accelerator_type.py | 173 +- .../bundles/jobs/_models/condition.py | 167 +- .../bundles/jobs/_models/condition_task.py | 180 ++- .../bundles/jobs/_models/condition_task_op.py | 182 ++- .../jobs/_models/confidential_compute_type.py | 174 +- .../bundles/jobs/_models/continuous.py | 183 ++- .../continuous_trigger_configuration.py | 189 ++- .../bundles/jobs/_models/cron_schedule.py | 179 ++- .../_models/cron_trigger_configuration.py | 177 +- .../bundles/jobs/_models/dashboard_task.py | 181 ++- .../jobs/_models/data_security_mode.py | 194 ++- .../bundles/jobs/_models/day_of_week.py | 175 +- .../bundles/jobs/_models/dbfs_storage_info.py | 177 +- .../bundles/jobs/_models/dbt_platform_task.py | 177 +- .../bundles/jobs/_models/dbt_task.py | 175 +- .../bundles/jobs/_models/dependency_mode.py | 179 ++- .../bundles/jobs/_models/deployment_spec.py | 181 ++- .../bundles/jobs/_models/docker_basic_auth.py | 176 +- .../bundles/jobs/_models/docker_image.py | 180 ++- .../bundles/jobs/_models/ebs_volume_type.py | 172 +- .../bundles/jobs/_models/environment.py | 177 +- .../file_arrival_trigger_configuration.py | 180 ++- .../bundles/jobs/_models/for_each_task.py | 178 ++- .../bundles/jobs/_models/gcp_attributes.py | 185 ++- .../bundles/jobs/_models/gcp_availability.py | 173 +- .../bundles/jobs/_models/gcs_storage_info.py | 177 +- .../jobs/_models/gen_ai_compute_task.py | 182 ++- .../bundles/jobs/_models/git_provider.py | 181 ++- .../bundles/jobs/_models/git_source.py | 184 ++- .../jobs/_models/hardware_accelerator_type.py | 173 +- .../bundles/jobs/_models/init_script_info.py | 205 ++- python/databricks/bundles/jobs/_models/job.py | 244 ++- .../bundles/jobs/_models/job_cluster.py | 177 +- .../jobs/_models/job_email_notifications.py | 180 ++- .../bundles/jobs/_models/job_environment.py | 177 +- .../jobs/_models/job_environment_variables.py | 181 ++- .../_models/job_environment_variables_spec.py | 181 ++- .../jobs/_models/job_notification_settings.py | 176 +- .../jobs/_models/job_parameter_definition.py | 176 +- .../bundles/jobs/_models/job_permission.py | 180 ++- .../jobs/_models/job_permission_level.py | 172 +- .../bundles/jobs/_models/job_run_as.py | 179 ++- .../jobs/_models/jobs_health_metric.py | 181 ++- .../jobs/_models/jobs_health_operator.py | 168 +- .../bundles/jobs/_models/jobs_health_rule.py | 184 ++- .../bundles/jobs/_models/jobs_health_rules.py | 181 ++- .../databricks/bundles/jobs/_models/kind.py | 174 +- .../bundles/jobs/_models/library.py | 188 ++- .../bundles/jobs/_models/lifecycle.py | 176 +- .../bundles/jobs/_models/local_file_info.py | 176 +- .../jobs/_models/log_analytics_info.py | 176 +- .../jobs/_models/maintenance_window.py | 176 +- .../bundles/jobs/_models/maven_library.py | 180 ++- .../_models/model_trigger_configuration.py | 189 ++- .../model_trigger_configuration_condition.py | 173 +- .../jobs/_models/node_type_flexibility.py | 177 +- .../bundles/jobs/_models/notebook_task.py | 179 ++- .../bundles/jobs/_models/pause_status.py | 167 +- .../jobs/_models/performance_target.py | 172 +- .../_models/periodic_trigger_configuration.py | 184 ++- ...eriodic_trigger_configuration_time_unit.py | 171 +- .../bundles/jobs/_models/pipeline_params.py | 176 +- .../bundles/jobs/_models/pipeline_task.py | 181 ++- .../bundles/jobs/_models/power_bi_model.py | 179 ++- .../bundles/jobs/_models/power_bi_table.py | 175 +- .../bundles/jobs/_models/power_bi_task.py | 184 ++- .../jobs/_models/python_operator_task.py | 185 ++- .../_models/python_operator_task_parameter.py | 181 ++- .../jobs/_models/python_py_pi_library.py | 176 +- .../bundles/jobs/_models/python_wheel_task.py | 176 +- .../bundles/jobs/_models/queue_settings.py | 176 +- .../bundles/jobs/_models/r_cran_library.py | 176 +- .../databricks/bundles/jobs/_models/run_if.py | 182 ++- .../bundles/jobs/_models/run_job_task.py | 184 ++- .../bundles/jobs/_models/runtime_engine.py | 167 +- .../bundles/jobs/_models/s3_storage_info.py | 177 +- .../databricks/bundles/jobs/_models/source.py | 168 +- .../bundles/jobs/_models/spark_jar_task.py | 180 ++- .../bundles/jobs/_models/spark_python_task.py | 179 ++- .../bundles/jobs/_models/spark_submit_task.py | 179 ++- .../bundles/jobs/_models/sparse_checkout.py | 176 +- .../_models/sql_condition_configuration.py | 185 ++- .../_models/sql_condition_trigger_mode.py | 176 +- .../bundles/jobs/_models/sql_task.py | 193 ++- .../bundles/jobs/_models/sql_task_alert.py | 184 ++- .../jobs/_models/sql_task_dashboard.py | 184 ++- .../bundles/jobs/_models/sql_task_file.py | 175 +- .../bundles/jobs/_models/sql_task_query.py | 176 +- .../jobs/_models/sql_task_subscription.py | 176 +- .../bundles/jobs/_models/storage_mode.py | 167 +- .../bundles/jobs/_models/subscription.py | 188 ++- .../jobs/_models/subscription_attachment.py | 195 ++- ...subscription_attachment_data_attachment.py | 211 ++- ..._attachment_data_attachment_file_format.py | 175 +- ...ption_attachment_data_attachment_widget.py | 194 ++- ...subscription_attachment_page_attachment.py | 203 ++- ...tachment_page_attachment_page_selection.py | 175 +- .../jobs/_models/subscription_subscriber.py | 176 +- .../table_update_trigger_configuration.py | 179 ++- .../databricks/bundles/jobs/_models/task.py | 263 +-- .../bundles/jobs/_models/task_dependency.py | 176 +- .../jobs/_models/task_email_notifications.py | 180 ++- .../_models/task_notification_settings.py | 176 +- .../bundles/jobs/_models/task_retry_mode.py | 168 +- .../jobs/_models/trigger_configuration.py | 204 ++- .../bundles/jobs/_models/trigger_settings.py | 195 ++- .../jobs/_models/volumes_storage_info.py | 177 +- .../bundles/jobs/_models/webhook.py | 176 +- .../jobs/_models/webhook_notifications.py | 181 ++- .../bundles/jobs/_models/workload_type.py | 181 ++- .../jobs/_models/workspace_storage_info.py | 177 +- .../bundles/mcp_services/__init__.py | 97 +- .../bundles/mcp_services/_models/lifecycle.py | 28 +- .../mcp_services/_models/mcp_service.py | 43 +- .../_models/mcp_service_config.py | 37 +- .../mcp_service_config_source_connection.py | 33 +- .../bundles/mcp_services/_models/privilege.py | 77 +- .../_models/privilege_assignment.py | 27 +- .../mcp_services/_models/rate_limit.py | 37 +- .../_models/rate_limit_rate_limit_key.py | 31 +- .../rate_limit_rate_limit_renewal_period.py | 25 +- .../model_provider_services/__init__.py | 388 ++--- .../_models/inference_table_config.py | 54 +- .../_models/lifecycle.py | 53 +- .../_models/model_provider_service.py | 68 +- .../_models/model_provider_service_config.py | 172 +- ...e_config_amazon_bedrock_provider_config.py | 73 +- ...g_amazon_bedrock_provider_direct_config.py | 85 +- ...ervice_config_anthropic_provider_config.py | 85 +- ...config_anthropic_provider_direct_config.py | 69 +- ...onfig_anthropic_provider_relayed_config.py | 64 +- ..._provider_service_config_aws_access_key.py | 66 +- ...ce_config_azure_open_ai_provider_config.py | 73 +- ...ig_azure_open_ai_provider_direct_config.py | 95 +- ...fig_custom_provider_api_key_header_auth.py | 69 +- ...r_service_config_custom_provider_config.py | 76 +- ...ce_config_custom_provider_direct_config.py | 77 +- ..._service_config_entra_service_principal.py | 65 +- ...ice_config_external_model_provider_type.py | 81 +- ...onfig_gemini_enterprise_provider_config.py | 77 +- ...emini_enterprise_provider_direct_config.py | 83 +- ...onfig_microsoft_foundry_provider_config.py | 77 +- ...icrosoft_foundry_provider_direct_config.py | 97 +- ...ider_service_config_model_target_config.py | 65 +- ..._service_config_open_ai_provider_config.py | 73 +- ...e_config_open_ai_provider_direct_config.py | 65 +- ..._service_config_provider_pricing_config.py | 61 +- ...provider_service_config_provider_secret.py | 67 +- ...rovider_service_config_secret_reference.py | 61 +- ...vider_service_config_service_credential.py | 61 +- .../_models/privilege.py | 102 +- .../_models/privilege_assignment.py | 57 +- .../_models/rate_limit.py | 62 +- .../_models/rate_limit_rate_limit_key.py | 56 +- .../rate_limit_rate_limit_renewal_period.py | 50 +- .../bundles/model_services/__init__.py | 196 +-- .../_models/inference_table_config.py | 38 +- .../model_services/_models/lifecycle.py | 37 +- ...ider_service_config_model_target_config.py | 49 +- .../model_services/_models/model_service.py | 52 +- .../_models/model_service_config.py | 50 +- ...model_service_config_destination_config.py | 78 +- ...fig_destination_config_destination_type.py | 50 +- ...el_service_config_external_model_config.py | 46 +- .../model_service_config_fallback_config.py | 50 +- ...del_service_config_pay_per_token_config.py | 42 +- ...ce_config_provisioned_throughput_config.py | 45 +- .../model_service_config_routing_config.py | 54 +- .../model_services/_models/privilege.py | 86 +- .../_models/privilege_assignment.py | 41 +- .../model_services/_models/rate_limit.py | 46 +- .../_models/rate_limit_rate_limit_key.py | 40 +- .../rate_limit_rate_limit_renewal_period.py | 34 +- .../model_serving_endpoints/__init__.py | 530 +++--- .../_models/ai21_labs_config.py | 73 +- .../_models/ai_gateway_config.py | 93 +- .../ai_gateway_guardrail_parameters.py | 81 +- .../ai_gateway_guardrail_pii_behavior.py | 81 +- ...gateway_guardrail_pii_behavior_behavior.py | 68 +- .../_models/ai_gateway_guardrails.py | 77 +- .../ai_gateway_inference_table_config.py | 77 +- .../_models/ai_gateway_rate_limit.py | 81 +- .../_models/ai_gateway_rate_limit_key.py | 69 +- .../ai_gateway_rate_limit_renewal_period.py | 68 +- .../ai_gateway_usage_tracking_config.py | 77 +- .../_models/amazon_bedrock_config.py | 77 +- .../amazon_bedrock_config_bedrock_provider.py | 69 +- .../_models/anthropic_config.py | 73 +- .../_models/api_key_auth.py | 73 +- .../_models/auto_capture_config_input.py | 74 +- .../_models/bearer_token_auth.py | 73 +- .../_models/cohere_config.py | 73 +- .../_models/custom_provider_config.py | 82 +- .../databricks_model_serving_config.py | 77 +- .../_models/email_notifications.py | 73 +- .../_models/endpoint_core_config_input.py | 89 +- .../_models/endpoint_tag.py | 73 +- .../_models/external_model.py | 121 +- .../_models/external_model_provider.py | 79 +- .../_models/fallback_config.py | 73 +- .../_models/google_cloud_vertex_ai_config.py | 77 +- .../_models/lifecycle.py | 73 +- .../_models/model_serving_endpoint.py | 112 +- .../model_serving_endpoint_permission.py | 81 +- .../_models/open_ai_config.py | 74 +- .../_models/pa_lm_config.py | 73 +- .../_models/rate_limit.py | 84 +- .../_models/rate_limit_key.py | 67 +- .../_models/rate_limit_renewal_period.py | 67 +- .../model_serving_endpoints/_models/route.py | 73 +- .../_models/served_entity_input.py | 81 +- .../_models/served_model_input.py | 81 +- .../served_model_input_workload_type.py | 80 +- .../serving_endpoint_permission_level.py | 69 +- .../_models/serving_model_workload_type.py | 80 +- .../_models/telemetry_config.py | 89 +- .../_models/telemetry_feature.py | 75 +- .../telemetry_inference_table_config.py | 78 +- .../_models/traffic_config.py | 77 +- .../_models/unity_catalog_table_names.py | 73 +- python/databricks/bundles/models/__init__.py | 58 +- .../bundles/models/_models/lifecycle.py | 22 +- .../bundles/models/_models/mlflow_model.py | 31 +- .../models/_models/mlflow_model_permission.py | 26 +- .../bundles/models/_models/model_tag.py | 23 +- .../registered_model_permission_level.py | 25 +- .../databricks/bundles/pipelines/__init__.py | 1164 +++++--------- .../pipelines/_models/adlsgen2_info.py | 142 +- .../_models/api_source_connector_config.py | 144 +- .../_models/api_source_connector_options.py | 148 +- .../_models/auto_full_refresh_policy.py | 142 +- .../_models/avro_transformer_options.py | 144 +- .../pipelines/_models/aws_attributes.py | 150 +- .../pipelines/_models/aws_availability.py | 139 +- .../pipelines/_models/azure_attributes.py | 150 +- .../pipelines/_models/azure_availability.py | 138 +- .../pipelines/_models/cluster_log_conf.py | 154 +- .../_models/confidential_compute_type.py | 139 +- .../_models/confluence_connector_options.py | 146 +- .../confluent_schema_registry_options.py | 145 +- .../_models/connection_parameters.py | 142 +- .../pipelines/_models/connector_options.py | 206 ++- .../pipelines/_models/connector_type.py | 133 +- .../pipelines/_models/data_staging_options.py | 142 +- .../bundles/pipelines/_models/day_of_week.py | 140 +- .../pipelines/_models/dbfs_storage_info.py | 142 +- .../pipelines/_models/ebs_volume_type.py | 137 +- .../pipelines/_models/event_log_spec.py | 142 +- .../bundles/pipelines/_models/file_filter.py | 142 +- .../_models/file_ingestion_options.py | 163 +- .../file_ingestion_options_file_format.py | 138 +- ...ingestion_options_schema_evolution_mode.py | 144 +- .../bundles/pipelines/_models/file_library.py | 141 +- .../bundles/pipelines/_models/filters.py | 141 +- .../pipelines/_models/gcp_attributes.py | 150 +- .../pipelines/_models/gcp_availability.py | 138 +- .../pipelines/_models/gcs_storage_info.py | 142 +- .../pipelines/_models/google_ads_config.py | 142 +- .../google_ads_custom_report_options.py | 148 +- .../pipelines/_models/google_ads_options.py | 148 +- .../pipelines/_models/google_drive_options.py | 150 +- ..._drive_options_google_drive_entity_type.py | 138 +- .../pipelines/_models/ingestion_config.py | 147 +- .../ingestion_gateway_pipeline_definition.py | 150 +- .../_models/ingestion_pipeline_definition.py | 169 +- ...tion_pipeline_definition_fanout_options.py | 151 +- ...fic_config_query_based_connector_config.py | 156 +- ...ne_definition_workday_report_parameters.py | 149 +- .../pipelines/_models/init_script_info.py | 170 +- .../_models/jira_connector_options.py | 142 +- .../_models/json_transformer_options.py | 153 +- .../pipelines/_models/kafka_options.py | 149 +- .../bundles/pipelines/_models/lifecycle.py | 141 +- .../_models/linked_in_ads_options.py | 156 +- ...ons_linked_in_ads_custom_report_options.py | 191 ++- ...ptions_linked_in_ads_entity_granularity.py | 144 +- ...tom_report_options_linked_in_ads_finder.py | 140 +- ..._options_linked_in_ads_time_granularity.py | 140 +- .../pipelines/_models/local_file_info.py | 141 +- .../pipelines/_models/log_analytics_info.py | 141 +- .../pipelines/_models/marketo_options.py | 144 +- .../pipelines/_models/maven_library.py | 145 +- .../_models/meta_marketing_options.py | 154 +- ...ns_meta_marketing_custom_report_options.py | 151 +- .../pipelines/_models/notebook_library.py | 141 +- .../pipelines/_models/notifications.py | 141 +- .../_models/operation_time_window.py | 145 +- .../_models/outlook_attachment_mode.py | 139 +- .../pipelines/_models/outlook_body_format.py | 135 +- .../pipelines/_models/outlook_options.py | 152 +- .../bundles/pipelines/_models/parse_mode.py | 133 +- .../bundles/pipelines/_models/path_pattern.py | 141 +- .../bundles/pipelines/_models/pipeline.py | 186 ++- .../pipelines/_models/pipeline_cluster.py | 169 +- .../_models/pipeline_cluster_autoscale.py | 145 +- .../pipeline_cluster_autoscale_mode.py | 137 +- .../pipelines/_models/pipeline_library.py | 157 +- .../pipelines/_models/pipeline_permission.py | 145 +- .../_models/pipeline_permission_level.py | 137 +- .../_models/pipelines_environment.py | 142 +- .../_models/postgres_catalog_config.py | 146 +- .../pipelines/_models/postgres_slot_config.py | 142 +- .../_models/protobuf_transformer_options.py | 148 +- .../pipelines/_models/rabbitmq_options.py | 142 +- .../pipelines/_models/reddit_ads_options.py | 156 +- ...ptions_reddit_ads_custom_report_options.py | 151 +- .../bundles/pipelines/_models/report_spec.py | 145 +- .../pipelines/_models/restart_window.py | 145 +- .../bundles/pipelines/_models/run_as.py | 144 +- .../pipelines/_models/s3_storage_info.py | 142 +- .../_models/schema_registry_config.py | 145 +- .../bundles/pipelines/_models/schema_spec.py | 153 +- .../pipelines/_models/sharepoint_options.py | 150 +- ...arepoint_options_sharepoint_entity_type.py | 138 +- .../pipelines/_models/smartsheet_options.py | 142 +- .../_models/source_catalog_config.py | 146 +- .../pipelines/_models/source_config.py | 153 +- .../bundles/pipelines/_models/table_spec.py | 149 +- .../_models/table_specific_config.py | 177 +- .../_models/table_specific_config_scd_type.py | 137 +- .../pipelines/_models/tik_tok_ads_options.py | 162 +- ...tions_tik_tok_ads_custom_report_options.py | 159 +- .../tik_tok_ads_options_tik_tok_data_level.py | 138 +- ...tik_tok_ads_options_tik_tok_report_type.py | 138 +- .../bundles/pipelines/_models/transformer.py | 158 +- .../pipelines/_models/transformer_format.py | 136 +- .../pipelines/_models/volumes_storage_info.py | 142 +- .../_models/workspace_storage_info.py | 142 +- .../_models/zendesk_support_options.py | 142 +- .../bundles/quality_monitors/__init__.py | 150 +- .../quality_monitors/_models/lifecycle.py | 32 +- .../_models/monitor_cron_schedule.py | 36 +- .../monitor_cron_schedule_pause_status.py | 28 +- .../monitor_data_classification_config.py | 39 +- .../_models/monitor_destination.py | 32 +- .../_models/monitor_inference_log.py | 40 +- .../monitor_inference_log_problem_type.py | 28 +- .../_models/monitor_metric.py | 37 +- .../_models/monitor_metric_type.py | 33 +- .../_models/monitor_notifications.py | 36 +- .../_models/monitor_snapshot.py | 32 +- .../_models/monitor_time_series.py | 33 +- .../_models/quality_monitor.py | 71 +- .../bundles/registered_models/__init__.py | 58 +- .../registered_models/_models/lifecycle.py | 22 +- .../registered_models/_models/privilege.py | 71 +- .../_models/privilege_assignment.py | 26 +- .../_models/registered_model.py | 33 +- .../_models/registered_model_alias.py | 22 +- python/databricks/bundles/schemas/__init__.py | 47 +- .../bundles/schemas/_models/lifecycle.py | 21 +- .../bundles/schemas/_models/privilege.py | 70 +- .../schemas/_models/privilege_assignment.py | 20 +- .../bundles/schemas/_models/schema.py | 30 +- .../bundles/secret_scopes/__init__.py | 68 +- .../azure_key_vault_secret_scope_metadata.py | 29 +- .../secret_scopes/_models/lifecycle.py | 24 +- .../_models/scope_backend_type.py | 16 +- .../secret_scopes/_models/secret_scope.py | 40 +- .../_models/secret_scope_permission.py | 28 +- .../_models/secret_scope_permission_level.py | 19 +- python/databricks/bundles/secrets/__init__.py | 37 +- .../bundles/secrets/_models/lifecycle.py | 21 +- .../bundles/secrets/_models/privilege.py | 70 +- .../secrets/_models/privilege_assignment.py | 20 +- .../bundles/secrets/_models/secret.py | 30 +- .../bundles/sql_warehouses/__init__.py | 112 +- .../bundles/sql_warehouses/_models/channel.py | 35 +- .../sql_warehouses/_models/channel_name.py | 31 +- ...create_warehouse_request_warehouse_type.py | 26 +- .../_models/endpoint_tag_pair.py | 30 +- .../sql_warehouses/_models/endpoint_tags.py | 34 +- .../_models/lifecycle_with_started.py | 30 +- .../_models/spot_instance_policy.py | 31 +- .../sql_warehouses/_models/sql_warehouse.py | 99 +- .../_models/sql_warehouse_permission.py | 34 +- .../_models/warehouse_permission_level.py | 27 +- .../synced_database_tables/__init__.py | 80 +- .../_models/lifecycle.py | 25 +- .../_models/new_pipeline_spec.py | 26 +- .../_models/synced_database_table.py | 32 +- .../_models/synced_table_scheduling_policy.py | 20 +- .../_models/synced_table_spec.py | 42 +- .../synced_table_spec_pg_specific_type.py | 28 +- .../synced_table_spec_type_override.py | 36 +- .../vector_search_endpoints/__init__.py | 56 +- .../_models/endpoint_type.py | 15 +- .../_models/lifecycle.py | 23 +- .../_models/vector_search_endpoint.py | 42 +- .../vector_search_endpoint_permission.py | 31 +- ...vector_search_endpoint_permission_level.py | 19 +- .../bundles/vector_search_indexes/__init__.py | 124 +- .../delta_sync_vector_index_spec_request.py | 55 +- .../direct_access_vector_index_spec.py | 51 +- .../_models/embedding_source_column.py | 31 +- .../_models/embedding_vector_column.py | 31 +- .../_models/index_subtype.py | 23 +- .../_models/lifecycle.py | 31 +- .../_models/pipeline_type.py | 23 +- .../_models/privilege.py | 80 +- .../_models/privilege_assignment.py | 35 +- .../_models/vector_index_type.py | 23 +- .../_models/vector_search_index.py | 58 +- python/databricks/bundles/volumes/__init__.py | 52 +- .../bundles/volumes/_models/lifecycle.py | 23 +- .../bundles/volumes/_models/privilege.py | 72 +- .../volumes/_models/privilege_assignment.py | 22 +- .../bundles/volumes/_models/volume.py | 30 +- .../bundles/volumes/_models/volume_type.py | 14 +- .../core/_generated/alerts.py | 44 +- .../databricks_tests/core/_generated/apps.py | 36 +- .../core/_generated/catalogs.py | 24 +- .../core/_generated/cluster_policies.py | 28 +- .../core/_generated/clusters.py | 66 +- .../core/_generated/dashboards.py | 14 +- .../core/_generated/database_catalogs.py | 20 +- .../core/_generated/database_instances.py | 28 +- .../core/_generated/experiments.py | 28 +- .../core/_generated/external_locations.py | 38 +- .../core/_generated/features.py | 20 +- .../core/_generated/genie_spaces.py | 7 +- .../core/_generated/instance_pools.py | 58 +- .../core/_generated/job_runs.py | 22 +- .../databricks_tests/core/_generated/jobs.py | 64 +- .../core/_generated/mcp_services.py | 22 +- .../_generated/model_provider_services.py | 30 +- .../core/_generated/model_services.py | 26 +- .../_generated/model_serving_endpoints.py | 56 +- .../core/_generated/models.py | 28 +- .../core/_generated/pipelines.py | 46 +- .../core/_generated/quality_monitors.py | 93 +- .../core/_generated/registered_models.py | 20 +- .../core/_generated/schemas.py | 18 +- .../core/_generated/secret_scopes.py | 37 +- .../core/_generated/secrets.py | 20 +- .../core/_generated/sql_warehouses.py | 42 +- .../core/_generated/synced_database_tables.py | 14 +- .../_generated/vector_search_endpoints.py | 36 +- .../core/_generated/vector_search_indexes.py | 44 +- .../core/_generated/volumes.py | 20 +- 718 files changed, 53696 insertions(+), 14327 deletions(-) create mode 100644 .nextchanges/dependency-updates/go-sdk-v0.190.0.md create mode 100644 bundle/direct/dresources/configs/sql_warehouses.generated.yml create mode 100644 python/databricks/bundles/features/_models/sdp_source.py diff --git a/.codegen/_openapi_sha b/.codegen/_openapi_sha index 165042179b5..bd00da3e923 100644 --- a/.codegen/_openapi_sha +++ b/.codegen/_openapi_sha @@ -1 +1 @@ -1603cd4e00e208260dc36be11f5ba9822ff73360 \ No newline at end of file +155fbabb3529267ce58c1a37573911f59d90b06c \ No newline at end of file diff --git a/.codegen/cli.json b/.codegen/cli.json index 372ab22fee1..c3e114bb11e 100644 --- a/.codegen/cli.json +++ b/.codegen/cli.json @@ -4632,7 +4632,7 @@ "launch_stage": "PUBLIC_PREVIEW" }, "alert_configurations": { - "description": "Alerts to configure when this budget is in a triggered state. Budgets must have exactly one alert configuration.", + "description": "Alerts to configure when this budget is in a triggered state. Budgets must have at least one alert configuration.\nA budget can have up to four shared alerts per time period, each with a unique `quantity_threshold`.\nBudgets with `resource_type` set to `BUDGET_RESOURCE_TYPE_UNITY_AI_GATEWAY` can also use per-user alerts, with optional `principal_overrides`.\nThese budgets can have at most one per-user alert per time period, and can use shared alerts, a per-user alert, or both.\nThe `BLOCK_USAGE` action is only supported on budgets with `resource_type` set to `BUDGET_RESOURCE_TYPE_UNITY_AI_GATEWAY`.\nAt most one shared alert per time period can have a `BLOCK_USAGE` action, and it must be the shared alert with the highest threshold in that period.", "launch_stage": "PUBLIC_PREVIEW" }, "budget_configuration_id": { @@ -4794,7 +4794,7 @@ "launch_stage": "PUBLIC_PREVIEW" }, "alert_configurations": { - "description": "Alerts to configure when this budget is in a triggered state. Budgets must have exactly one alert configuration.", + "description": "Alerts to configure when this budget is in a triggered state. Budgets must have at least one alert configuration.\nA budget can have up to four shared alerts per time period, each with a unique `quantity_threshold`.\nBudgets with `resource_type` set to `BUDGET_RESOURCE_TYPE_UNITY_AI_GATEWAY` can also use per-user alerts, with optional `principal_overrides`.\nThese budgets can have at most one per-user alert per time period, and can use shared alerts, a per-user alert, or both.\nThe `BLOCK_USAGE` action is only supported on budgets with `resource_type` set to `BUDGET_RESOURCE_TYPE_UNITY_AI_GATEWAY`.\nAt most one shared alert per time period can have a `BLOCK_USAGE` action, and it must be the shared alert with the highest threshold in that period.", "launch_stage": "PUBLIC_PREVIEW" }, "display_name": { @@ -5320,7 +5320,7 @@ "launch_stage": "PUBLIC_PREVIEW" }, "alert_configurations": { - "description": "Alerts to configure when this budget is in a triggered state. Budgets must have exactly one alert configuration.", + "description": "Alerts to configure when this budget is in a triggered state. Budgets must have at least one alert configuration.\nA budget can have up to four shared alerts per time period, each with a unique `quantity_threshold`.\nBudgets with `resource_type` set to `BUDGET_RESOURCE_TYPE_UNITY_AI_GATEWAY` can also use per-user alerts, with optional `principal_overrides`.\nThese budgets can have at most one per-user alert per time period, and can use shared alerts, a per-user alert, or both.\nThe `BLOCK_USAGE` action is only supported on budgets with `resource_type` set to `BUDGET_RESOURCE_TYPE_UNITY_AI_GATEWAY`.\nAt most one shared alert per time period can have a `BLOCK_USAGE` action, and it must be the shared alert with the highest threshold in that period.", "launch_stage": "PUBLIC_PREVIEW" }, "budget_configuration_id": { @@ -5945,7 +5945,7 @@ ] }, "state": { - "description": "Serialized local config state after the operation. Its presence records\nwhether the resource still exists. Explicitly clearing state removes the resource;\nomitting it from the update mask leaves resource state unchanged.\nState is stored only on Resources. UpdateOperation accepts state but omits it\nfrom its response, as do subsequent GetOperation and ListOperations.\n\nMutable: may be updated after creation via UpdateOperation. When updating,\nthe caller must echo the last-observed `sequence_id` as a concurrency\nprecondition.\n\nThe string is not parsed; it is projected onto the corresponding Resource's\nstate. It is deliberately not google.protobuf.Value, whose only numeric case\nis `double number_value`, so parsing the client's JSON into it rewrites every\ninteger as a double - `1` reads back as `1.0`, which no longer deserializes\ninto an integer field - and silently loses precision above 2^53, which is\nwithin range for IDs the client records.\n\nA string rather than bytes: the payload is always UTF-8 JSON, and proto3 JSON\nmaps bytes to base64, which inflates every request and response by a third\nand makes state unreadable in logs and API responses. Both generate the same\nOpenAPI schema (\"type\": \"string\"), so the SDKs are identical either way.", + "description": "Serialized local config state after the operation. Its presence records\nwhether the resource is still tracked by the deployment. Explicitly clearing\nstate removes the resource from the deployment; omitting it from the update\nmask leaves resource state unchanged.\nState is stored only on Resources. UpdateOperation accepts state but omits it\nfrom its response, as do subsequent GetOperation and ListOperations.\n\nMutable: may be updated after creation via UpdateOperation. When updating,\nthe caller must echo the last-observed `sequence_id` as a concurrency\nprecondition.\n\nThe string is not parsed; it is projected onto the corresponding Resource's\nstate. It is deliberately not google.protobuf.Value, whose only numeric case\nis `double number_value`, so parsing the client's JSON into it rewrites every\ninteger as a double - `1` reads back as `1.0`, which no longer deserializes\ninto an integer field - and silently loses precision above 2^53, which is\nwithin range for IDs the client records.\n\nA string rather than bytes: the payload is always UTF-8 JSON, and proto3 JSON\nmaps bytes to base64, which inflates every request and response by a third\nand makes state unreadable in logs and API responses. Both generate the same\nOpenAPI schema (\"type\": \"string\"), so the SDKs are identical either way.", "launch_stage": "PRIVATE_PREVIEW", "behaviors": [ "OPTIONAL" @@ -10565,7 +10565,7 @@ "catalog.ListMcpServicesRequest": { "fields": { "page_size": { - "description": "Maximum number of MCP services to return. Defaults to 100 when unset or 0;\nthe maximum is 200. Use `page_token` to retrieve additional pages.", + "description": "Maximum number of MCP services to return. Defaults to 200 when unset or 0;\nthe maximum is 200. Use `page_token` to retrieve additional pages.", "launch_stage": "GA", "behaviors": [ "OPTIONAL" @@ -10646,7 +10646,7 @@ "catalog.ListModelProviderServicesRequest": { "fields": { "page_size": { - "description": "Maximum number of provider services to return. Defaults to 100 when unset or\n0; the maximum is 200. Use `page_token` to retrieve additional pages.", + "description": "Maximum number of provider services to return. Defaults to 200 when unset or\n0; the maximum is 200. Use `page_token` to retrieve additional pages.", "launch_stage": "GA", "behaviors": [ "OPTIONAL" @@ -10703,7 +10703,7 @@ "catalog.ListModelServicesRequest": { "fields": { "page_size": { - "description": "Maximum number of model services to return. Defaults to 100 when unset or 0;\nthe maximum is 200. Use `page_token` to retrieve additional pages.", + "description": "Maximum number of model services to return. Defaults to 200 when unset or 0;\nthe maximum is 200. Use `page_token` to retrieve additional pages.", "launch_stage": "GA", "behaviors": [ "OPTIONAL" @@ -11009,7 +11009,7 @@ "catalog.ListSkillsRequest": { "fields": { "page_size": { - "description": "Maximum number of skills to return. Defaults to 100 when unset or 0; the\nmaximum is 200. Use `page_token` to retrieve additional pages.", + "description": "Maximum number of skills to return. Defaults to 200 when unset or 0; the\nmaximum is 200. Use `page_token` to retrieve additional pages.", "launch_stage": "PUBLIC_BETA", "behaviors": [ "OPTIONAL" @@ -11869,7 +11869,7 @@ } }, "catalog.ModelProviderServiceConfigCustomProviderConfig": { - "description": "Custom OpenAI-compatible provider configuration with bearer-token\nauthentication.", + "description": "Custom OpenAI-compatible provider configuration.", "fields": { "direct": { "description": "Endpoint and authentication configuration for the custom provider.", @@ -35620,7 +35620,7 @@ "attachment": { "description": "Optional: The configuration for the files attached to the dashboard snapshot notification. It\ncontrols the underlying data of dashboard widgets that is attached as data files and which\npages are included in the attached PDF. If not specified, the notification is sent with only\nthe default snapshot.", "ref": "jobs.SubscriptionAttachment", - "launch_stage": "PRIVATE_PREVIEW", + "launch_stage": "PUBLIC_BETA", "behaviors": [ "OPTIONAL" ] @@ -35645,7 +35645,7 @@ "data_attachment": { "description": "Optional: The configuration for the underlying data of dashboard widgets attached as tabular\ndata files. If not specified, no tabular data is attached.", "ref": "jobs.SubscriptionAttachmentDataAttachment", - "launch_stage": "PRIVATE_PREVIEW", + "launch_stage": "PUBLIC_BETA", "behaviors": [ "OPTIONAL" ] @@ -35653,7 +35653,7 @@ "page_attachment": { "description": "Optional: The selection of dashboard pages included in the attached PDF. If not specified,\nonly the first page is included.", "ref": "jobs.SubscriptionAttachmentPageAttachment", - "launch_stage": "PRIVATE_PREVIEW", + "launch_stage": "PUBLIC_BETA", "behaviors": [ "OPTIONAL" ] @@ -35666,18 +35666,18 @@ "file_format": { "description": "The file format of the exported tabular data files.", "ref": "jobs.SubscriptionAttachmentDataAttachmentFileFormat", - "launch_stage": "PRIVATE_PREVIEW" + "launch_stage": "PUBLIC_BETA" }, "include_applied_filters": { "description": "When true, the dashboard's applied filter values are included alongside the tabular data as\nan additional attached file.", - "launch_stage": "PRIVATE_PREVIEW", + "launch_stage": "PUBLIC_BETA", "behaviors": [ "OPTIONAL" ] }, "widgets": { "description": "The widgets whose underlying data is exported. At least one widget must be referenced.", - "launch_stage": "PRIVATE_PREVIEW" + "launch_stage": "PUBLIC_BETA" } } }, @@ -35689,9 +35689,9 @@ "FILE_FORMAT_EXCEL" ], "enum_launch_stages": { - "FILE_FORMAT_CSV": "PRIVATE_PREVIEW", - "FILE_FORMAT_EXCEL": "PRIVATE_PREVIEW", - "FILE_FORMAT_TSV": "PRIVATE_PREVIEW" + "FILE_FORMAT_CSV": "PUBLIC_BETA", + "FILE_FORMAT_EXCEL": "PUBLIC_BETA", + "FILE_FORMAT_TSV": "PUBLIC_BETA" } }, "jobs.SubscriptionAttachmentDataAttachmentWidget": { @@ -35699,11 +35699,11 @@ "fields": { "page_id": { "description": "The URL identifier of the page containing the widget. This is the value of the `name`\nfield of the corresponding entry under `pages` in the serialized dashboard definition.\nFormat: alphanumeric, hyphen, and underscore characters allowed, max length 63.", - "launch_stage": "PRIVATE_PREVIEW" + "launch_stage": "PUBLIC_BETA" }, "widget_id": { "description": "The URL identifier of the widget. This is the value of the `name` field of the\ncorresponding widget under that page's `layout` in the serialized dashboard definition.\nFormat: alphanumeric, hyphen, and underscore characters allowed, max length 63.", - "launch_stage": "PRIVATE_PREVIEW" + "launch_stage": "PUBLIC_BETA" } } }, @@ -35712,7 +35712,7 @@ "fields": { "page_ids": { "description": "The identifiers of the pages to include in the PDF. Only specified when `page_selection` is\n`PAGE_SELECTION_SUBSET`. Each value is the identifier of a page in the dashboard's\nserialized definition returned by the Get dashboard API (`serialized_dashboard`): the value\nof the `name` field of an entry under `pages`.\nFormat: alphanumeric, hyphen, and underscore characters allowed, max length 63.", - "launch_stage": "PRIVATE_PREVIEW", + "launch_stage": "PUBLIC_BETA", "behaviors": [ "OPTIONAL" ] @@ -35720,7 +35720,7 @@ "page_selection": { "description": "How the attached pages are selected. When set to `PAGE_SELECTION_SUBSET`, the pages to\ninclude are listed in `page_ids`.", "ref": "jobs.SubscriptionAttachmentPageAttachmentPageSelection", - "launch_stage": "PRIVATE_PREVIEW", + "launch_stage": "PUBLIC_BETA", "behaviors": [ "OPTIONAL" ] @@ -35734,8 +35734,8 @@ "PAGE_SELECTION_SUBSET" ], "enum_launch_stages": { - "PAGE_SELECTION_ALL": "PRIVATE_PREVIEW", - "PAGE_SELECTION_SUBSET": "PRIVATE_PREVIEW" + "PAGE_SELECTION_ALL": "PUBLIC_BETA", + "PAGE_SELECTION_SUBSET": "PUBLIC_BETA" } }, "jobs.SubscriptionSubscriber": { @@ -39762,6 +39762,14 @@ "OPTIONAL" ] }, + "sdp_source": { + "description": "A Spark Declarative Pipeline data source.", + "ref": "ml.SdpSource", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OPTIONAL" + ] + }, "stream_source": { "description": "A Stream data source.", "ref": "ml.StreamSource", @@ -43864,6 +43872,74 @@ } } }, + "ml.SdpSource": { + "description": "A Spark Declarative Pipeline (SDP) data source.", + "fields": { + "dataframe_schema": { + "description": "Schema of the resulting dataframe after transformations, in Spark StructType JSON format (from df.schema.json()).\nRequired if transformation_sql is specified.\nExample: {\"type\":\"struct\",\"fields\":[{\"name\":\"col_a\",\"type\":\"integer\",\"nullable\":true,\"metadata\":{}},{\"name\":\"col_c\",\"type\":\"integer\",\"nullable\":true,\"metadata\":{}}]}", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OPTIONAL" + ] + }, + "filter_condition": { + "description": "Single WHERE clause to filter the pipeline's output table before applying transformations. Will be row-wise evaluated, so should only include conditionals and projections.", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OPTIONAL" + ] + }, + "input_tables": { + "description": "The full three-part names of the tables the pipeline reads from. Required to materialize this\nfeature with a TableTrigger, which starts a run when these tables update. Each table must exist\nwhen the feature is created. After the pipeline runs, Feature Store checks that it reads from\nthese tables, and the run fails if it doesn't. At most 100 tables.", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OPTIONAL" + ] + }, + "output_table": { + "description": "The full three-part (catalog, schema, table) name of the pipeline's output table to read from.\nThe table doesn't need to exist until the pipeline first runs. Required.", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OPTIONAL" + ] + }, + "pipeline_id": { + "description": "The ID of the Spark Declarative Pipeline that Feature Store runs to produce this source, before\neach materialization or backfill. The pipeline must already exist when the feature is created,\nand you need CAN_RUN, CAN_MANAGE, or owner permission on it. It doesn't need to have run yet.\nDeleting the feature doesn't delete the pipeline. If the pipeline is deleted, this field reads\nback empty. Required.", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OPTIONAL" + ] + }, + "pipeline_state": { + "description": "State tracking pipeline execution and output table validation. PENDING until the pipeline\nhas run at least once and created the output table; then RESOLVED on success, or ERROR on failure.", + "ref": "ml.SdpSourceState", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OUTPUT_ONLY" + ] + }, + "transformation_sql": { + "description": "A single SQL SELECT expression applied after filter_condition.\nShould contain all the columns needed (eg. \"SELECT *, col_a + col_b AS col_c FROM x.y.z WHERE col_a \u003e 0\" would have `transformation_sql` \"*, col_a + col_b AS col_c\")\nIf transformation_sql is not provided, all columns of the pipeline's output table are present in the DataSource dataframe.", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OPTIONAL" + ] + } + } + }, + "ml.SdpSourceState": { + "description": "State of the SDP source relative to pipeline execution and output table validation.", + "enum": [ + "PENDING", + "RESOLVED", + "ERROR" + ], + "enum_launch_stages": { + "ERROR": "PRIVATE_PREVIEW", + "PENDING": "PRIVATE_PREVIEW", + "RESOLVED": "PRIVATE_PREVIEW" + } + }, "ml.SearchExperiments": { "fields": { "filter": { @@ -47952,7 +48028,7 @@ "launch_stage": "GA" }, "order_by": { - "description": "A list of strings specifying the order of results.\nSupported order_by fields are id and name. The default is id asc.\nThis field is optional.", + "description": "A list of strings specifying the order of results.\nThe supported order_by field is id. The default is id asc.\nThis field is optional.", "launch_stage": "GA" }, "page_token": { @@ -61644,11 +61720,6 @@ "description": "Name of the share.", "launch_stage": "GA" }, - "replication_config": { - "description": "Configuration for share replication.", - "ref": "sharing.ReplicationConfig", - "launch_stage": "PRIVATE_PREVIEW" - }, "storage_root": { "description": "Storage root URL for the share.", "launch_stage": "GA" @@ -62626,15 +62697,6 @@ } } }, - "sharing.ReplicationConfig": { - "description": "Configuration for Delta Sharing replication on a share. When enabled, the share's data is\ncopied into each recipient's region so recipients can query the share locally.", - "fields": { - "enabled": { - "description": "Whether replication is enabled for the share.", - "launch_stage": "PRIVATE_PREVIEW" - } - } - }, "sharing.RetrieveTokenRequest": {}, "sharing.RetrieveTokenResponse": { "fields": { @@ -62727,11 +62789,6 @@ "description": "Username of current owner of share.", "launch_stage": "GA" }, - "replication_config": { - "description": "Configuration for share replication.", - "ref": "sharing.ReplicationConfig", - "launch_stage": "PRIVATE_PREVIEW" - }, "storage_location": { "description": "Storage Location URL (full path) for the share.", "launch_stage": "GA", @@ -64183,6 +64240,13 @@ "description": "warehouse creator name", "launch_stage": "GA" }, + "customer_version_label": { + "description": "Compute version label for a REALTIME warehouse. This field is output only.", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OUTPUT_ONLY" + ] + }, "enable_photon": { "description": "Configures whether the warehouse should use Photon optimized clusters.\n\nDefaults to true.", "launch_stage": "GA" @@ -64196,6 +64260,10 @@ "deprecated": true, "launch_stage": "GA" }, + "max_dbu": { + "description": "Maximum compute capacity of a real-time SQL warehouse, in DBU per hour.\nMust be a multiple of 3 between 3 and 960. The effective maximum cannot be\nlower than the warehouse's minimum capacity.", + "launch_stage": "PRIVATE_PREVIEW" + }, "max_num_clusters": { "description": "Maximum number of clusters that the autoscaler will create to handle\nconcurrent queries.\n\nSupported values:\n- Must be \u003e= min_num_clusters\n- Must be \u003c= 40.\n\nDefaults to min_clusters if unset.", "launch_stage": "GA" @@ -64208,6 +64276,10 @@ "description": "Logical name for the cluster.\n\nSupported values:\n- Must be unique within an org.\n- Must be less than 100 characters.", "launch_stage": "GA" }, + "reyden_cluster_size_multiplier": { + "description": "Cluster size multiplier for Reyden warehouses. Scales the effective compute\ncapacity relative to the nominal virtual cluster size. Only meaningful when\nthe warehouse is of Reyden type.\n\nAccepted range: 1.0 to the workspace-configured maximum (decimals\nallowed). The default maximum is 100.0.\n- Values below 1.0 are normalized to 1.0.\n- Values above the configured maximum are capped at that maximum.\n- When absent, the warehouse is treated as having a multiplier of 1.0.", + "launch_stage": "PRIVATE_PREVIEW" + }, "spot_instance_policy": { "description": "Configurations whether the endpoint should use spot instances.", "ref": "sql.SpotInstancePolicy", @@ -64655,6 +64727,13 @@ "description": "warehouse creator name", "launch_stage": "GA" }, + "customer_version_label": { + "description": "Compute version label for a REALTIME warehouse. This field is output only.", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OUTPUT_ONLY" + ] + }, "enable_photon": { "description": "Configures whether the warehouse should use Photon optimized clusters.\n\nDefaults to true.", "launch_stage": "GA" @@ -64668,6 +64747,10 @@ "deprecated": true, "launch_stage": "GA" }, + "max_dbu": { + "description": "Maximum compute capacity of a real-time SQL warehouse, in DBU per hour.\nMust be a multiple of 3 between 3 and 960. The effective maximum cannot be\nlower than the warehouse's minimum capacity.", + "launch_stage": "PRIVATE_PREVIEW" + }, "max_num_clusters": { "description": "Maximum number of clusters that the autoscaler will create to handle\nconcurrent queries.\n\nSupported values:\n- Must be \u003e= min_num_clusters\n- Must be \u003c= 40.\n\nDefaults to min_clusters if unset.", "launch_stage": "GA" @@ -64680,6 +64763,10 @@ "description": "Logical name for the cluster.\n\nSupported values:\n- Must be unique within an org.\n- Must be less than 100 characters.", "launch_stage": "GA" }, + "reyden_cluster_size_multiplier": { + "description": "Cluster size multiplier for Reyden warehouses. Scales the effective compute\ncapacity relative to the nominal virtual cluster size. Only meaningful when\nthe warehouse is of Reyden type.\n\nAccepted range: 1.0 to the workspace-configured maximum (decimals\nallowed). The default maximum is 100.0.\n- Values below 1.0 are normalized to 1.0.\n- Values above the configured maximum are capped at that maximum.\n- When absent, the warehouse is treated as having a multiplier of 1.0.", + "launch_stage": "PRIVATE_PREVIEW" + }, "spot_instance_policy": { "description": "Configurations whether the endpoint should use spot instances.", "ref": "sql.SpotInstancePolicy", @@ -64778,6 +64865,13 @@ "description": "warehouse creator name", "launch_stage": "GA" }, + "customer_version_label": { + "description": "Compute version label for a REALTIME warehouse. This field is output only.", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OUTPUT_ONLY" + ] + }, "enable_photon": { "description": "Configures whether the warehouse should use Photon optimized clusters.\n\nDefaults to true.", "launch_stage": "GA" @@ -64810,6 +64904,10 @@ "description": "the jdbc connection string for this warehouse", "launch_stage": "GA" }, + "max_dbu": { + "description": "Maximum compute capacity of a real-time SQL warehouse, in DBU per hour.\nMust be a multiple of 3 between 3 and 960. The effective maximum cannot be\nlower than the warehouse's minimum capacity.", + "launch_stage": "PRIVATE_PREVIEW" + }, "max_num_clusters": { "description": "Maximum number of clusters that the autoscaler will create to handle\nconcurrent queries.\n\nSupported values:\n- Must be \u003e= min_num_clusters\n- Must be \u003c= 40.\n\nDefaults to min_clusters if unset.", "launch_stage": "GA" @@ -64836,6 +64934,10 @@ "ref": "sql.OdbcParams", "launch_stage": "GA" }, + "reyden_cluster_size_multiplier": { + "description": "Cluster size multiplier for Reyden warehouses. Scales the effective compute\ncapacity relative to the nominal virtual cluster size. Only meaningful when\nthe warehouse is of Reyden type.\n\nAccepted range: 1.0 to the workspace-configured maximum (decimals\nallowed). The default maximum is 100.0.\n- Values below 1.0 are normalized to 1.0.\n- Values above the configured maximum are capped at that maximum.\n- When absent, the warehouse is treated as having a multiplier of 1.0.", + "launch_stage": "PRIVATE_PREVIEW" + }, "spot_instance_policy": { "description": "Configurations whether the endpoint should use spot instances.", "ref": "sql.SpotInstancePolicy", @@ -65133,6 +65235,13 @@ "description": "warehouse creator name", "launch_stage": "GA" }, + "customer_version_label": { + "description": "Compute version label for a REALTIME warehouse. This field is output only.", + "launch_stage": "PRIVATE_PREVIEW", + "behaviors": [ + "OUTPUT_ONLY" + ] + }, "enable_photon": { "description": "Configures whether the warehouse should use Photon optimized clusters.\n\nDefaults to true.", "launch_stage": "GA" @@ -65165,6 +65274,10 @@ "description": "the jdbc connection string for this warehouse", "launch_stage": "GA" }, + "max_dbu": { + "description": "Maximum compute capacity of a real-time SQL warehouse, in DBU per hour.\nMust be a multiple of 3 between 3 and 960. The effective maximum cannot be\nlower than the warehouse's minimum capacity.", + "launch_stage": "PRIVATE_PREVIEW" + }, "max_num_clusters": { "description": "Maximum number of clusters that the autoscaler will create to handle\nconcurrent queries.\n\nSupported values:\n- Must be \u003e= min_num_clusters\n- Must be \u003c= 40.\n\nDefaults to min_clusters if unset.", "launch_stage": "GA" @@ -65191,6 +65304,10 @@ "ref": "sql.OdbcParams", "launch_stage": "GA" }, + "reyden_cluster_size_multiplier": { + "description": "Cluster size multiplier for Reyden warehouses. Scales the effective compute\ncapacity relative to the nominal virtual cluster size. Only meaningful when\nthe warehouse is of Reyden type.\n\nAccepted range: 1.0 to the workspace-configured maximum (decimals\nallowed). The default maximum is 100.0.\n- Values below 1.0 are normalized to 1.0.\n- Values above the configured maximum are capped at that maximum.\n- When absent, the warehouse is treated as having a multiplier of 1.0.", + "launch_stage": "PRIVATE_PREVIEW" + }, "spot_instance_policy": { "description": "Configurations whether the endpoint should use spot instances.", "ref": "sql.SpotInstancePolicy", @@ -77537,7 +77654,7 @@ }, { "name": "CreateSkill", - "description": "Creates a skill in a Unity Catalog schema and provisions its managed bundle\nstorage. Specify its name in `skill_id`. The request contains an optional\ncomment but no bundle bytes. Upload bundle files through the Files API,\nthen call FinalizeSkill.\n\nYou must be the owner of the parent schema or have `CREATE_VOLUME` and\n`USE_SCHEMA` on it, plus `USE_CATALOG` on the parent catalog.", + "description": "Creates a skill in a Unity Catalog schema and provisions its managed bundle\nstorage. Specify its name in `skill_id`. The request contains an optional\ncomment but no bundle bytes.\n\nTo publish a skill:\n\n1. Call CreateSkill to create the skill in Unity Catalog.\n2. Upload every bundle file, including `SKILL.md` at the bundle root,\nwith the Files API:\n`PUT /api/2.0/fs/files/Skills/{catalog}/{schema}/{skill}/{file_path}`.\nThe CreateSkill caller owns the skill and can upload. Other callers\nneed `READ_VOLUME` and `WRITE_VOLUME` on the skill.\n3. Call FinalizeSkill to validate `SKILL.md` and set the skill's\n`bundle_name` and `description` from its frontmatter `name` and\n`description`.\n\nYou must be the owner of the parent schema or have `CREATE_VOLUME` and\n`USE_SCHEMA` on it, plus `USE_CATALOG` on the parent catalog.", "summary": "Create a skill.", "path": "/api/2.1/unity-catalog/skills", "can_use_json": true, @@ -78127,7 +78244,7 @@ }, { "name": "FinalizeSkill", - "description": "Finalizes a skill after its bundle is uploaded. This method reads SKILL.md\nthrough the Files API using the caller's authorization. Its YAML\nfrontmatter must contain an agentskills.io-compliant `name` and a nonblank\n`description` within the configured UTF-8 byte limit. On success, it\nreplaces `bundle_name` and `description`; refreshes `finalize_time`,\n`update_time`, and `updated_by`; and returns the updated skill. `comment` is\npreserved. Re-finalization uses the latest SKILL.md and is last-write-wins\nwithout an etag precondition. Validation failures do not change metadata.\n\nYou must be the owner of the skill or have `READ_VOLUME` on it, plus\n`USE_CATALOG` on the parent catalog and `USE_SCHEMA` on the parent schema.", + "description": "Finalizes a skill after its bundle files are uploaded or changed through\nthe Files API. Call it again after every bundle change. This method reads\nSKILL.md through the Files API using the caller's authorization. Its YAML\nfrontmatter must contain an agentskills.io-compliant `name` and a nonblank\n`description` within the configured UTF-8 byte limit. On success, it\nreplaces `bundle_name` and `description`; refreshes `finalize_time`,\n`update_time`, and `updated_by`; and returns the updated skill. `comment` is\npreserved. Re-finalization uses the latest SKILL.md and is last-write-wins\nwithout an etag precondition. Validation failures do not change metadata.\n\nYou must be the owner of the skill or have `READ_VOLUME` on it, plus\n`USE_CATALOG` on the parent catalog and `USE_SCHEMA` on the parent schema.", "summary": "Finalize a skill.", "path": "/api/2.1/unity-catalog/{name=skills/*}/finalize", "has_required_positional_arguments": true, @@ -78521,7 +78638,7 @@ "all_fields": [ { "name": "page_size", - "description": "Maximum number of MCP services to return. Defaults to 100 when unset or 0;\nthe maximum is 200. Use `page_token` to retrieve additional pages.", + "description": "Maximum number of MCP services to return. Defaults to 200 when unset or 0;\nthe maximum is 200. Use `page_token` to retrieve additional pages.", "is_query": true, "entity": { "is_int": true @@ -78592,7 +78709,7 @@ "all_fields": [ { "name": "page_size", - "description": "Maximum number of provider services to return. Defaults to 100 when unset or\n0; the maximum is 200. Use `page_token` to retrieve additional pages.", + "description": "Maximum number of provider services to return. Defaults to 200 when unset or\n0; the maximum is 200. Use `page_token` to retrieve additional pages.", "is_query": true, "entity": { "is_int": true @@ -78663,7 +78780,7 @@ "all_fields": [ { "name": "page_size", - "description": "Maximum number of model services to return. Defaults to 100 when unset or 0;\nthe maximum is 200. Use `page_token` to retrieve additional pages.", + "description": "Maximum number of model services to return. Defaults to 200 when unset or 0;\nthe maximum is 200. Use `page_token` to retrieve additional pages.", "is_query": true, "entity": { "is_int": true @@ -78758,7 +78875,7 @@ "all_fields": [ { "name": "page_size", - "description": "Maximum number of skills to return. Defaults to 100 when unset or 0; the\nmaximum is 200. Use `page_token` to retrieve additional pages.", + "description": "Maximum number of skills to return. Defaults to 200 when unset or 0; the\nmaximum is 200. Use `page_token` to retrieve additional pages.", "is_query": true, "entity": { "is_int": true @@ -79543,7 +79660,7 @@ }, { "name": "UpdateSkill", - "description": "Updates a skill. Only fields named in `update_mask` are changed; currently\nonly `comment` is supported. The resource name is immutable. Optionally\nsupply an `etag` to make the update conditional on the skill not having\nchanged since it was read. Bundle files, grants, tags, and ownership are\nunchanged.\n\nYou must be the owner of the skill or have `MANAGE` on it, plus\n`USE_CATALOG` on the parent catalog and `USE_SCHEMA` on the parent schema.", + "description": "Updates a skill. Only fields named in `update_mask` are changed; currently\nonly `comment` is supported. The resource name is immutable. Optionally\nsupply an `etag` to make the update conditional on the skill not having\nchanged since it was read. Bundle files, grants, tags, and ownership are\nunchanged.\n\nTo update the bundle, add, overwrite (`overwrite=true`), or delete files\nunder `/Skills/{catalog}/{schema}/{skill}/` with the Files API, then call\nFinalizeSkill. Changing bundle files requires being the owner of the skill\nor having `READ_VOLUME` and `WRITE_VOLUME` on it; `MANAGE` does not grant\naccess to bundle files.\n\nTo call UpdateSkill, you must be the owner of the skill or have `MANAGE` on\nit, plus `USE_CATALOG` on the parent catalog and `USE_SCHEMA` on the parent\nschema.", "summary": "Update a skill.", "path": "/api/2.1/unity-catalog/{name=skills/*}", "can_use_json": true, @@ -93504,7 +93621,7 @@ }, { "name": "UpdateOperation", - "description": "Updates a resource operation's mutable fields.\n\n`state`, `error_message`, `resource_id`, `status`, and `dashboard_metadata`\nmay be updated, independently; `update_mask` must contain only those paths.\nAll other fields are immutable. The update is guarded by an\noptimistic-concurrency check: the caller sets `operation.sequence_id` to the\nvalue it last observed, and the server rejects the update with `ABORTED` if\nthe operation has been modified since. On success the server increments\n`sequence_id`; updates to `state`, `resource_id`, and `dashboard_metadata`\nare mirrored onto the corresponding deployment-level resource. Listing\n`state` in `update_mask` with no value clears it, which removes the\nresource, so a delete that is retried until it succeeds must clear `state`.\nThe parent version must be in progress, and after the update is applied a\nsucceeded operation cannot carry an `error_message`. See the `state` and\n`resource_id` fields for the rest.", + "description": "Updates a resource operation's mutable fields.\n\n`state`, `error_message`, `resource_id`, `status`, and `dashboard_metadata`\nmay be updated, independently; `update_mask` must contain only those paths.\nAll other fields are immutable. The update is guarded by an\noptimistic-concurrency check: the caller sets `operation.sequence_id` to the\nvalue it last observed, and the server rejects the update with `ABORTED` if\nthe operation has been modified since. On success the server increments\n`sequence_id`; updates to `state`, `resource_id`, and `dashboard_metadata`\nare mirrored onto the corresponding deployment-level resource. Listing\n`state` in `update_mask` with no value clears it, which removes the\ntracked resource, so a delete or unbind that succeeds must clear `state`.\nThe parent version must be in progress, and after the update is applied a\nsucceeded operation cannot carry an `error_message`. See the `state` and\n`resource_id` fields for the rest.", "summary": "Update an operation.", "path": "/api/2.0/bundle/{name=deployments/*/versions/*/operations/*}", "can_use_json": true, @@ -93837,7 +93954,7 @@ }, { "name": "state", - "description": "Serialized local config state after the operation. Its presence records\nwhether the resource still exists. Explicitly clearing state removes the resource;\nomitting it from the update mask leaves resource state unchanged.\nState is stored only on Resources. UpdateOperation accepts state but omits it\nfrom its response, as do subsequent GetOperation and ListOperations.\n\nMutable: may be updated after creation via UpdateOperation. When updating,\nthe caller must echo the last-observed `sequence_id` as a concurrency\nprecondition.\n\nThe string is not parsed; it is projected onto the corresponding Resource's\nstate. It is deliberately not google.protobuf.Value, whose only numeric case\nis `double number_value`, so parsing the client's JSON into it rewrites every\ninteger as a double - `1` reads back as `1.0`, which no longer deserializes\ninto an integer field - and silently loses precision above 2^53, which is\nwithin range for IDs the client records.\n\nA string rather than bytes: the payload is always UTF-8 JSON, and proto3 JSON\nmaps bytes to base64, which inflates every request and response by a third\nand makes state unreadable in logs and API responses. Both generate the same\nOpenAPI schema (\"type\": \"string\"), so the SDKs are identical either way.", + "description": "Serialized local config state after the operation. Its presence records\nwhether the resource is still tracked by the deployment. Explicitly clearing\nstate removes the resource from the deployment; omitting it from the update\nmask leaves resource state unchanged.\nState is stored only on Resources. UpdateOperation accepts state but omits it\nfrom its response, as do subsequent GetOperation and ListOperations.\n\nMutable: may be updated after creation via UpdateOperation. When updating,\nthe caller must echo the last-observed `sequence_id` as a concurrency\nprecondition.\n\nThe string is not parsed; it is projected onto the corresponding Resource's\nstate. It is deliberately not google.protobuf.Value, whose only numeric case\nis `double number_value`, so parsing the client's JSON into it rewrites every\ninteger as a double - `1` reads back as `1.0`, which no longer deserializes\ninto an integer field - and silently loses precision above 2^53, which is\nwithin range for IDs the client records.\n\nA string rather than bytes: the payload is always UTF-8 JSON, and proto3 JSON\nmaps bytes to base64, which inflates every request and response by a third\nand makes state unreadable in logs and API responses. Both generate the same\nOpenAPI schema (\"type\": \"string\"), so the SDKs are identical either way.", "is_request_body_field": true, "entity": { "is_string": true @@ -177531,7 +177648,7 @@ }, { "name": "order_by", - "description": "A list of strings specifying the order of results.\nSupported order_by fields are id and name. The default is id asc.\nThis field is optional.", + "description": "A list of strings specifying the order of results.\nThe supported order_by field is id. The default is id asc.\nThis field is optional.", "is_query": true, "entity": { "pascal_name": "List", @@ -211624,16 +211741,6 @@ "is_string": true } }, - { - "name": "replication_config", - "description": "Configuration for share replication.", - "is_request_body_field": true, - "is_optional_object": true, - "entity": { - "pascal_name": "ReplicationConfig", - "is_object": true - } - }, { "name": "storage_root", "description": "Storage root URL for the share.", @@ -225092,6 +225199,15 @@ "is_string": true } }, + { + "name": "customer_version_label", + "description": "Compute version label for a REALTIME warehouse. This field is output only.", + "is_output_only": true, + "is_request_body_field": true, + "entity": { + "is_string": true + } + }, { "name": "enable_photon", "description": "Configures whether the warehouse should use Photon optimized clusters.\n\nDefaults to true.", @@ -225116,6 +225232,14 @@ "is_string": true } }, + { + "name": "max_dbu", + "description": "Maximum compute capacity of a real-time SQL warehouse, in DBU per hour.\nMust be a multiple of 3 between 3 and 960. The effective maximum cannot be\nlower than the warehouse's minimum capacity.", + "is_request_body_field": true, + "entity": { + "is_int": true + } + }, { "name": "max_num_clusters", "description": "Maximum number of clusters that the autoscaler will create to handle\nconcurrent queries.\n\nSupported values:\n- Must be \u003e= min_num_clusters\n- Must be \u003c= 40.\n\nDefaults to min_clusters if unset.", @@ -225140,6 +225264,14 @@ "is_string": true } }, + { + "name": "reyden_cluster_size_multiplier", + "description": "Cluster size multiplier for Reyden warehouses. Scales the effective compute\ncapacity relative to the nominal virtual cluster size. Only meaningful when\nthe warehouse is of Reyden type.\n\nAccepted range: 1.0 to the workspace-configured maximum (decimals\nallowed). The default maximum is 100.0.\n- Values below 1.0 are normalized to 1.0.\n- Values above the configured maximum are capped at that maximum.\n- When absent, the warehouse is treated as having a multiplier of 1.0.", + "is_request_body_field": true, + "entity": { + "is_float64": true + } + }, { "name": "spot_instance_policy", "description": "Configurations whether the endpoint should use spot instances.", @@ -225714,6 +225846,15 @@ "is_string": true } }, + { + "name": "customer_version_label", + "description": "Compute version label for a REALTIME warehouse. This field is output only.", + "is_output_only": true, + "is_request_body_field": true, + "entity": { + "is_string": true + } + }, { "name": "enable_photon", "description": "Configures whether the warehouse should use Photon optimized clusters.\n\nDefaults to true.", @@ -225747,6 +225888,14 @@ "is_string": true } }, + { + "name": "max_dbu", + "description": "Maximum compute capacity of a real-time SQL warehouse, in DBU per hour.\nMust be a multiple of 3 between 3 and 960. The effective maximum cannot be\nlower than the warehouse's minimum capacity.", + "is_request_body_field": true, + "entity": { + "is_int": true + } + }, { "name": "max_num_clusters", "description": "Maximum number of clusters that the autoscaler will create to handle\nconcurrent queries.\n\nSupported values:\n- Must be \u003e= min_num_clusters\n- Must be \u003c= 40.\n\nDefaults to min_clusters if unset.", @@ -225771,6 +225920,14 @@ "is_string": true } }, + { + "name": "reyden_cluster_size_multiplier", + "description": "Cluster size multiplier for Reyden warehouses. Scales the effective compute\ncapacity relative to the nominal virtual cluster size. Only meaningful when\nthe warehouse is of Reyden type.\n\nAccepted range: 1.0 to the workspace-configured maximum (decimals\nallowed). The default maximum is 100.0.\n- Values below 1.0 are normalized to 1.0.\n- Values above the configured maximum are capped at that maximum.\n- When absent, the warehouse is treated as having a multiplier of 1.0.", + "is_request_body_field": true, + "entity": { + "is_float64": true + } + }, { "name": "spot_instance_policy", "description": "Configurations whether the endpoint should use spot instances.", diff --git a/.nextchanges/dependency-updates/go-sdk-v0.190.0.md b/.nextchanges/dependency-updates/go-sdk-v0.190.0.md new file mode 100644 index 00000000000..e36c8f8d41f --- /dev/null +++ b/.nextchanges/dependency-updates/go-sdk-v0.190.0.md @@ -0,0 +1 @@ +* Update the Go SDK to [v0.190.0](https://github.com/databricks/databricks-sdk-go/releases/tag/v0.190.0). diff --git a/acceptance/bundle/refschema/out.fields.txt b/acceptance/bundle/refschema/out.fields.txt index 8663853ceb1..978c468c0ac 100644 --- a/acceptance/bundle/refschema/out.fields.txt +++ b/acceptance/bundle/refschema/out.fields.txt @@ -967,6 +967,15 @@ resources.features.*.source.request_source.flat_schema.fields []ml.FieldDefiniti resources.features.*.source.request_source.flat_schema.fields[*] ml.FieldDefinition ALL resources.features.*.source.request_source.flat_schema.fields[*].data_type ml.ScalarDataType ALL resources.features.*.source.request_source.flat_schema.fields[*].name string ALL +resources.features.*.source.sdp_source *ml.SdpSource ALL +resources.features.*.source.sdp_source.dataframe_schema string ALL +resources.features.*.source.sdp_source.filter_condition string ALL +resources.features.*.source.sdp_source.input_tables []string ALL +resources.features.*.source.sdp_source.input_tables[*] string ALL +resources.features.*.source.sdp_source.output_table string ALL +resources.features.*.source.sdp_source.pipeline_id string ALL +resources.features.*.source.sdp_source.pipeline_state ml.SdpSourceState ALL +resources.features.*.source.sdp_source.transformation_sql string ALL resources.features.*.source.stream_source *ml.StreamSource ALL resources.features.*.source.stream_source.dataframe_schema string ALL resources.features.*.source.stream_source.filter_condition string ALL @@ -4396,6 +4405,7 @@ resources.sql_warehouses.*.channel.dbsql_version string ALL resources.sql_warehouses.*.channel.name sql.ChannelName ALL resources.sql_warehouses.*.cluster_size string ALL resources.sql_warehouses.*.creator_name string ALL +resources.sql_warehouses.*.customer_version_label string ALL resources.sql_warehouses.*.enable_photon bool ALL resources.sql_warehouses.*.enable_serverless_compute bool ALL resources.sql_warehouses.*.health *sql.EndpointHealth REMOTE @@ -4416,6 +4426,7 @@ resources.sql_warehouses.*.lifecycle *resources.LifecycleWithStarted INPUT resources.sql_warehouses.*.lifecycle resources.Lifecycle INPUT resources.sql_warehouses.*.lifecycle.prevent_destroy bool INPUT resources.sql_warehouses.*.lifecycle.started *bool ALL +resources.sql_warehouses.*.max_dbu int ALL resources.sql_warehouses.*.max_num_clusters int ALL resources.sql_warehouses.*.min_num_clusters int ALL resources.sql_warehouses.*.modified_status string INPUT @@ -4427,6 +4438,7 @@ resources.sql_warehouses.*.odbc_params.hostname string REMOTE resources.sql_warehouses.*.odbc_params.path string REMOTE resources.sql_warehouses.*.odbc_params.port int REMOTE resources.sql_warehouses.*.odbc_params.protocol string REMOTE +resources.sql_warehouses.*.reyden_cluster_size_multiplier float64 ALL resources.sql_warehouses.*.spot_instance_policy sql.SpotInstancePolicy ALL resources.sql_warehouses.*.state sql.State REMOTE resources.sql_warehouses.*.statement_timeout int ALL diff --git a/bundle/direct/dresources/configs/features.generated.yml b/bundle/direct/dresources/configs/features.generated.yml index 51dae44ab90..e845514a9e7 100644 --- a/bundle/direct/dresources/configs/features.generated.yml +++ b/bundle/direct/dresources/configs/features.generated.yml @@ -34,3 +34,5 @@ ignore_remote_changes: reason: spec:output_only - field: schema_name reason: spec:output_only + - field: source.sdp_source.pipeline_state + reason: spec:output_only diff --git a/bundle/direct/dresources/configs/sql_warehouses.generated.yml b/bundle/direct/dresources/configs/sql_warehouses.generated.yml new file mode 100644 index 00000000000..e770f766b9c --- /dev/null +++ b/bundle/direct/dresources/configs/sql_warehouses.generated.yml @@ -0,0 +1,5 @@ +# Generated, do not edit. + +ignore_remote_changes: + - field: customer_version_label + reason: spec:output_only diff --git a/bundle/internal/validation/generated/enum_fields.go b/bundle/internal/validation/generated/enum_fields.go index e811c87414b..e67e605689d 100644 --- a/bundle/internal/validation/generated/enum_fields.go +++ b/bundle/internal/validation/generated/enum_fields.go @@ -68,6 +68,7 @@ var EnumFields = map[string][]string{ "resources.features.*.function.function_type": {"APPROX_COUNT_DISTINCT", "APPROX_PERCENTILE", "AVG", "COUNT", "FIRST", "FUNCTION_TYPE_UNSPECIFIED", "LAST", "MAX", "MIN", "STDDEV_POP", "STDDEV_SAMP", "SUM", "VAR_POP", "VAR_SAMP"}, "resources.features.*.source.request_source.flat_schema.fields[*].data_type": {"BINARY", "BOOLEAN", "DATE", "DECIMAL", "DOUBLE", "FLOAT", "INTEGER", "LONG", "SHORT", "STRING", "TIMESTAMP"}, + "resources.features.*.source.sdp_source.pipeline_state": {"ERROR", "PENDING", "RESOLVED"}, "resources.genie_spaces.*.permissions[*].level": {"CAN_ATTACH_TO", "CAN_BIND", "CAN_CREATE", "CAN_CREATE_APP", "CAN_EDIT", "CAN_EDIT_METADATA", "CAN_MANAGE", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_MANAGE_RUN", "CAN_MANAGE_STAGING_VERSIONS", "CAN_MONITOR", "CAN_MONITOR_ONLY", "CAN_QUERY", "CAN_READ", "CAN_RESTART", "CAN_RUN", "CAN_USE", "CAN_VIEW", "CAN_VIEW_METADATA", "IS_OWNER"}, diff --git a/bundle/schema/jsonschema.json b/bundle/schema/jsonschema.json index 90a5e8af5d4..76a608b5275 100644 --- a/bundle/schema/jsonschema.json +++ b/bundle/schema/jsonschema.json @@ -3279,6 +3279,12 @@ "description": "Settings that control the deployment lifecycle of the resource, such as preventing it from being destroyed.", "$ref": "#/$defs/github.com/databricks/cli/bundle/config/resources.LifecycleWithStarted" }, + "max_dbu": { + "description": "[Private Preview] Maximum compute capacity of a real-time SQL warehouse, in DBU per hour.\nMust be a multiple of 3 between 3 and 960. The effective maximum cannot be\nlower than the warehouse's minimum capacity.", + "$ref": "#/$defs/int", + "x-databricks-launch-stage": "PRIVATE_PREVIEW", + "doNotSuggest": true + }, "max_num_clusters": { "description": "Maximum number of clusters that the autoscaler will create to handle\nconcurrent queries.\n\nSupported values:\n- Must be \u003e= min_num_clusters\n- Must be \u003c= 40.\n\nDefaults to min_clusters if unset.", "$ref": "#/$defs/int", @@ -3299,6 +3305,12 @@ "$ref": "#/$defs/slice/github.com/databricks/cli/bundle/config/resources.SqlWarehousePermission", "markdownDescription": "A Sequence of permissions to apply to this resource, where each item grants a permission `level` to a single `user_name`, `group_name`, or `service_principal_name`. A principal cannot be set in both a resource's `permissions` and the top-level `permissions` mapping.\n\nSee [permissions](https://docs.databricks.com/dev-tools/bundles/settings.html#permissions) and [link](https://docs.databricks.com/dev-tools/bundles/permissions.html)." }, + "reyden_cluster_size_multiplier": { + "description": "[Private Preview] Cluster size multiplier for Reyden warehouses. Scales the effective compute\ncapacity relative to the nominal virtual cluster size. Only meaningful when\nthe warehouse is of Reyden type.\n\nAccepted range: 1.0 to the workspace-configured maximum (decimals\nallowed). The default maximum is 100.0.\n- Values below 1.0 are normalized to 1.0.\n- Values above the configured maximum are capped at that maximum.\n- When absent, the warehouse is treated as having a multiplier of 1.0.", + "$ref": "#/$defs/float64", + "x-databricks-launch-stage": "PRIVATE_PREVIEW", + "doNotSuggest": true + }, "spot_instance_policy": { "description": "Configurations whether the endpoint should use spot instances.", "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/sql.SpotInstancePolicy", @@ -5725,7 +5737,7 @@ "x-databricks-launch-stage": "GA" }, "custom": { - "description": "Custom OpenAI-compatible provider configuration with bearer-token\nauthentication.", + "description": "Custom OpenAI-compatible provider configuration.", "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/catalog.ModelProviderServiceConfigCustomProviderConfig", "x-databricks-launch-stage": "GA" }, @@ -6010,7 +6022,7 @@ "oneOf": [ { "type": "object", - "description": "Custom OpenAI-compatible provider configuration with bearer-token\nauthentication.", + "description": "Custom OpenAI-compatible provider configuration.", "properties": { "direct": { "description": "Endpoint and authentication configuration for the custom provider.", @@ -11685,10 +11697,9 @@ "type": "object", "properties": { "attachment": { - "description": "[Private Preview] Optional: The configuration for the files attached to the dashboard snapshot notification. It\ncontrols the underlying data of dashboard widgets that is attached as data files and which\npages are included in the attached PDF. If not specified, the notification is sent with only\nthe default snapshot.", + "description": "[Beta] Optional: The configuration for the files attached to the dashboard snapshot notification. It\ncontrols the underlying data of dashboard widgets that is attached as data files and which\npages are included in the attached PDF. If not specified, the notification is sent with only\nthe default snapshot.", "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/jobs.SubscriptionAttachment", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" }, "custom_subject": { "description": "Optional: Allows users to specify a custom subject line on the email sent\nto subscribers.", @@ -11721,16 +11732,14 @@ "description": "The configuration for the files attached to the dashboard snapshot notification sent to\nsubscribers. `data_attachment` describes the underlying data of dashboard widgets attached\nas data files, and `page_attachment` describes which dashboard pages are included in the\nattached PDF.", "properties": { "data_attachment": { - "description": "[Private Preview] Optional: The configuration for the underlying data of dashboard widgets attached as tabular\ndata files. If not specified, no tabular data is attached.", + "description": "[Beta] Optional: The configuration for the underlying data of dashboard widgets attached as tabular\ndata files. If not specified, no tabular data is attached.", "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/jobs.SubscriptionAttachmentDataAttachment", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" }, "page_attachment": { - "description": "[Private Preview] Optional: The selection of dashboard pages included in the attached PDF. If not specified,\nonly the first page is included.", + "description": "[Beta] Optional: The selection of dashboard pages included in the attached PDF. If not specified,\nonly the first page is included.", "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/jobs.SubscriptionAttachmentPageAttachment", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" } }, "additionalProperties": false @@ -11748,22 +11757,19 @@ "description": "The configuration for the tabular data files attached to the dashboard snapshot. `widgets`\nlists the widgets whose data is exported and `file_format` is the output format; each\nreferenced widget's data is exported as a separate file.", "properties": { "file_format": { - "description": "[Private Preview] The file format of the exported tabular data files.", + "description": "[Beta] The file format of the exported tabular data files.", "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/jobs.SubscriptionAttachmentDataAttachmentFileFormat", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" }, "include_applied_filters": { - "description": "[Private Preview] When true, the dashboard's applied filter values are included alongside the tabular data as\nan additional attached file.", + "description": "[Beta] When true, the dashboard's applied filter values are included alongside the tabular data as\nan additional attached file.", "$ref": "#/$defs/bool", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" }, "widgets": { - "description": "[Private Preview] The widgets whose underlying data is exported. At least one widget must be referenced.", + "description": "[Beta] The widgets whose underlying data is exported. At least one widget must be referenced.", "$ref": "#/$defs/slice/github.com/databricks/databricks-sdk-go/service/jobs.SubscriptionAttachmentDataAttachmentWidget", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" } }, "additionalProperties": false, @@ -11789,9 +11795,9 @@ "FILE_FORMAT_EXCEL" ], "enumDescriptions": [ - "[Private Preview]", - "[Private Preview]", - "[Private Preview]" + "[Beta]", + "[Beta]", + "[Beta]" ] }, { @@ -11807,16 +11813,14 @@ "description": "Identifies a dashboard widget whose data to attach. `page_id` and `widget_id` are\nthe URL identifiers from the dashboard's serialized definition returned by the Get dashboard\nAPI (`serialized_dashboard`), and together they uniquely identify the widget.", "properties": { "page_id": { - "description": "[Private Preview] The URL identifier of the page containing the widget. This is the value of the `name`\nfield of the corresponding entry under `pages` in the serialized dashboard definition.\nFormat: alphanumeric, hyphen, and underscore characters allowed, max length 63.", + "description": "[Beta] The URL identifier of the page containing the widget. This is the value of the `name`\nfield of the corresponding entry under `pages` in the serialized dashboard definition.\nFormat: alphanumeric, hyphen, and underscore characters allowed, max length 63.", "$ref": "#/$defs/string", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" }, "widget_id": { - "description": "[Private Preview] The URL identifier of the widget. This is the value of the `name` field of the\ncorresponding widget under that page's `layout` in the serialized dashboard definition.\nFormat: alphanumeric, hyphen, and underscore characters allowed, max length 63.", + "description": "[Beta] The URL identifier of the widget. This is the value of the `name` field of the\ncorresponding widget under that page's `layout` in the serialized dashboard definition.\nFormat: alphanumeric, hyphen, and underscore characters allowed, max length 63.", "$ref": "#/$defs/string", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" } }, "additionalProperties": false, @@ -11838,16 +11842,14 @@ "description": "The selection of dashboard pages included in the attached PDF. `page_selection` chooses\nbetween all pages, the first page only, or an explicit subset listed in `page_ids`.", "properties": { "page_ids": { - "description": "[Private Preview] The identifiers of the pages to include in the PDF. Only specified when `page_selection` is\n`PAGE_SELECTION_SUBSET`. Each value is the identifier of a page in the dashboard's\nserialized definition returned by the Get dashboard API (`serialized_dashboard`): the value\nof the `name` field of an entry under `pages`.\nFormat: alphanumeric, hyphen, and underscore characters allowed, max length 63.", + "description": "[Beta] The identifiers of the pages to include in the PDF. Only specified when `page_selection` is\n`PAGE_SELECTION_SUBSET`. Each value is the identifier of a page in the dashboard's\nserialized definition returned by the Get dashboard API (`serialized_dashboard`): the value\nof the `name` field of an entry under `pages`.\nFormat: alphanumeric, hyphen, and underscore characters allowed, max length 63.", "$ref": "#/$defs/slice/string", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" }, "page_selection": { - "description": "[Private Preview] How the attached pages are selected. When set to `PAGE_SELECTION_SUBSET`, the pages to\ninclude are listed in `page_ids`.", + "description": "[Beta] How the attached pages are selected. When set to `PAGE_SELECTION_SUBSET`, the pages to\ninclude are listed in `page_ids`.", "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/jobs.SubscriptionAttachmentPageAttachmentPageSelection", - "x-databricks-launch-stage": "PRIVATE_PREVIEW", - "doNotSuggest": true + "x-databricks-launch-stage": "PUBLIC_BETA" } }, "additionalProperties": false @@ -11868,8 +11870,8 @@ "PAGE_SELECTION_SUBSET" ], "enumDescriptions": [ - "[Private Preview]", - "[Private Preview]" + "[Beta]", + "[Beta]" ] }, { @@ -12824,6 +12826,12 @@ "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/ml.RequestSource", "x-databricks-launch-stage": "GA" }, + "sdp_source": { + "description": "[Private Preview] A Spark Declarative Pipeline data source.", + "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/ml.SdpSource", + "x-databricks-launch-stage": "PRIVATE_PREVIEW", + "doNotSuggest": true + }, "stream_source": { "description": "A Stream data source.", "$ref": "#/$defs/github.com/databricks/databricks-sdk-go/service/ml.StreamSource", @@ -13682,6 +13690,79 @@ } ] }, + "ml.SdpSource": { + "oneOf": [ + { + "type": "object", + "description": "A Spark Declarative Pipeline (SDP) data source.", + "properties": { + "dataframe_schema": { + "description": "[Private Preview] Schema of the resulting dataframe after transformations, in Spark StructType JSON format (from df.schema.json()).\nRequired if transformation_sql is specified.\nExample: {\"type\":\"struct\",\"fields\":[{\"name\":\"col_a\",\"type\":\"integer\",\"nullable\":true,\"metadata\":{}},{\"name\":\"col_c\",\"type\":\"integer\",\"nullable\":true,\"metadata\":{}}]}", + "$ref": "#/$defs/string", + "x-databricks-launch-stage": "PRIVATE_PREVIEW", + "doNotSuggest": true + }, + "filter_condition": { + "description": "[Private Preview] Single WHERE clause to filter the pipeline's output table before applying transformations. Will be row-wise evaluated, so should only include conditionals and projections.", + "$ref": "#/$defs/string", + "x-databricks-launch-stage": "PRIVATE_PREVIEW", + "doNotSuggest": true + }, + "input_tables": { + "description": "[Private Preview] The full three-part names of the tables the pipeline reads from. Required to materialize this\nfeature with a TableTrigger, which starts a run when these tables update. Each table must exist\nwhen the feature is created. After the pipeline runs, Feature Store checks that it reads from\nthese tables, and the run fails if it doesn't. At most 100 tables.", + "$ref": "#/$defs/slice/string", + "x-databricks-launch-stage": "PRIVATE_PREVIEW", + "doNotSuggest": true + }, + "output_table": { + "description": "[Private Preview] The full three-part (catalog, schema, table) name of the pipeline's output table to read from.\nThe table doesn't need to exist until the pipeline first runs. Required.", + "$ref": "#/$defs/string", + "x-databricks-launch-stage": "PRIVATE_PREVIEW", + "doNotSuggest": true + }, + "pipeline_id": { + "description": "[Private Preview] The ID of the Spark Declarative Pipeline that Feature Store runs to produce this source, before\neach materialization or backfill. The pipeline must already exist when the feature is created,\nand you need CAN_RUN, CAN_MANAGE, or owner permission on it. It doesn't need to have run yet.\nDeleting the feature doesn't delete the pipeline. If the pipeline is deleted, this field reads\nback empty. Required.", + "$ref": "#/$defs/string", + "x-databricks-launch-stage": "PRIVATE_PREVIEW", + "doNotSuggest": true + }, + "transformation_sql": { + "description": "[Private Preview] A single SQL SELECT expression applied after filter_condition.\nShould contain all the columns needed (eg. \"SELECT *, col_a + col_b AS col_c FROM x.y.z WHERE col_a \u003e 0\" would have `transformation_sql` \"*, col_a + col_b AS col_c\")\nIf transformation_sql is not provided, all columns of the pipeline's output table are present in the DataSource dataframe.", + "$ref": "#/$defs/string", + "x-databricks-launch-stage": "PRIVATE_PREVIEW", + "doNotSuggest": true + } + }, + "additionalProperties": false + }, + { + "type": "string", + "pattern": "\\$\\{(var(\\._*\\p{L}+([-_]*[\\p{L}\\p{N}]+)*(\\[[0-9]+\\])*)+)\\}" + } + ] + }, + "ml.SdpSourceState": { + "oneOf": [ + { + "type": "string", + "description": "State of the SDP source relative to pipeline execution and output table validation.", + "enum": [ + "PENDING", + "RESOLVED", + "ERROR" + ], + "enumDescriptions": [ + "[Private Preview]", + "[Private Preview]", + "[Private Preview]" + ] + }, + { + "type": "string", + "pattern": "\\$\\{(var(\\._*\\p{L}+([-_]*[\\p{L}\\p{N}]+)*(\\[[0-9]+\\])*)+)\\}" + } + ] + }, "ml.SlidingWindow": { "oneOf": [ { diff --git a/cmd/workspace/ai-gateway/ai-gateway.go b/cmd/workspace/ai-gateway/ai-gateway.go index 3fe1d3da088..8dd93402cc2 100644 --- a/cmd/workspace/ai-gateway/ai-gateway.go +++ b/cmd/workspace/ai-gateway/ai-gateway.go @@ -459,8 +459,17 @@ Create a skill. Creates a skill in a Unity Catalog schema and provisions its managed bundle storage. Specify its name in skill_id. The request contains an optional - comment but no bundle bytes. Upload bundle files through the Files API, then - call FinalizeSkill. + comment but no bundle bytes. + + To publish a skill: + + 1. Call CreateSkill to create the skill in Unity Catalog. 2. Upload every + bundle file, including SKILL.md at the bundle root, with the Files API: PUT + /api/2.0/fs/files/Skills/{catalog}/{schema}/{skill}/{file_path}. The + CreateSkill caller owns the skill and can upload. Other callers need + READ_VOLUME and WRITE_VOLUME on the skill. 3. Call FinalizeSkill to + validate SKILL.md and set the skill's bundle_name and description from + its frontmatter name and description. You must be the owner of the parent schema or have CREATE_VOLUME and USE_SCHEMA on it, plus USE_CATALOG on the parent catalog. @@ -879,7 +888,8 @@ func newFinalizeSkill() *cobra.Command { Finalize a skill. - Finalizes a skill after its bundle is uploaded. This method reads SKILL.md + Finalizes a skill after its bundle files are uploaded or changed through the + Files API. Call it again after every bundle change. This method reads SKILL.md through the Files API using the caller's authorization. Its YAML frontmatter must contain an agentskills.io-compliant name and a nonblank description within the configured UTF-8 byte limit. On success, it replaces bundle_name @@ -1938,8 +1948,15 @@ Update a skill. an etag to make the update conditional on the skill not having changed since it was read. Bundle files, grants, tags, and ownership are unchanged. - You must be the owner of the skill or have MANAGE on it, plus USE_CATALOG - on the parent catalog and USE_SCHEMA on the parent schema. + To update the bundle, add, overwrite (overwrite=true), or delete files under + /Skills/{catalog}/{schema}/{skill}/ with the Files API, then call + FinalizeSkill. Changing bundle files requires being the owner of the skill or + having READ_VOLUME and WRITE_VOLUME on it; MANAGE does not grant access + to bundle files. + + To call UpdateSkill, you must be the owner of the skill or have MANAGE on + it, plus USE_CATALOG on the parent catalog and USE_SCHEMA on the parent + schema. Arguments: NAME: Resource name of the skill. Format: skills/{catalog}.{schema}.{skill}. diff --git a/cmd/workspace/bundle-deployments/bundle-deployments.go b/cmd/workspace/bundle-deployments/bundle-deployments.go index ddf35aeaf97..47c856736ef 100644 --- a/cmd/workspace/bundle-deployments/bundle-deployments.go +++ b/cmd/workspace/bundle-deployments/bundle-deployments.go @@ -1134,8 +1134,8 @@ func newUpdateOperation() *cobra.Command { the operation has been modified since. On success the server increments sequence_id; updates to state, resource_id, and dashboard_metadata are mirrored onto the corresponding deployment-level resource. Listing state in - update_mask with no value clears it, which removes the resource, so a delete - that is retried until it succeeds must clear state. The parent version must + update_mask with no value clears it, which removes the tracked resource, so + a delete or unbind that succeeds must clear state. The parent version must be in progress, and after the update is applied a succeeded operation cannot carry an error_message. See the state and resource_id fields for the rest. diff --git a/cmd/workspace/shares/shares.go b/cmd/workspace/shares/shares.go index 40baf4d5b66..4f3869ae300 100644 --- a/cmd/workspace/shares/shares.go +++ b/cmd/workspace/shares/shares.go @@ -70,7 +70,6 @@ func newCreate() *cobra.Command { cmd.Flags().Var(&createJson, "json", `either inline JSON string or @path/to/file.json with request body`) cmd.Flags().StringVar(&createReq.Comment, "comment", createReq.Comment, `User-provided free-form text description.`) - // TODO: complex arg: replication_config cmd.Flags().StringVar(&createReq.StorageRoot, "storage-root", createReq.StorageRoot, `Storage root URL for the share.`) cmd.Use = "create NAME" diff --git a/cmd/workspace/warehouses/warehouses.go b/cmd/workspace/warehouses/warehouses.go index 2958b509e0c..8995fadc49b 100644 --- a/cmd/workspace/warehouses/warehouses.go +++ b/cmd/workspace/warehouses/warehouses.go @@ -94,9 +94,11 @@ func newCreate() *cobra.Command { cmd.Flags().BoolVar(&createReq.EnablePhoton, "enable-photon", createReq.EnablePhoton, `Configures whether the warehouse should use Photon optimized clusters.`) cmd.Flags().BoolVar(&createReq.EnableServerlessCompute, "enable-serverless-compute", createReq.EnableServerlessCompute, `Configures whether the warehouse should use serverless compute.`) cmd.Flags().StringVar(&createReq.InstanceProfileArn, "instance-profile-arn", createReq.InstanceProfileArn, `Deprecated.`) + cmd.Flags().IntVar(&createReq.MaxDbu, "max-dbu", createReq.MaxDbu, `Maximum compute capacity of a real-time SQL warehouse, in DBU per hour.`) cmd.Flags().IntVar(&createReq.MaxNumClusters, "max-num-clusters", createReq.MaxNumClusters, `Maximum number of clusters that the autoscaler will create to handle concurrent queries.`) cmd.Flags().IntVar(&createReq.MinNumClusters, "min-num-clusters", createReq.MinNumClusters, `Minimum number of available clusters that will be maintained for this SQL warehouse.`) cmd.Flags().StringVar(&createReq.Name, "name", createReq.Name, `Logical name for the cluster.`) + cmd.Flags().Float64Var(&createReq.ReydenClusterSizeMultiplier, "reyden-cluster-size-multiplier", createReq.ReydenClusterSizeMultiplier, `Cluster size multiplier for Reyden warehouses.`) cmd.Flags().Var(&createReq.SpotInstancePolicy, "spot-instance-policy", `Configurations whether the endpoint should use spot instances. Supported values: [COST_OPTIMIZED, POLICY_UNSPECIFIED, RELIABILITY_OPTIMIZED]`) cmd.Flags().IntVar(&createReq.StatementTimeout, "statement-timeout", createReq.StatementTimeout, `Warehouse statement timeout in seconds.`) // TODO: complex arg: tags @@ -449,9 +451,11 @@ func newEdit() *cobra.Command { cmd.Flags().BoolVar(&editReq.EnablePhoton, "enable-photon", editReq.EnablePhoton, `Configures whether the warehouse should use Photon optimized clusters.`) cmd.Flags().BoolVar(&editReq.EnableServerlessCompute, "enable-serverless-compute", editReq.EnableServerlessCompute, `Configures whether the warehouse should use serverless compute.`) cmd.Flags().StringVar(&editReq.InstanceProfileArn, "instance-profile-arn", editReq.InstanceProfileArn, `Deprecated.`) + cmd.Flags().IntVar(&editReq.MaxDbu, "max-dbu", editReq.MaxDbu, `Maximum compute capacity of a real-time SQL warehouse, in DBU per hour.`) cmd.Flags().IntVar(&editReq.MaxNumClusters, "max-num-clusters", editReq.MaxNumClusters, `Maximum number of clusters that the autoscaler will create to handle concurrent queries.`) cmd.Flags().IntVar(&editReq.MinNumClusters, "min-num-clusters", editReq.MinNumClusters, `Minimum number of available clusters that will be maintained for this SQL warehouse.`) cmd.Flags().StringVar(&editReq.Name, "name", editReq.Name, `Logical name for the cluster.`) + cmd.Flags().Float64Var(&editReq.ReydenClusterSizeMultiplier, "reyden-cluster-size-multiplier", editReq.ReydenClusterSizeMultiplier, `Cluster size multiplier for Reyden warehouses.`) cmd.Flags().Var(&editReq.SpotInstancePolicy, "spot-instance-policy", `Configurations whether the endpoint should use spot instances. Supported values: [COST_OPTIMIZED, POLICY_UNSPECIFIED, RELIABILITY_OPTIMIZED]`) cmd.Flags().IntVar(&editReq.StatementTimeout, "statement-timeout", editReq.StatementTimeout, `Warehouse statement timeout in seconds.`) // TODO: complex arg: tags diff --git a/go.mod b/go.mod index c2b8ef27741..9c4a2775369 100644 --- a/go.mod +++ b/go.mod @@ -13,7 +13,7 @@ require ( github.com/charmbracelet/huh v1.0.0 // MIT github.com/charmbracelet/lipgloss v1.1.0 // MIT github.com/charmbracelet/x/ansi v0.11.8 // MIT - github.com/databricks/databricks-sdk-go v0.189.0 // Apache-2.0 + github.com/databricks/databricks-sdk-go v0.190.0 // Apache-2.0 github.com/databricks/sdk-go/auth v0.1.0 // Apache-2.0 github.com/databricks/sdk-go/core v0.1.0 // Apache-2.0 github.com/databricks/sdk-go/files v0.1.0 // Apache-2.0 diff --git a/go.sum b/go.sum index bba164c3a69..d1de9e67b81 100644 --- a/go.sum +++ b/go.sum @@ -59,8 +59,8 @@ github.com/creack/pty v1.1.24 h1:bJrF4RRfyJnbTJqzRLHzcGaZK1NeM5kTC9jGgovnR1s= github.com/creack/pty v1.1.24/go.mod h1:08sCNb52WyoAwi2QDyzUCTgcvVFhUzewun7wtTfvcwE= github.com/danieljoos/wincred v1.2.3 h1:v7dZC2x32Ut3nEfRH+vhoZGvN72+dQ/snVXo/vMFLdQ= github.com/danieljoos/wincred v1.2.3/go.mod h1:6qqX0WNrS4RzPZ1tnroDzq9kY3fu1KwE7MRLQK4X0bs= -github.com/databricks/databricks-sdk-go v0.189.0 h1:ayn+g67dGFl/LTxsg3bbrHZAslX9CJrdttQQjK0gDg4= -github.com/databricks/databricks-sdk-go v0.189.0/go.mod h1:udBw6nxkaqa0fgPQ7ZW1kGFkltQncYx+qFJeVFmxtz8= +github.com/databricks/databricks-sdk-go v0.190.0 h1:kU2Bq0RGZInRYk35JbdX/w5CFDAUoNyikT5Ze1YdURg= +github.com/databricks/databricks-sdk-go v0.190.0/go.mod h1:udBw6nxkaqa0fgPQ7ZW1kGFkltQncYx+qFJeVFmxtz8= github.com/databricks/sdk-go/auth v0.1.0 h1:NZyXdvKE1fX72/4EQbiX3AhfUI63wyPJhFCdZp35L2A= github.com/databricks/sdk-go/auth v0.1.0/go.mod h1:eNJdFzBAfu1y32a3wPcTnjaUkjhS9VfyCNblKFPIPEo= github.com/databricks/sdk-go/core v0.1.0 h1:ZolzW/KVIJy1xXQ0zTsu/QYsEB4vQAEApbbq4D3RGHc= diff --git a/python/codegen/uv.lock b/python/codegen/uv.lock index 681ead3c4f0..4835268ec96 100644 --- a/python/codegen/uv.lock +++ b/python/codegen/uv.lock @@ -20,52 +20,52 @@ dev = [{ name = "pytest" }] [[package]] name = "colorama" version = "0.4.6" -source = { registry = "https://pypi.org/simple" } -sdist = { url = "https://files.pythonhosted.org/packages/d8/53/6f443c9a4a8358a93a6792e2acffb9d9d5cb0a5cfd8802644b7b1c9a02e4/colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44", size = 27697, upload-time = "2022-10-25T02:36:22.414Z" } +source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } +sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/d8/53/6f443c9a4a8358a93a6792e2acffb9d9d5cb0a5cfd8802644b7b1c9a02e4/colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44", size = 27697, upload-time = "2022-10-25T02:36:22.414Z" } wheels = [ - { url = "https://files.pythonhosted.org/packages/d1/d6/3965ed04c63042e047cb6a3e6ed1a63a35087b6a609aa3a15ed8ac56c221/colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6", size = 25335, upload-time = "2022-10-25T02:36:20.889Z" }, + { url = "https://pypi-proxy.cloud.databricks.com/packages/d1/d6/3965ed04c63042e047cb6a3e6ed1a63a35087b6a609aa3a15ed8ac56c221/colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6", size = 25335, upload-time = "2022-10-25T02:36:20.889Z" }, ] [[package]] name = "iniconfig" version = "2.1.0" -source = { registry = "https://pypi.org/simple" } -sdist = { url = "https://files.pythonhosted.org/packages/f2/97/ebf4da567aa6827c909642694d71c9fcf53e5b504f2d96afea02718862f3/iniconfig-2.1.0.tar.gz", hash = "sha256:3abbd2e30b36733fee78f9c7f7308f2d0050e88f0087fd25c2645f63c773e1c7", size = 4793, upload-time = "2025-03-19T20:09:59.721Z" } +source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } +sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/f2/97/ebf4da567aa6827c909642694d71c9fcf53e5b504f2d96afea02718862f3/iniconfig-2.1.0.tar.gz", hash = "sha256:3abbd2e30b36733fee78f9c7f7308f2d0050e88f0087fd25c2645f63c773e1c7", size = 4793, upload-time = "2025-03-19T20:09:59.721Z" } wheels = [ - { url = "https://files.pythonhosted.org/packages/2c/e1/e6716421ea10d38022b952c159d5161ca1193197fb744506875fbb87ea7b/iniconfig-2.1.0-py3-none-any.whl", hash = "sha256:9deba5723312380e77435581c6bf4935c94cbfab9b1ed33ef8d238ea168eb760", size = 6050, upload-time = "2025-03-19T20:10:01.071Z" }, + { url = "https://pypi-proxy.cloud.databricks.com/packages/2c/e1/e6716421ea10d38022b952c159d5161ca1193197fb744506875fbb87ea7b/iniconfig-2.1.0-py3-none-any.whl", hash = "sha256:9deba5723312380e77435581c6bf4935c94cbfab9b1ed33ef8d238ea168eb760", size = 6050, upload-time = "2025-03-19T20:10:01.071Z" }, ] [[package]] name = "packaging" version = "24.2" -source = { registry = "https://pypi.org/simple" } -sdist = { url = "https://files.pythonhosted.org/packages/d0/63/68dbb6eb2de9cb10ee4c9c14a0148804425e13c4fb20d61cce69f53106da/packaging-24.2.tar.gz", hash = "sha256:c228a6dc5e932d346bc5739379109d49e8853dd8223571c7c5b55260edc0b97f", size = 163950, upload-time = "2024-11-08T09:47:47.202Z" } +source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } +sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/d0/63/68dbb6eb2de9cb10ee4c9c14a0148804425e13c4fb20d61cce69f53106da/packaging-24.2.tar.gz", hash = "sha256:c228a6dc5e932d346bc5739379109d49e8853dd8223571c7c5b55260edc0b97f", size = 163950, upload-time = "2024-11-08T09:47:47.202Z" } wheels = [ - { url = "https://files.pythonhosted.org/packages/88/ef/eb23f262cca3c0c4eb7ab1933c3b1f03d021f2c48f54763065b6f0e321be/packaging-24.2-py3-none-any.whl", hash = "sha256:09abb1bccd265c01f4a3aa3f7a7db064b36514d2cba19a2f694fe6150451a759", size = 65451, upload-time = "2024-11-08T09:47:44.722Z" }, + { url = "https://pypi-proxy.cloud.databricks.com/packages/88/ef/eb23f262cca3c0c4eb7ab1933c3b1f03d021f2c48f54763065b6f0e321be/packaging-24.2-py3-none-any.whl", hash = "sha256:09abb1bccd265c01f4a3aa3f7a7db064b36514d2cba19a2f694fe6150451a759", size = 65451, upload-time = "2024-11-08T09:47:44.722Z" }, ] [[package]] name = "pluggy" version = "1.5.0" -source = { registry = "https://pypi.org/simple" } -sdist = { url = "https://files.pythonhosted.org/packages/96/2d/02d4312c973c6050a18b314a5ad0b3210edb65a906f868e31c111dede4a6/pluggy-1.5.0.tar.gz", hash = "sha256:2cffa88e94fdc978c4c574f15f9e59b7f4201d439195c3715ca9e2486f1d0cf1", size = 67955, upload-time = "2024-04-20T21:34:42.531Z" } +source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } +sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/96/2d/02d4312c973c6050a18b314a5ad0b3210edb65a906f868e31c111dede4a6/pluggy-1.5.0.tar.gz", hash = "sha256:2cffa88e94fdc978c4c574f15f9e59b7f4201d439195c3715ca9e2486f1d0cf1", size = 67955, upload-time = "2024-04-20T21:34:42.531Z" } wheels = [ - { url = "https://files.pythonhosted.org/packages/88/5f/e351af9a41f866ac3f1fac4ca0613908d9a41741cfcf2228f4ad853b697d/pluggy-1.5.0-py3-none-any.whl", hash = "sha256:44e1ad92c8ca002de6377e165f3e0f1be63266ab4d554740532335b9d75ea669", size = 20556, upload-time = "2024-04-20T21:34:40.434Z" }, + { url = "https://pypi-proxy.cloud.databricks.com/packages/88/5f/e351af9a41f866ac3f1fac4ca0613908d9a41741cfcf2228f4ad853b697d/pluggy-1.5.0-py3-none-any.whl", hash = "sha256:44e1ad92c8ca002de6377e165f3e0f1be63266ab4d554740532335b9d75ea669", size = 20556, upload-time = "2024-04-20T21:34:40.434Z" }, ] [[package]] name = "pygments" version = "2.20.0" -source = { registry = "https://pypi.org/simple" } -sdist = { url = "https://files.pythonhosted.org/packages/c3/b2/bc9c9196916376152d655522fdcebac55e66de6603a76a02bca1b6414f6c/pygments-2.20.0.tar.gz", hash = "sha256:6757cd03768053ff99f3039c1a36d6c0aa0b263438fcab17520b30a303a82b5f", size = 4955991, upload-time = "2026-03-29T13:29:33.898Z" } +source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } +sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/c3/b2/bc9c9196916376152d655522fdcebac55e66de6603a76a02bca1b6414f6c/pygments-2.20.0.tar.gz", hash = "sha256:6757cd03768053ff99f3039c1a36d6c0aa0b263438fcab17520b30a303a82b5f", size = 4955991, upload-time = "2026-03-29T13:29:33.898Z" } wheels = [ - { url = "https://files.pythonhosted.org/packages/f4/7e/a72dd26f3b0f4f2bf1dd8923c85f7ceb43172af56d63c7383eb62b332364/pygments-2.20.0-py3-none-any.whl", hash = "sha256:81a9e26dd42fd28a23a2d169d86d7ac03b46e2f8b59ed4698fb4785f946d0176", size = 1231151, upload-time = "2026-03-29T13:29:30.038Z" }, + { url = "https://pypi-proxy.cloud.databricks.com/packages/f4/7e/a72dd26f3b0f4f2bf1dd8923c85f7ceb43172af56d63c7383eb62b332364/pygments-2.20.0-py3-none-any.whl", hash = "sha256:81a9e26dd42fd28a23a2d169d86d7ac03b46e2f8b59ed4698fb4785f946d0176", size = 1231151, upload-time = "2026-03-29T13:29:30.038Z" }, ] [[package]] name = "pytest" version = "9.0.3" -source = { registry = "https://pypi.org/simple" } +source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } dependencies = [ { name = "colorama", marker = "sys_platform == 'win32'" }, { name = "iniconfig" }, @@ -73,7 +73,7 @@ dependencies = [ { name = "pluggy" }, { name = "pygments" }, ] -sdist = { url = "https://files.pythonhosted.org/packages/7d/0d/549bd94f1a0a402dc8cf64563a117c0f3765662e2e668477624baeec44d5/pytest-9.0.3.tar.gz", hash = "sha256:b86ada508af81d19edeb213c681b1d48246c1a91d304c6c81a427674c17eb91c", size = 1572165, upload-time = "2026-04-07T17:16:18.027Z" } +sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/7d/0d/549bd94f1a0a402dc8cf64563a117c0f3765662e2e668477624baeec44d5/pytest-9.0.3.tar.gz", hash = "sha256:b86ada508af81d19edeb213c681b1d48246c1a91d304c6c81a427674c17eb91c", size = 1572165, upload-time = "2026-04-07T17:16:18.027Z" } wheels = [ - { url = "https://files.pythonhosted.org/packages/d4/24/a372aaf5c9b7208e7112038812994107bc65a84cd00e0354a88c2c77a617/pytest-9.0.3-py3-none-any.whl", hash = "sha256:2c5efc453d45394fdd706ade797c0a81091eccd1d6e4bccfcd476e2b8e0ab5d9", size = 375249, upload-time = "2026-04-07T17:16:16.13Z" }, + { url = "https://pypi-proxy.cloud.databricks.com/packages/d4/24/a372aaf5c9b7208e7112038812994107bc65a84cd00e0354a88c2c77a617/pytest-9.0.3-py3-none-any.whl", hash = "sha256:2c5efc453d45394fdd706ade797c0a81091eccd1d6e4bccfcd476e2b8e0ab5d9", size = 375249, upload-time = "2026-04-07T17:16:16.13Z" }, ] diff --git a/python/databricks/bundles/alerts/__init__.py b/python/databricks/bundles/alerts/__init__.py index 3d8987ba271..a64c273c35e 100644 --- a/python/databricks/bundles/alerts/__init__.py +++ b/python/databricks/bundles/alerts/__init__.py @@ -1,131 +1,78 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Aggregation", - "AggregationParam", - "Alert", - "AlertDict", - "AlertEvaluationState", - "AlertEvaluationStateParam", - "AlertParam", - "AlertStatementParameter", - "AlertStatementParameterDict", - "AlertStatementParameterParam", - "AlertV2Evaluation", - "AlertV2EvaluationDict", - "AlertV2EvaluationParam", - "AlertV2Notification", - "AlertV2NotificationDict", - "AlertV2NotificationParam", - "AlertV2Operand", - "AlertV2OperandColumn", - "AlertV2OperandColumnDict", - "AlertV2OperandColumnParam", - "AlertV2OperandDict", - "AlertV2OperandParam", - "AlertV2OperandValue", - "AlertV2OperandValueDict", - "AlertV2OperandValueParam", - "AlertV2RunAs", - "AlertV2RunAsDict", - "AlertV2RunAsParam", - "AlertV2Subscription", - "AlertV2SubscriptionDict", - "AlertV2SubscriptionParam", - "ComparisonOperator", - "ComparisonOperatorParam", - "CronSchedule", - "CronScheduleDict", - "CronScheduleParam", - "CustomTemplateFormat", - "CustomTemplateFormatParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "Permission", - "PermissionDict", - "PermissionLevel", - "PermissionLevelParam", - "PermissionParam", - "SchedulePauseStatus", - "SchedulePauseStatusParam", + 'Aggregation', + 'AggregationParam', + 'Alert', + 'AlertDict', + 'AlertEvaluationState', + 'AlertEvaluationStateParam', + 'AlertParam', + 'AlertStatementParameter', + 'AlertStatementParameterDict', + 'AlertStatementParameterParam', + 'AlertV2Evaluation', + 'AlertV2EvaluationDict', + 'AlertV2EvaluationParam', + 'AlertV2Notification', + 'AlertV2NotificationDict', + 'AlertV2NotificationParam', + 'AlertV2Operand', + 'AlertV2OperandColumn', + 'AlertV2OperandColumnDict', + 'AlertV2OperandColumnParam', + 'AlertV2OperandDict', + 'AlertV2OperandParam', + 'AlertV2OperandValue', + 'AlertV2OperandValueDict', + 'AlertV2OperandValueParam', + 'AlertV2RunAs', + 'AlertV2RunAsDict', + 'AlertV2RunAsParam', + 'AlertV2Subscription', + 'AlertV2SubscriptionDict', + 'AlertV2SubscriptionParam', + 'ComparisonOperator', + 'ComparisonOperatorParam', + 'CronSchedule', + 'CronScheduleDict', + 'CronScheduleParam', + 'CustomTemplateFormat', + 'CustomTemplateFormatParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'Permission', + 'PermissionDict', + 'PermissionLevel', + 'PermissionLevelParam', + 'PermissionParam', + 'SchedulePauseStatus', + 'SchedulePauseStatusParam', ] -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.alert_evaluation_state import ( - AlertEvaluationState, - AlertEvaluationStateParam, -) -from databricks.bundles.alerts._models.alert_statement_parameter import ( - AlertStatementParameter, - AlertStatementParameterDict, - AlertStatementParameterParam, -) -from databricks.bundles.alerts._models.alert_v2_evaluation import ( - AlertV2Evaluation, - AlertV2EvaluationDict, - AlertV2EvaluationParam, -) -from databricks.bundles.alerts._models.alert_v2_notification import ( - AlertV2Notification, - AlertV2NotificationDict, - AlertV2NotificationParam, -) -from databricks.bundles.alerts._models.alert_v2_operand import ( - AlertV2Operand, - AlertV2OperandDict, - AlertV2OperandParam, -) -from databricks.bundles.alerts._models.alert_v2_operand_column import ( - AlertV2OperandColumn, - AlertV2OperandColumnDict, - AlertV2OperandColumnParam, -) -from databricks.bundles.alerts._models.alert_v2_operand_value import ( - AlertV2OperandValue, - AlertV2OperandValueDict, - AlertV2OperandValueParam, -) -from databricks.bundles.alerts._models.alert_v2_run_as import ( - AlertV2RunAs, - AlertV2RunAsDict, - AlertV2RunAsParam, -) -from databricks.bundles.alerts._models.alert_v2_subscription import ( - AlertV2Subscription, - AlertV2SubscriptionDict, - AlertV2SubscriptionParam, -) -from databricks.bundles.alerts._models.comparison_operator import ( - ComparisonOperator, - ComparisonOperatorParam, -) -from databricks.bundles.alerts._models.cron_schedule import ( - CronSchedule, - CronScheduleDict, - CronScheduleParam, -) -from databricks.bundles.alerts._models.custom_template_format import ( - CustomTemplateFormat, - CustomTemplateFormatParam, -) -from databricks.bundles.alerts._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.alerts._models.permission import ( - Permission, - PermissionDict, - PermissionParam, -) -from databricks.bundles.alerts._models.permission_level import ( - PermissionLevel, - PermissionLevelParam, -) -from databricks.bundles.alerts._models.schedule_pause_status import ( - SchedulePauseStatus, - SchedulePauseStatusParam, -) +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + diff --git a/python/databricks/bundles/alerts/_models/aggregation.py b/python/databricks/bundles/alerts/_models/aggregation.py index ff9efddf6e9..809aae0b3c1 100644 --- a/python/databricks/bundles/alerts/_models/aggregation.py +++ b/python/databricks/bundles/alerts/_models/aggregation.py @@ -1,7 +1,35 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam class Aggregation(Enum): @@ -14,8 +42,4 @@ class Aggregation(Enum): MAX = "MAX" STDDEV = "STDDEV" - -AggregationParam = ( - Literal["SUM", "COUNT", "COUNT_DISTINCT", "AVG", "MEDIAN", "MIN", "MAX", "STDDEV"] - | Aggregation -) +AggregationParam = Literal["SUM", "COUNT", "COUNT_DISTINCT", "AVG", "MEDIAN", "MIN", "MAX", "STDDEV"] | Aggregation diff --git a/python/databricks/bundles/alerts/_models/alert.py b/python/databricks/bundles/alerts/_models/alert.py index 2a66b1d169c..6232280649e 100644 --- a/python/databricks/bundles/alerts/_models/alert.py +++ b/python/databricks/bundles/alerts/_models/alert.py @@ -1,43 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.alerts._models.alert_statement_parameter import ( - AlertStatementParameter, - AlertStatementParameterParam, -) -from databricks.bundles.alerts._models.alert_v2_evaluation import ( - AlertV2Evaluation, - AlertV2EvaluationParam, -) -from databricks.bundles.alerts._models.alert_v2_run_as import ( - AlertV2RunAs, - AlertV2RunAsParam, -) -from databricks.bundles.alerts._models.cron_schedule import ( - CronSchedule, - CronScheduleParam, -) -from databricks.bundles.alerts._models.custom_template_format import ( - CustomTemplateFormat, - CustomTemplateFormatParam, -) -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionParam +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Alert(Resource): """""" @@ -137,11 +135,12 @@ class Alert(Resource): """ @classmethod - def from_dict(cls, value: "AlertDict") -> "Self": + def from_dict(cls, value: 'AlertDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertDict': + return _transform_to_json_value(self) # type:ignore + class AlertDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_evaluation_state.py b/python/databricks/bundles/alerts/_models/alert_evaluation_state.py index abee35a5a4a..4cf7627c088 100644 --- a/python/databricks/bundles/alerts/_models/alert_evaluation_state.py +++ b/python/databricks/bundles/alerts/_models/alert_evaluation_state.py @@ -1,7 +1,35 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam class AlertEvaluationState(Enum): @@ -12,13 +40,9 @@ class AlertEvaluationState(Enum): OK - alert is not triggered ERROR - alert evaluation failed """ - UNKNOWN = "UNKNOWN" TRIGGERED = "TRIGGERED" OK = "OK" ERROR = "ERROR" - -AlertEvaluationStateParam = ( - Literal["UNKNOWN", "TRIGGERED", "OK", "ERROR"] | AlertEvaluationState -) +AlertEvaluationStateParam = Literal["UNKNOWN", "TRIGGERED", "OK", "ERROR"] | AlertEvaluationState diff --git a/python/databricks/bundles/alerts/_models/alert_statement_parameter.py b/python/databricks/bundles/alerts/_models/alert_statement_parameter.py index def206d148d..b2d33f0eb0e 100644 --- a/python/databricks/bundles/alerts/_models/alert_statement_parameter.py +++ b/python/databricks/bundles/alerts/_models/alert_statement_parameter.py @@ -1,25 +1,49 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertStatementParameter: """ :meta private: [EXPERIMENTAL] - + A named parameter bound to the alert query. Only flat, named scalar parameters are supported; complex types such as ARRAY, MAP, and STRUCT are not. """ - name: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -44,11 +68,12 @@ class AlertStatementParameter: """ @classmethod - def from_dict(cls, value: "AlertStatementParameterDict") -> "Self": + def from_dict(cls, value: 'AlertStatementParameterDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertStatementParameterDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertStatementParameterDict': + return _transform_to_json_value(self) # type:ignore + class AlertStatementParameterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_evaluation.py b/python/databricks/bundles/alerts/_models/alert_v2_evaluation.py index 4d599872ccb..88c6f05a946 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_evaluation.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_evaluation.py @@ -1,36 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.alerts._models.alert_evaluation_state import ( - AlertEvaluationState, - AlertEvaluationStateParam, -) -from databricks.bundles.alerts._models.alert_v2_notification import ( - AlertV2Notification, - AlertV2NotificationParam, -) -from databricks.bundles.alerts._models.alert_v2_operand import ( - AlertV2Operand, - AlertV2OperandParam, -) -from databricks.bundles.alerts._models.alert_v2_operand_column import ( - AlertV2OperandColumn, - AlertV2OperandColumnParam, -) -from databricks.bundles.alerts._models.comparison_operator import ( - ComparisonOperator, - ComparisonOperatorParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertV2Evaluation: """""" @@ -61,11 +66,12 @@ class AlertV2Evaluation: """ @classmethod - def from_dict(cls, value: "AlertV2EvaluationDict") -> "Self": + def from_dict(cls, value: 'AlertV2EvaluationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertV2EvaluationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertV2EvaluationDict': + return _transform_to_json_value(self) # type:ignore + class AlertV2EvaluationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_notification.py b/python/databricks/bundles/alerts/_models/alert_v2_notification.py index 81c892f1d90..33507c80c81 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_notification.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_notification.py @@ -1,20 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.alerts._models.alert_v2_subscription import ( - AlertV2Subscription, - AlertV2SubscriptionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertV2Notification: """""" @@ -34,11 +55,12 @@ class AlertV2Notification: subscriptions: VariableOrList[AlertV2Subscription] = field(default_factory=list) @classmethod - def from_dict(cls, value: "AlertV2NotificationDict") -> "Self": + def from_dict(cls, value: 'AlertV2NotificationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertV2NotificationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertV2NotificationDict': + return _transform_to_json_value(self) # type:ignore + class AlertV2NotificationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_operand.py b/python/databricks/bundles/alerts/_models/alert_v2_operand.py index 5e39c938860..fa391b35c2d 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_operand.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_operand.py @@ -1,24 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.alerts._models.alert_v2_operand_column import ( - AlertV2OperandColumn, - AlertV2OperandColumnParam, -) -from databricks.bundles.alerts._models.alert_v2_operand_value import ( - AlertV2OperandValue, - AlertV2OperandValueParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertV2Operand: """""" @@ -28,11 +45,12 @@ class AlertV2Operand: value: VariableOrOptional[AlertV2OperandValue] = None @classmethod - def from_dict(cls, value: "AlertV2OperandDict") -> "Self": + def from_dict(cls, value: 'AlertV2OperandDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertV2OperandDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertV2OperandDict': + return _transform_to_json_value(self) # type:ignore + class AlertV2OperandDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_operand_column.py b/python/databricks/bundles/alerts/_models/alert_v2_operand_column.py index da2f90107e8..3a6f83d2614 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_operand_column.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_operand_column.py @@ -1,17 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertV2OperandColumn: """""" @@ -26,11 +50,12 @@ class AlertV2OperandColumn: display: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: "AlertV2OperandColumnDict") -> "Self": + def from_dict(cls, value: 'AlertV2OperandColumnDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertV2OperandColumnDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertV2OperandColumnDict': + return _transform_to_json_value(self) # type:ignore + class AlertV2OperandColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_operand_value.py b/python/databricks/bundles/alerts/_models/alert_v2_operand_value.py index 5abc7f386ee..8fda36d762e 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_operand_value.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_operand_value.py @@ -1,16 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertV2OperandValue: """""" @@ -22,11 +47,12 @@ class AlertV2OperandValue: string_value: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: "AlertV2OperandValueDict") -> "Self": + def from_dict(cls, value: 'AlertV2OperandValueDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertV2OperandValueDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertV2OperandValueDict': + return _transform_to_json_value(self) # type:ignore + class AlertV2OperandValueDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_run_as.py b/python/databricks/bundles/alerts/_models/alert_v2_run_as.py index 164318184ab..07fafdc17d3 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_run_as.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_run_as.py @@ -1,16 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertV2RunAs: """""" @@ -26,11 +51,12 @@ class AlertV2RunAs: """ @classmethod - def from_dict(cls, value: "AlertV2RunAsDict") -> "Self": + def from_dict(cls, value: 'AlertV2RunAsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertV2RunAsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertV2RunAsDict': + return _transform_to_json_value(self) # type:ignore + class AlertV2RunAsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_subscription.py b/python/databricks/bundles/alerts/_models/alert_v2_subscription.py index 8d379a1ced7..d87b69d0198 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_subscription.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_subscription.py @@ -1,16 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertV2Subscription: """""" @@ -20,11 +45,12 @@ class AlertV2Subscription: user_email: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: "AlertV2SubscriptionDict") -> "Self": + def from_dict(cls, value: 'AlertV2SubscriptionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertV2SubscriptionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertV2SubscriptionDict': + return _transform_to_json_value(self) # type:ignore + class AlertV2SubscriptionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/comparison_operator.py b/python/databricks/bundles/alerts/_models/comparison_operator.py index 95dc77260a0..268914118ca 100644 --- a/python/databricks/bundles/alerts/_models/comparison_operator.py +++ b/python/databricks/bundles/alerts/_models/comparison_operator.py @@ -1,7 +1,35 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam class ComparisonOperator(Enum): @@ -14,17 +42,4 @@ class ComparisonOperator(Enum): IS_NULL = "IS_NULL" IS_NOT_NULL = "IS_NOT_NULL" - -ComparisonOperatorParam = ( - Literal[ - "LESS_THAN", - "GREATER_THAN", - "EQUAL", - "NOT_EQUAL", - "GREATER_THAN_OR_EQUAL", - "LESS_THAN_OR_EQUAL", - "IS_NULL", - "IS_NOT_NULL", - ] - | ComparisonOperator -) +ComparisonOperatorParam = Literal["LESS_THAN", "GREATER_THAN", "EQUAL", "NOT_EQUAL", "GREATER_THAN_OR_EQUAL", "LESS_THAN_OR_EQUAL", "IS_NULL", "IS_NOT_NULL"] | ComparisonOperator diff --git a/python/databricks/bundles/alerts/_models/cron_schedule.py b/python/databricks/bundles/alerts/_models/cron_schedule.py index 987a9a1056b..474e5e841f1 100644 --- a/python/databricks/bundles/alerts/_models/cron_schedule.py +++ b/python/databricks/bundles/alerts/_models/cron_schedule.py @@ -1,20 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.alerts._models.schedule_pause_status import ( - SchedulePauseStatus, - SchedulePauseStatusParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class CronSchedule: """""" @@ -38,11 +59,12 @@ class CronSchedule: """ @classmethod - def from_dict(cls, value: "CronScheduleDict") -> "Self": + def from_dict(cls, value: 'CronScheduleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CronScheduleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CronScheduleDict': + return _transform_to_json_value(self) # type:ignore + class CronScheduleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/custom_template_format.py b/python/databricks/bundles/alerts/_models/custom_template_format.py index fbddaecfc0e..d22bbab2303 100644 --- a/python/databricks/bundles/alerts/_models/custom_template_format.py +++ b/python/databricks/bundles/alerts/_models/custom_template_format.py @@ -1,18 +1,44 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam class CustomTemplateFormat(Enum): """ :meta private: [EXPERIMENTAL] - + Rendering format for the `custom_summary` and `custom_description` templates. """ - HTML = "HTML" MARKDOWN = "MARKDOWN" - CustomTemplateFormatParam = Literal["HTML", "MARKDOWN"] | CustomTemplateFormat diff --git a/python/databricks/bundles/alerts/_models/lifecycle.py b/python/databricks/bundles/alerts/_models/lifecycle.py index 697776a198e..ef6a9fe2352 100644 --- a/python/databricks/bundles/alerts/_models/lifecycle.py +++ b/python/databricks/bundles/alerts/_models/lifecycle.py @@ -1,16 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +46,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/permission.py b/python/databricks/bundles/alerts/_models/permission.py index e4091c6d061..9d4b5e813d5 100644 --- a/python/databricks/bundles/alerts/_models/permission.py +++ b/python/databricks/bundles/alerts/_models/permission.py @@ -1,20 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.alerts._models.permission_level import ( - PermissionLevel, - PermissionLevelParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Permission: """""" @@ -40,11 +61,12 @@ class Permission: """ @classmethod - def from_dict(cls, value: "PermissionDict") -> "Self": + def from_dict(cls, value: 'PermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PermissionDict': + return _transform_to_json_value(self) # type:ignore + class PermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/permission_level.py b/python/databricks/bundles/alerts/_models/permission_level.py index c6111911b29..c709c9ac644 100644 --- a/python/databricks/bundles/alerts/_models/permission_level.py +++ b/python/databricks/bundles/alerts/_models/permission_level.py @@ -1,14 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam + +from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam class PermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" @@ -30,29 +57,4 @@ class PermissionLevel(Enum): CAN_MONITOR_ONLY = "CAN_MONITOR_ONLY" CAN_CREATE_APP = "CAN_CREATE_APP" - -PermissionLevelParam = ( - Literal[ - "CAN_MANAGE", - "CAN_RESTART", - "CAN_ATTACH_TO", - "IS_OWNER", - "CAN_MANAGE_RUN", - "CAN_VIEW", - "CAN_READ", - "CAN_RUN", - "CAN_EDIT", - "CAN_USE", - "CAN_MANAGE_STAGING_VERSIONS", - "CAN_MANAGE_PRODUCTION_VERSIONS", - "CAN_EDIT_METADATA", - "CAN_VIEW_METADATA", - "CAN_BIND", - "CAN_QUERY", - "CAN_MONITOR", - "CAN_CREATE", - "CAN_MONITOR_ONLY", - "CAN_CREATE_APP", - ] - | PermissionLevel -) +PermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW", "CAN_READ", "CAN_RUN", "CAN_EDIT", "CAN_USE", "CAN_MANAGE_STAGING_VERSIONS", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_EDIT_METADATA", "CAN_VIEW_METADATA", "CAN_BIND", "CAN_QUERY", "CAN_MONITOR", "CAN_CREATE", "CAN_MONITOR_ONLY", "CAN_CREATE_APP"] | PermissionLevel diff --git a/python/databricks/bundles/alerts/_models/schedule_pause_status.py b/python/databricks/bundles/alerts/_models/schedule_pause_status.py index ab05367f8c1..a99d86c8174 100644 --- a/python/databricks/bundles/alerts/_models/schedule_pause_status.py +++ b/python/databricks/bundles/alerts/_models/schedule_pause_status.py @@ -1,12 +1,39 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam +from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam +from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam +from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam +from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam +from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam +from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam +from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam + +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam + +from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam + +from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam + +from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam class SchedulePauseStatus(Enum): UNPAUSED = "UNPAUSED" PAUSED = "PAUSED" - SchedulePauseStatusParam = Literal["UNPAUSED", "PAUSED"] | SchedulePauseStatus diff --git a/python/databricks/bundles/apps/__init__.py b/python/databricks/bundles/apps/__init__.py index 7b92e6d61bb..442928f9265 100644 --- a/python/databricks/bundles/apps/__init__.py +++ b/python/databricks/bundles/apps/__init__.py @@ -1,237 +1,138 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "App", - "AppConfig", - "AppConfigDict", - "AppConfigParam", - "AppDict", - "AppEnvVar", - "AppEnvVarDict", - "AppEnvVarParam", - "AppParam", - "AppPermission", - "AppPermissionDict", - "AppPermissionLevel", - "AppPermissionLevelParam", - "AppPermissionParam", - "AppResource", - "AppResourceApp", - "AppResourceAppAppPermission", - "AppResourceAppAppPermissionParam", - "AppResourceAppDict", - "AppResourceAppParam", - "AppResourceDatabase", - "AppResourceDatabaseDatabasePermission", - "AppResourceDatabaseDatabasePermissionParam", - "AppResourceDatabaseDict", - "AppResourceDatabaseParam", - "AppResourceDict", - "AppResourceExperiment", - "AppResourceExperimentDict", - "AppResourceExperimentExperimentPermission", - "AppResourceExperimentExperimentPermissionParam", - "AppResourceExperimentParam", - "AppResourceGenieSpace", - "AppResourceGenieSpaceDict", - "AppResourceGenieSpaceGenieSpacePermission", - "AppResourceGenieSpaceGenieSpacePermissionParam", - "AppResourceGenieSpaceParam", - "AppResourceJob", - "AppResourceJobDict", - "AppResourceJobJobPermission", - "AppResourceJobJobPermissionParam", - "AppResourceJobParam", - "AppResourceParam", - "AppResourcePostgres", - "AppResourcePostgresDict", - "AppResourcePostgresParam", - "AppResourcePostgresPostgresPermission", - "AppResourcePostgresPostgresPermissionParam", - "AppResourceSecret", - "AppResourceSecretDict", - "AppResourceSecretParam", - "AppResourceSecretSecretPermission", - "AppResourceSecretSecretPermissionParam", - "AppResourceServingEndpoint", - "AppResourceServingEndpointDict", - "AppResourceServingEndpointParam", - "AppResourceServingEndpointServingEndpointPermission", - "AppResourceServingEndpointServingEndpointPermissionParam", - "AppResourceSqlWarehouse", - "AppResourceSqlWarehouseDict", - "AppResourceSqlWarehouseParam", - "AppResourceSqlWarehouseSqlWarehousePermission", - "AppResourceSqlWarehouseSqlWarehousePermissionParam", - "AppResourceUcSecurable", - "AppResourceUcSecurableDict", - "AppResourceUcSecurableParam", - "AppResourceUcSecurableUcSecurablePermission", - "AppResourceUcSecurableUcSecurablePermissionParam", - "AppResourceUcSecurableUcSecurableType", - "AppResourceUcSecurableUcSecurableTypeParam", - "ComputeSize", - "ComputeSizeParam", - "GitRepository", - "GitRepositoryDict", - "GitRepositoryParam", - "GitSource", - "GitSourceDict", - "GitSourceParam", - "LifecycleWithStarted", - "LifecycleWithStartedDict", - "LifecycleWithStartedParam", - "TelemetryExportDestination", - "TelemetryExportDestinationDict", - "TelemetryExportDestinationParam", - "UnityCatalog", - "UnityCatalogDict", - "UnityCatalogParam", + 'App', + 'AppConfig', + 'AppConfigDict', + 'AppConfigParam', + 'AppDict', + 'AppEnvVar', + 'AppEnvVarDict', + 'AppEnvVarParam', + 'AppParam', + 'AppPermission', + 'AppPermissionDict', + 'AppPermissionLevel', + 'AppPermissionLevelParam', + 'AppPermissionParam', + 'AppResource', + 'AppResourceApp', + 'AppResourceAppAppPermission', + 'AppResourceAppAppPermissionParam', + 'AppResourceAppDict', + 'AppResourceAppParam', + 'AppResourceDatabase', + 'AppResourceDatabaseDatabasePermission', + 'AppResourceDatabaseDatabasePermissionParam', + 'AppResourceDatabaseDict', + 'AppResourceDatabaseParam', + 'AppResourceDict', + 'AppResourceExperiment', + 'AppResourceExperimentDict', + 'AppResourceExperimentExperimentPermission', + 'AppResourceExperimentExperimentPermissionParam', + 'AppResourceExperimentParam', + 'AppResourceGenieSpace', + 'AppResourceGenieSpaceDict', + 'AppResourceGenieSpaceGenieSpacePermission', + 'AppResourceGenieSpaceGenieSpacePermissionParam', + 'AppResourceGenieSpaceParam', + 'AppResourceJob', + 'AppResourceJobDict', + 'AppResourceJobJobPermission', + 'AppResourceJobJobPermissionParam', + 'AppResourceJobParam', + 'AppResourceParam', + 'AppResourcePostgres', + 'AppResourcePostgresDict', + 'AppResourcePostgresParam', + 'AppResourcePostgresPostgresPermission', + 'AppResourcePostgresPostgresPermissionParam', + 'AppResourceSecret', + 'AppResourceSecretDict', + 'AppResourceSecretParam', + 'AppResourceSecretSecretPermission', + 'AppResourceSecretSecretPermissionParam', + 'AppResourceServingEndpoint', + 'AppResourceServingEndpointDict', + 'AppResourceServingEndpointParam', + 'AppResourceServingEndpointServingEndpointPermission', + 'AppResourceServingEndpointServingEndpointPermissionParam', + 'AppResourceSqlWarehouse', + 'AppResourceSqlWarehouseDict', + 'AppResourceSqlWarehouseParam', + 'AppResourceSqlWarehouseSqlWarehousePermission', + 'AppResourceSqlWarehouseSqlWarehousePermissionParam', + 'AppResourceUcSecurable', + 'AppResourceUcSecurableDict', + 'AppResourceUcSecurableParam', + 'AppResourceUcSecurableUcSecurablePermission', + 'AppResourceUcSecurableUcSecurablePermissionParam', + 'AppResourceUcSecurableUcSecurableType', + 'AppResourceUcSecurableUcSecurableTypeParam', + 'ComputeSize', + 'ComputeSizeParam', + 'GitRepository', + 'GitRepositoryDict', + 'GitRepositoryParam', + 'GitSource', + 'GitSourceDict', + 'GitSourceParam', + 'LifecycleWithStarted', + 'LifecycleWithStartedDict', + 'LifecycleWithStartedParam', + 'TelemetryExportDestination', + 'TelemetryExportDestinationDict', + 'TelemetryExportDestinationParam', + 'UnityCatalog', + 'UnityCatalogDict', + 'UnityCatalogParam', ] +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import ( - AppConfig, - AppConfigDict, - AppConfigParam, -) -from databricks.bundles.apps._models.app_env_var import ( - AppEnvVar, - AppEnvVarDict, - AppEnvVarParam, -) -from databricks.bundles.apps._models.app_permission import ( - AppPermission, - AppPermissionDict, - AppPermissionParam, -) -from databricks.bundles.apps._models.app_permission_level import ( - AppPermissionLevel, - AppPermissionLevelParam, -) -from databricks.bundles.apps._models.app_resource import ( - AppResource, - AppResourceDict, - AppResourceParam, -) -from databricks.bundles.apps._models.app_resource_app import ( - AppResourceApp, - AppResourceAppDict, - AppResourceAppParam, -) -from databricks.bundles.apps._models.app_resource_app_app_permission import ( - AppResourceAppAppPermission, - AppResourceAppAppPermissionParam, -) -from databricks.bundles.apps._models.app_resource_database import ( - AppResourceDatabase, - AppResourceDatabaseDict, - AppResourceDatabaseParam, -) -from databricks.bundles.apps._models.app_resource_database_database_permission import ( - AppResourceDatabaseDatabasePermission, - AppResourceDatabaseDatabasePermissionParam, -) -from databricks.bundles.apps._models.app_resource_experiment import ( - AppResourceExperiment, - AppResourceExperimentDict, - AppResourceExperimentParam, -) -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import ( - AppResourceExperimentExperimentPermission, - AppResourceExperimentExperimentPermissionParam, -) -from databricks.bundles.apps._models.app_resource_genie_space import ( - AppResourceGenieSpace, - AppResourceGenieSpaceDict, - AppResourceGenieSpaceParam, -) -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import ( - AppResourceGenieSpaceGenieSpacePermission, - AppResourceGenieSpaceGenieSpacePermissionParam, -) -from databricks.bundles.apps._models.app_resource_job import ( - AppResourceJob, - AppResourceJobDict, - AppResourceJobParam, -) -from databricks.bundles.apps._models.app_resource_job_job_permission import ( - AppResourceJobJobPermission, - AppResourceJobJobPermissionParam, -) -from databricks.bundles.apps._models.app_resource_postgres import ( - AppResourcePostgres, - AppResourcePostgresDict, - AppResourcePostgresParam, -) -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import ( - AppResourcePostgresPostgresPermission, - AppResourcePostgresPostgresPermissionParam, -) -from databricks.bundles.apps._models.app_resource_secret import ( - AppResourceSecret, - AppResourceSecretDict, - AppResourceSecretParam, -) -from databricks.bundles.apps._models.app_resource_secret_secret_permission import ( - AppResourceSecretSecretPermission, - AppResourceSecretSecretPermissionParam, -) -from databricks.bundles.apps._models.app_resource_serving_endpoint import ( - AppResourceServingEndpoint, - AppResourceServingEndpointDict, - AppResourceServingEndpointParam, -) -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import ( - AppResourceServingEndpointServingEndpointPermission, - AppResourceServingEndpointServingEndpointPermissionParam, -) -from databricks.bundles.apps._models.app_resource_sql_warehouse import ( - AppResourceSqlWarehouse, - AppResourceSqlWarehouseDict, - AppResourceSqlWarehouseParam, -) -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import ( - AppResourceSqlWarehouseSqlWarehousePermission, - AppResourceSqlWarehouseSqlWarehousePermissionParam, -) -from databricks.bundles.apps._models.app_resource_uc_securable import ( - AppResourceUcSecurable, - AppResourceUcSecurableDict, - AppResourceUcSecurableParam, -) -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import ( - AppResourceUcSecurableUcSecurablePermission, - AppResourceUcSecurableUcSecurablePermissionParam, -) -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import ( - AppResourceUcSecurableUcSecurableType, - AppResourceUcSecurableUcSecurableTypeParam, -) +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam -from databricks.bundles.apps._models.git_repository import ( - GitRepository, - GitRepositoryDict, - GitRepositoryParam, -) -from databricks.bundles.apps._models.git_source import ( - GitSource, - GitSourceDict, - GitSourceParam, -) -from databricks.bundles.apps._models.lifecycle_with_started import ( - LifecycleWithStarted, - LifecycleWithStartedDict, - LifecycleWithStartedParam, -) -from databricks.bundles.apps._models.telemetry_export_destination import ( - TelemetryExportDestination, - TelemetryExportDestinationDict, - TelemetryExportDestinationParam, -) -from databricks.bundles.apps._models.unity_catalog import ( - UnityCatalog, - UnityCatalogDict, - UnityCatalogParam, -) + diff --git a/python/databricks/bundles/apps/_models/app.py b/python/databricks/bundles/apps/_models/app.py index 533172ae6b2..3826caa1379 100644 --- a/python/databricks/bundles/apps/_models/app.py +++ b/python/databricks/bundles/apps/_models/app.py @@ -1,44 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.apps._models.app_config import ( - AppConfig, - AppConfigParam, -) -from databricks.bundles.apps._models.app_permission import ( - AppPermission, - AppPermissionParam, -) -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceParam -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam -from databricks.bundles.apps._models.git_repository import ( - GitRepository, - GitRepositoryParam, -) -from databricks.bundles.apps._models.git_source import GitSource, GitSourceParam -from databricks.bundles.apps._models.lifecycle_with_started import ( - LifecycleWithStarted, - LifecycleWithStartedParam, -) -from databricks.bundles.apps._models.telemetry_export_destination import ( - TelemetryExportDestination, - TelemetryExportDestinationParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class App(Resource): """""" @@ -117,9 +136,7 @@ class App(Resource): [Private Preview] Name of the space this app belongs to. """ - telemetry_export_destinations: VariableOrList[TelemetryExportDestination] = field( - default_factory=list - ) + telemetry_export_destinations: VariableOrList[TelemetryExportDestination] = field(default_factory=list) usage_policy_id: VariableOrOptional[str] = None """ @@ -132,11 +149,12 @@ class App(Resource): """ @classmethod - def from_dict(cls, value: "AppDict") -> "Self": + def from_dict(cls, value: 'AppDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppDict': + return _transform_to_json_value(self) # type:ignore + class AppDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_config.py b/python/databricks/bundles/apps/_models/app_config.py index e8a8f5fd70e..e689c834e26 100644 --- a/python/databricks/bundles/apps/_models/app_config.py +++ b/python/databricks/bundles/apps/_models/app_config.py @@ -1,17 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarParam +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppConfig: """""" @@ -21,11 +67,12 @@ class AppConfig: env: VariableOrList[AppEnvVar] = field(default_factory=list) @classmethod - def from_dict(cls, value: "AppConfigDict") -> "Self": + def from_dict(cls, value: 'AppConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppConfigDict': + return _transform_to_json_value(self) # type:ignore + class AppConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_env_var.py b/python/databricks/bundles/apps/_models/app_env_var.py index a0afec4223a..f02879f73b9 100644 --- a/python/databricks/bundles/apps/_models/app_env_var.py +++ b/python/databricks/bundles/apps/_models/app_env_var.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppEnvVar: """""" @@ -22,11 +69,12 @@ class AppEnvVar: value_from: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: "AppEnvVarDict") -> "Self": + def from_dict(cls, value: 'AppEnvVarDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppEnvVarDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppEnvVarDict': + return _transform_to_json_value(self) # type:ignore + class AppEnvVarDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_permission.py b/python/databricks/bundles/apps/_models/app_permission.py index f04638d0df2..7c00cb07c69 100644 --- a/python/databricks/bundles/apps/_models/app_permission.py +++ b/python/databricks/bundles/apps/_models/app_permission.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_permission_level import ( - AppPermissionLevel, - AppPermissionLevelParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppPermission: """""" @@ -40,11 +83,12 @@ class AppPermission: """ @classmethod - def from_dict(cls, value: "AppPermissionDict") -> "Self": + def from_dict(cls, value: 'AppPermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppPermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppPermissionDict': + return _transform_to_json_value(self) # type:ignore + class AppPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_permission_level.py b/python/databricks/bundles/apps/_models/app_permission_level.py index a951c137ea7..00b62909dec 100644 --- a/python/databricks/bundles/apps/_models/app_permission_level.py +++ b/python/databricks/bundles/apps/_models/app_permission_level.py @@ -1,16 +1,64 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppPermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_USE = "CAN_USE" - AppPermissionLevelParam = Literal["CAN_MANAGE", "CAN_USE"] | AppPermissionLevel diff --git a/python/databricks/bundles/apps/_models/app_resource.py b/python/databricks/bundles/apps/_models/app_resource.py index cc96b083c2c..bbb77c9604b 100644 --- a/python/databricks/bundles/apps/_models/app_resource.py +++ b/python/databricks/bundles/apps/_models/app_resource.py @@ -1,56 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.apps._models.app_resource_app import ( - AppResourceApp, - AppResourceAppParam, -) -from databricks.bundles.apps._models.app_resource_database import ( - AppResourceDatabase, - AppResourceDatabaseParam, -) -from databricks.bundles.apps._models.app_resource_experiment import ( - AppResourceExperiment, - AppResourceExperimentParam, -) -from databricks.bundles.apps._models.app_resource_genie_space import ( - AppResourceGenieSpace, - AppResourceGenieSpaceParam, -) -from databricks.bundles.apps._models.app_resource_job import ( - AppResourceJob, - AppResourceJobParam, -) -from databricks.bundles.apps._models.app_resource_postgres import ( - AppResourcePostgres, - AppResourcePostgresParam, -) -from databricks.bundles.apps._models.app_resource_secret import ( - AppResourceSecret, - AppResourceSecretParam, -) -from databricks.bundles.apps._models.app_resource_serving_endpoint import ( - AppResourceServingEndpoint, - AppResourceServingEndpointParam, -) -from databricks.bundles.apps._models.app_resource_sql_warehouse import ( - AppResourceSqlWarehouse, - AppResourceSqlWarehouseParam, -) -from databricks.bundles.apps._models.app_resource_uc_securable import ( - AppResourceUcSecurable, - AppResourceUcSecurableParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResource: """""" @@ -86,11 +93,12 @@ class AppResource: uc_securable: VariableOrOptional[AppResourceUcSecurable] = None @classmethod - def from_dict(cls, value: "AppResourceDict") -> "Self": + def from_dict(cls, value: 'AppResourceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_app.py b/python/databricks/bundles/apps/_models/app_resource_app.py index a7a05689c2e..ef048ba8f0b 100644 --- a/python/databricks/bundles/apps/_models/app_resource_app.py +++ b/python/databricks/bundles/apps/_models/app_resource_app.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_resource_app_app_permission import ( - AppResourceAppAppPermission, - AppResourceAppAppPermissionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourceApp: """""" @@ -24,11 +67,12 @@ class AppResourceApp: permission: VariableOrOptional[AppResourceAppAppPermission] = None @classmethod - def from_dict(cls, value: "AppResourceAppDict") -> "Self": + def from_dict(cls, value: 'AppResourceAppDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceAppDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceAppDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceAppDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_app_app_permission.py b/python/databricks/bundles/apps/_models/app_resource_app_app_permission.py index a58dca414e4..1c91a920547 100644 --- a/python/databricks/bundles/apps/_models/app_resource_app_app_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_app_app_permission.py @@ -1,11 +1,60 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceAppAppPermission(Enum): CAN_USE = "CAN_USE" - AppResourceAppAppPermissionParam = Literal["CAN_USE"] | AppResourceAppAppPermission diff --git a/python/databricks/bundles/apps/_models/app_resource_database.py b/python/databricks/bundles/apps/_models/app_resource_database.py index b29ce973105..c7f963650d0 100644 --- a/python/databricks/bundles/apps/_models/app_resource_database.py +++ b/python/databricks/bundles/apps/_models/app_resource_database.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_resource_database_database_permission import ( - AppResourceDatabaseDatabasePermission, - AppResourceDatabaseDatabasePermissionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourceDatabase: """""" @@ -26,11 +69,12 @@ class AppResourceDatabase: permission: VariableOr[AppResourceDatabaseDatabasePermission] @classmethod - def from_dict(cls, value: "AppResourceDatabaseDict") -> "Self": + def from_dict(cls, value: 'AppResourceDatabaseDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceDatabaseDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceDatabaseDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceDatabaseDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_database_database_permission.py b/python/databricks/bundles/apps/_models/app_resource_database_database_permission.py index 447d015a1c2..ceb3547b2b6 100644 --- a/python/databricks/bundles/apps/_models/app_resource_database_database_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_database_database_permission.py @@ -1,13 +1,60 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceDatabaseDatabasePermission(Enum): CAN_CONNECT_AND_CREATE = "CAN_CONNECT_AND_CREATE" - -AppResourceDatabaseDatabasePermissionParam = ( - Literal["CAN_CONNECT_AND_CREATE"] | AppResourceDatabaseDatabasePermission -) +AppResourceDatabaseDatabasePermissionParam = Literal["CAN_CONNECT_AND_CREATE"] | AppResourceDatabaseDatabasePermission diff --git a/python/databricks/bundles/apps/_models/app_resource_experiment.py b/python/databricks/bundles/apps/_models/app_resource_experiment.py index e69d4be1a17..5c3d8c63446 100644 --- a/python/databricks/bundles/apps/_models/app_resource_experiment.py +++ b/python/databricks/bundles/apps/_models/app_resource_experiment.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import ( - AppResourceExperimentExperimentPermission, - AppResourceExperimentExperimentPermissionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourceExperiment: """""" @@ -24,11 +67,12 @@ class AppResourceExperiment: permission: VariableOr[AppResourceExperimentExperimentPermission] @classmethod - def from_dict(cls, value: "AppResourceExperimentDict") -> "Self": + def from_dict(cls, value: 'AppResourceExperimentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceExperimentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceExperimentDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceExperimentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_experiment_experiment_permission.py b/python/databricks/bundles/apps/_models/app_resource_experiment_experiment_permission.py index 230b8d5c06e..f8e4ab24765 100644 --- a/python/databricks/bundles/apps/_models/app_resource_experiment_experiment_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_experiment_experiment_permission.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceExperimentExperimentPermission(Enum): @@ -9,8 +59,4 @@ class AppResourceExperimentExperimentPermission(Enum): CAN_EDIT = "CAN_EDIT" CAN_READ = "CAN_READ" - -AppResourceExperimentExperimentPermissionParam = ( - Literal["CAN_MANAGE", "CAN_EDIT", "CAN_READ"] - | AppResourceExperimentExperimentPermission -) +AppResourceExperimentExperimentPermissionParam = Literal["CAN_MANAGE", "CAN_EDIT", "CAN_READ"] | AppResourceExperimentExperimentPermission diff --git a/python/databricks/bundles/apps/_models/app_resource_genie_space.py b/python/databricks/bundles/apps/_models/app_resource_genie_space.py index 0cfdaf316f4..cb3f30a767c 100644 --- a/python/databricks/bundles/apps/_models/app_resource_genie_space.py +++ b/python/databricks/bundles/apps/_models/app_resource_genie_space.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import ( - AppResourceGenieSpaceGenieSpacePermission, - AppResourceGenieSpaceGenieSpacePermissionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourceGenieSpace: """""" @@ -26,11 +69,12 @@ class AppResourceGenieSpace: space_id: VariableOr[str] @classmethod - def from_dict(cls, value: "AppResourceGenieSpaceDict") -> "Self": + def from_dict(cls, value: 'AppResourceGenieSpaceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceGenieSpaceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceGenieSpaceDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceGenieSpaceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_genie_space_genie_space_permission.py b/python/databricks/bundles/apps/_models/app_resource_genie_space_genie_space_permission.py index 312897a8cce..591978b1637 100644 --- a/python/databricks/bundles/apps/_models/app_resource_genie_space_genie_space_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_genie_space_genie_space_permission.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceGenieSpaceGenieSpacePermission(Enum): @@ -10,8 +60,4 @@ class AppResourceGenieSpaceGenieSpacePermission(Enum): CAN_RUN = "CAN_RUN" CAN_VIEW = "CAN_VIEW" - -AppResourceGenieSpaceGenieSpacePermissionParam = ( - Literal["CAN_MANAGE", "CAN_EDIT", "CAN_RUN", "CAN_VIEW"] - | AppResourceGenieSpaceGenieSpacePermission -) +AppResourceGenieSpaceGenieSpacePermissionParam = Literal["CAN_MANAGE", "CAN_EDIT", "CAN_RUN", "CAN_VIEW"] | AppResourceGenieSpaceGenieSpacePermission diff --git a/python/databricks/bundles/apps/_models/app_resource_job.py b/python/databricks/bundles/apps/_models/app_resource_job.py index 08ca1be6cfc..83a3c397197 100644 --- a/python/databricks/bundles/apps/_models/app_resource_job.py +++ b/python/databricks/bundles/apps/_models/app_resource_job.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_resource_job_job_permission import ( - AppResourceJobJobPermission, - AppResourceJobJobPermissionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourceJob: """""" @@ -30,11 +73,12 @@ class AppResourceJob: """ @classmethod - def from_dict(cls, value: "AppResourceJobDict") -> "Self": + def from_dict(cls, value: 'AppResourceJobDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceJobDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceJobDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceJobDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_job_job_permission.py b/python/databricks/bundles/apps/_models/app_resource_job_job_permission.py index 29f08d0aab1..14fdea47b31 100644 --- a/python/databricks/bundles/apps/_models/app_resource_job_job_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_job_job_permission.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceJobJobPermission(Enum): @@ -10,8 +60,4 @@ class AppResourceJobJobPermission(Enum): CAN_MANAGE_RUN = "CAN_MANAGE_RUN" CAN_VIEW = "CAN_VIEW" - -AppResourceJobJobPermissionParam = ( - Literal["CAN_MANAGE", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW"] - | AppResourceJobJobPermission -) +AppResourceJobJobPermissionParam = Literal["CAN_MANAGE", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW"] | AppResourceJobJobPermission diff --git a/python/databricks/bundles/apps/_models/app_resource_postgres.py b/python/databricks/bundles/apps/_models/app_resource_postgres.py index 0d3b6e39b17..5c6bf80ed85 100644 --- a/python/databricks/bundles/apps/_models/app_resource_postgres.py +++ b/python/databricks/bundles/apps/_models/app_resource_postgres.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import ( - AppResourcePostgresPostgresPermission, - AppResourcePostgresPostgresPermissionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourcePostgres: """""" @@ -26,11 +69,12 @@ class AppResourcePostgres: permission: VariableOrOptional[AppResourcePostgresPostgresPermission] = None @classmethod - def from_dict(cls, value: "AppResourcePostgresDict") -> "Self": + def from_dict(cls, value: 'AppResourcePostgresDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourcePostgresDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourcePostgresDict': + return _transform_to_json_value(self) # type:ignore + class AppResourcePostgresDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_postgres_postgres_permission.py b/python/databricks/bundles/apps/_models/app_resource_postgres_postgres_permission.py index 85657edfc35..19687d44b81 100644 --- a/python/databricks/bundles/apps/_models/app_resource_postgres_postgres_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_postgres_postgres_permission.py @@ -1,13 +1,60 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourcePostgresPostgresPermission(Enum): CAN_CONNECT_AND_CREATE = "CAN_CONNECT_AND_CREATE" - -AppResourcePostgresPostgresPermissionParam = ( - Literal["CAN_CONNECT_AND_CREATE"] | AppResourcePostgresPostgresPermission -) +AppResourcePostgresPostgresPermissionParam = Literal["CAN_CONNECT_AND_CREATE"] | AppResourcePostgresPostgresPermission diff --git a/python/databricks/bundles/apps/_models/app_resource_secret.py b/python/databricks/bundles/apps/_models/app_resource_secret.py index 3ecca5e2199..c09c69d7fa5 100644 --- a/python/databricks/bundles/apps/_models/app_resource_secret.py +++ b/python/databricks/bundles/apps/_models/app_resource_secret.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_resource_secret_secret_permission import ( - AppResourceSecretSecretPermission, - AppResourceSecretSecretPermissionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourceSecret: """""" @@ -35,11 +78,12 @@ class AppResourceSecret: """ @classmethod - def from_dict(cls, value: "AppResourceSecretDict") -> "Self": + def from_dict(cls, value: 'AppResourceSecretDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceSecretDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceSecretDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceSecretDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_secret_secret_permission.py b/python/databricks/bundles/apps/_models/app_resource_secret_secret_permission.py index c30b6d977ff..76a03686e7f 100644 --- a/python/databricks/bundles/apps/_models/app_resource_secret_secret_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_secret_secret_permission.py @@ -1,19 +1,65 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceSecretSecretPermission(Enum): """ Permission to grant on the secret scope. Supported permissions are: "READ", "WRITE", "MANAGE". """ - READ = "READ" WRITE = "WRITE" MANAGE = "MANAGE" - -AppResourceSecretSecretPermissionParam = ( - Literal["READ", "WRITE", "MANAGE"] | AppResourceSecretSecretPermission -) +AppResourceSecretSecretPermissionParam = Literal["READ", "WRITE", "MANAGE"] | AppResourceSecretSecretPermission diff --git a/python/databricks/bundles/apps/_models/app_resource_serving_endpoint.py b/python/databricks/bundles/apps/_models/app_resource_serving_endpoint.py index ee444cd7692..a8c00664f9c 100644 --- a/python/databricks/bundles/apps/_models/app_resource_serving_endpoint.py +++ b/python/databricks/bundles/apps/_models/app_resource_serving_endpoint.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import ( - AppResourceServingEndpointServingEndpointPermission, - AppResourceServingEndpointServingEndpointPermissionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourceServingEndpoint: """""" @@ -30,11 +73,12 @@ class AppResourceServingEndpoint: """ @classmethod - def from_dict(cls, value: "AppResourceServingEndpointDict") -> "Self": + def from_dict(cls, value: 'AppResourceServingEndpointDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceServingEndpointDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceServingEndpointDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceServingEndpointDict(TypedDict, total=False): @@ -51,6 +95,4 @@ class AppResourceServingEndpointDict(TypedDict, total=False): """ -AppResourceServingEndpointParam = ( - AppResourceServingEndpointDict | AppResourceServingEndpoint -) +AppResourceServingEndpointParam = AppResourceServingEndpointDict | AppResourceServingEndpoint diff --git a/python/databricks/bundles/apps/_models/app_resource_serving_endpoint_serving_endpoint_permission.py b/python/databricks/bundles/apps/_models/app_resource_serving_endpoint_serving_endpoint_permission.py index abfe5e63be0..10320643658 100644 --- a/python/databricks/bundles/apps/_models/app_resource_serving_endpoint_serving_endpoint_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_serving_endpoint_serving_endpoint_permission.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceServingEndpointServingEndpointPermission(Enum): @@ -9,8 +59,4 @@ class AppResourceServingEndpointServingEndpointPermission(Enum): CAN_QUERY = "CAN_QUERY" CAN_VIEW = "CAN_VIEW" - -AppResourceServingEndpointServingEndpointPermissionParam = ( - Literal["CAN_MANAGE", "CAN_QUERY", "CAN_VIEW"] - | AppResourceServingEndpointServingEndpointPermission -) +AppResourceServingEndpointServingEndpointPermissionParam = Literal["CAN_MANAGE", "CAN_QUERY", "CAN_VIEW"] | AppResourceServingEndpointServingEndpointPermission diff --git a/python/databricks/bundles/apps/_models/app_resource_sql_warehouse.py b/python/databricks/bundles/apps/_models/app_resource_sql_warehouse.py index 691528a0450..00d50b41d8a 100644 --- a/python/databricks/bundles/apps/_models/app_resource_sql_warehouse.py +++ b/python/databricks/bundles/apps/_models/app_resource_sql_warehouse.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import ( - AppResourceSqlWarehouseSqlWarehousePermission, - AppResourceSqlWarehouseSqlWarehousePermissionParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourceSqlWarehouse: """""" @@ -30,11 +73,12 @@ class AppResourceSqlWarehouse: """ @classmethod - def from_dict(cls, value: "AppResourceSqlWarehouseDict") -> "Self": + def from_dict(cls, value: 'AppResourceSqlWarehouseDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceSqlWarehouseDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceSqlWarehouseDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceSqlWarehouseDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_sql_warehouse_sql_warehouse_permission.py b/python/databricks/bundles/apps/_models/app_resource_sql_warehouse_sql_warehouse_permission.py index ceb417a6edb..7911bec3f99 100644 --- a/python/databricks/bundles/apps/_models/app_resource_sql_warehouse_sql_warehouse_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_sql_warehouse_sql_warehouse_permission.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceSqlWarehouseSqlWarehousePermission(Enum): @@ -9,8 +59,4 @@ class AppResourceSqlWarehouseSqlWarehousePermission(Enum): CAN_USE = "CAN_USE" IS_OWNER = "IS_OWNER" - -AppResourceSqlWarehouseSqlWarehousePermissionParam = ( - Literal["CAN_MANAGE", "CAN_USE", "IS_OWNER"] - | AppResourceSqlWarehouseSqlWarehousePermission -) +AppResourceSqlWarehouseSqlWarehousePermissionParam = Literal["CAN_MANAGE", "CAN_USE", "IS_OWNER"] | AppResourceSqlWarehouseSqlWarehousePermission diff --git a/python/databricks/bundles/apps/_models/app_resource_uc_securable.py b/python/databricks/bundles/apps/_models/app_resource_uc_securable.py index 196b2396592..ef587a9c3c2 100644 --- a/python/databricks/bundles/apps/_models/app_resource_uc_securable.py +++ b/python/databricks/bundles/apps/_models/app_resource_uc_securable.py @@ -1,24 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import ( - AppResourceUcSecurableUcSecurablePermission, - AppResourceUcSecurableUcSecurablePermissionParam, -) -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import ( - AppResourceUcSecurableUcSecurableType, - AppResourceUcSecurableUcSecurableTypeParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AppResourceUcSecurable: """""" @@ -30,11 +69,12 @@ class AppResourceUcSecurable: securable_type: VariableOr[AppResourceUcSecurableUcSecurableType] @classmethod - def from_dict(cls, value: "AppResourceUcSecurableDict") -> "Self": + def from_dict(cls, value: 'AppResourceUcSecurableDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AppResourceUcSecurableDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AppResourceUcSecurableDict': + return _transform_to_json_value(self) # type:ignore + class AppResourceUcSecurableDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_permission.py b/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_permission.py index bd4ed5005db..b163818ff50 100644 --- a/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_permission.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceUcSecurableUcSecurablePermission(Enum): @@ -12,10 +62,4 @@ class AppResourceUcSecurableUcSecurablePermission(Enum): USE_CONNECTION = "USE_CONNECTION" MODIFY = "MODIFY" - -AppResourceUcSecurableUcSecurablePermissionParam = ( - Literal[ - "READ_VOLUME", "WRITE_VOLUME", "SELECT", "EXECUTE", "USE_CONNECTION", "MODIFY" - ] - | AppResourceUcSecurableUcSecurablePermission -) +AppResourceUcSecurableUcSecurablePermissionParam = Literal["READ_VOLUME", "WRITE_VOLUME", "SELECT", "EXECUTE", "USE_CONNECTION", "MODIFY"] | AppResourceUcSecurableUcSecurablePermission diff --git a/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_type.py b/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_type.py index 46fc5390e24..ed3bd8009fd 100644 --- a/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_type.py +++ b/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_type.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam class AppResourceUcSecurableUcSecurableType(Enum): @@ -11,8 +61,4 @@ class AppResourceUcSecurableUcSecurableType(Enum): CONNECTION = "CONNECTION" MODEL_SERVICE = "MODEL_SERVICE" - -AppResourceUcSecurableUcSecurableTypeParam = ( - Literal["VOLUME", "TABLE", "FUNCTION", "CONNECTION", "MODEL_SERVICE"] - | AppResourceUcSecurableUcSecurableType -) +AppResourceUcSecurableUcSecurableTypeParam = Literal["VOLUME", "TABLE", "FUNCTION", "CONNECTION", "MODEL_SERVICE"] | AppResourceUcSecurableUcSecurableType diff --git a/python/databricks/bundles/apps/_models/compute_size.py b/python/databricks/bundles/apps/_models/compute_size.py index 4881d034935..a8e07dcde5f 100644 --- a/python/databricks/bundles/apps/_models/compute_size.py +++ b/python/databricks/bundles/apps/_models/compute_size.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam class ComputeSize(Enum): @@ -9,5 +59,4 @@ class ComputeSize(Enum): LARGE = "LARGE" XLARGE = "XLARGE" - ComputeSizeParam = Literal["MEDIUM", "LARGE", "XLARGE"] | ComputeSize diff --git a/python/databricks/bundles/apps/_models/git_repository.py b/python/databricks/bundles/apps/_models/git_repository.py index d13ae5b2a56..b8877195bfa 100644 --- a/python/databricks/bundles/apps/_models/git_repository.py +++ b/python/databricks/bundles/apps/_models/git_repository.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GitRepository: """ Git repository configuration specifying the location of the repository. """ - provider: VariableOr[str] """ Git provider. Case insensitive. Supported values: gitHub, gitHubEnterprise, bitbucketCloud, @@ -41,11 +87,12 @@ class GitRepository: """ @classmethod - def from_dict(cls, value: "GitRepositoryDict") -> "Self": + def from_dict(cls, value: 'GitRepositoryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GitRepositoryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GitRepositoryDict': + return _transform_to_json_value(self) # type:ignore + class GitRepositoryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/git_source.py b/python/databricks/bundles/apps/_models/git_source.py index 93bd61f5aa4..cecee53d65a 100644 --- a/python/databricks/bundles/apps/_models/git_source.py +++ b/python/databricks/bundles/apps/_models/git_source.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GitSource: """ Complete git source specification including repository location and reference. """ - branch: VariableOrOptional[str] = None """ Git branch to checkout. @@ -39,11 +85,12 @@ class GitSource: """ @classmethod - def from_dict(cls, value: "GitSourceDict") -> "Self": + def from_dict(cls, value: 'GitSourceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GitSourceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GitSourceDict': + return _transform_to_json_value(self) # type:ignore + class GitSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/lifecycle_with_started.py b/python/databricks/bundles/apps/_models/lifecycle_with_started.py index f9cb9b8d49c..9b04e6ddfbd 100644 --- a/python/databricks/bundles/apps/_models/lifecycle_with_started.py +++ b/python/databricks/bundles/apps/_models/lifecycle_with_started.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LifecycleWithStarted: """""" @@ -26,11 +73,12 @@ class LifecycleWithStarted: """ @classmethod - def from_dict(cls, value: "LifecycleWithStartedDict") -> "Self": + def from_dict(cls, value: 'LifecycleWithStartedDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleWithStartedDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleWithStartedDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleWithStartedDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/telemetry_export_destination.py b/python/databricks/bundles/apps/_models/telemetry_export_destination.py index edf9108c8a1..fc346ec938d 100644 --- a/python/databricks/bundles/apps/_models/telemetry_export_destination.py +++ b/python/databricks/bundles/apps/_models/telemetry_export_destination.py @@ -1,37 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.apps._models.unity_catalog import ( - UnityCatalog, - UnityCatalogParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TelemetryExportDestination: """ A single telemetry export destination with its configuration and status. """ - unity_catalog: VariableOrOptional[UnityCatalog] = None """ Unity Catalog Destinations for OTEL telemetry export. """ @classmethod - def from_dict(cls, value: "TelemetryExportDestinationDict") -> "Self": + def from_dict(cls, value: 'TelemetryExportDestinationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TelemetryExportDestinationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TelemetryExportDestinationDict': + return _transform_to_json_value(self) # type:ignore + class TelemetryExportDestinationDict(TypedDict, total=False): @@ -43,6 +86,4 @@ class TelemetryExportDestinationDict(TypedDict, total=False): """ -TelemetryExportDestinationParam = ( - TelemetryExportDestinationDict | TelemetryExportDestination -) +TelemetryExportDestinationParam = TelemetryExportDestinationDict | TelemetryExportDestination diff --git a/python/databricks/bundles/apps/_models/unity_catalog.py b/python/databricks/bundles/apps/_models/unity_catalog.py index f20d1c373f3..3dafe0f017d 100644 --- a/python/databricks/bundles/apps/_models/unity_catalog.py +++ b/python/databricks/bundles/apps/_models/unity_catalog.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam +from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam +from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam +from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam +from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam +from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam +from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam +from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam +from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam +from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam +from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam +from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam +from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam +from databricks.bundles.apps._models.app import App, AppDict, AppParam +from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam +from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam +from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam + +from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam + +from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam + +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam + +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam + +from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam + +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam + +from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam + +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam + +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam + +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class UnityCatalog: """ Unity Catalog Destinations for OTEL telemetry export. """ - logs_table: VariableOr[str] """ Unity Catalog table for OTEL logs. @@ -33,11 +79,12 @@ class UnityCatalog: """ @classmethod - def from_dict(cls, value: "UnityCatalogDict") -> "Self": + def from_dict(cls, value: 'UnityCatalogDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "UnityCatalogDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'UnityCatalogDict': + return _transform_to_json_value(self) # type:ignore + class UnityCatalogDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/__init__.py b/python/databricks/bundles/catalogs/__init__.py index f36750b4816..cdd8527365f 100644 --- a/python/databricks/bundles/catalogs/__init__.py +++ b/python/databricks/bundles/catalogs/__init__.py @@ -1,57 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "AzureEncryptionSettings", - "AzureEncryptionSettingsDict", - "AzureEncryptionSettingsParam", - "Catalog", - "CatalogDict", - "CatalogGrant", - "CatalogGrantDict", - "CatalogGrantParam", - "CatalogGrantPrivilege", - "CatalogGrantPrivilegeParam", - "CatalogParam", - "EncryptionSettings", - "EncryptionSettingsDict", - "EncryptionSettingsParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", + 'AzureEncryptionSettings', + 'AzureEncryptionSettingsDict', + 'AzureEncryptionSettingsParam', + 'Catalog', + 'CatalogDict', + 'CatalogParam', + 'EncryptionSettings', + 'EncryptionSettingsDict', + 'EncryptionSettingsParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'CatalogGrant', + 'CatalogGrantDict', + 'CatalogGrantParam', + 'CatalogGrantPrivilege', + 'CatalogGrantPrivilegeParam', ] -from databricks.bundles.catalogs._models.azure_encryption_settings import ( - AzureEncryptionSettings, - AzureEncryptionSettingsDict, - AzureEncryptionSettingsParam, -) -from databricks.bundles.catalogs._models.catalog import ( - Catalog, - CatalogDict, - CatalogParam, -) -from databricks.bundles.catalogs._models.encryption_settings import ( - EncryptionSettings, - EncryptionSettingsDict, - EncryptionSettingsParam, -) -from databricks.bundles.catalogs._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) +from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam +from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam +from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam +from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam -from databricks.bundles.catalogs._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) + CatalogGrant = PrivilegeAssignment CatalogGrantDict = PrivilegeAssignmentDict diff --git a/python/databricks/bundles/catalogs/_models/azure_encryption_settings.py b/python/databricks/bundles/catalogs/_models/azure_encryption_settings.py index d68ba739a24..dcab98af41b 100644 --- a/python/databricks/bundles/catalogs/_models/azure_encryption_settings.py +++ b/python/databricks/bundles/catalogs/_models/azure_encryption_settings.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam +from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam +from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AzureEncryptionSettings: """""" @@ -22,11 +30,12 @@ class AzureEncryptionSettings: azure_cmk_managed_identity_id: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: "AzureEncryptionSettingsDict") -> "Self": + def from_dict(cls, value: 'AzureEncryptionSettingsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AzureEncryptionSettingsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AzureEncryptionSettingsDict': + return _transform_to_json_value(self) # type:ignore + class AzureEncryptionSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/_models/catalog.py b/python/databricks/bundles/catalogs/_models/catalog.py index bdb397b0cb0..43e3874b93f 100644 --- a/python/databricks/bundles/catalogs/_models/catalog.py +++ b/python/databricks/bundles/catalogs/_models/catalog.py @@ -1,31 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.catalogs._models.encryption_settings import ( - EncryptionSettings, - EncryptionSettingsParam, -) -from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.catalogs._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam +from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam +from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Catalog(Resource): """""" @@ -93,11 +86,12 @@ class Catalog(Resource): """ @classmethod - def from_dict(cls, value: "CatalogDict") -> "Self": + def from_dict(cls, value: 'CatalogDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CatalogDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CatalogDict': + return _transform_to_json_value(self) # type:ignore + class CatalogDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/_models/encryption_settings.py b/python/databricks/bundles/catalogs/_models/encryption_settings.py index dfb53029dbe..8f197ef3367 100644 --- a/python/databricks/bundles/catalogs/_models/encryption_settings.py +++ b/python/databricks/bundles/catalogs/_models/encryption_settings.py @@ -1,27 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.catalogs._models.azure_encryption_settings import ( - AzureEncryptionSettings, - AzureEncryptionSettingsParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam +from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam +from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EncryptionSettings: """ Encryption Settings are used to carry metadata for securable encryption at rest. Currently used for catalogs, we can use the information supplied here to interact with a CMK. """ - azure_encryption_settings: VariableOrOptional[AzureEncryptionSettings] = None """ optional Azure settings - only required if an Azure CMK is used. @@ -38,11 +41,12 @@ class EncryptionSettings: """ @classmethod - def from_dict(cls, value: "EncryptionSettingsDict") -> "Self": + def from_dict(cls, value: 'EncryptionSettingsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EncryptionSettingsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EncryptionSettingsDict': + return _transform_to_json_value(self) # type:ignore + class EncryptionSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/_models/lifecycle.py b/python/databricks/bundles/catalogs/_models/lifecycle.py index 697776a198e..f247f3c9202 100644 --- a/python/databricks/bundles/catalogs/_models/lifecycle.py +++ b/python/databricks/bundles/catalogs/_models/lifecycle.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam +from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam +from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam +from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +29,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/_models/privilege.py b/python/databricks/bundles/catalogs/_models/privilege.py index df45211a76a..13d8ac1ea54 100644 --- a/python/databricks/bundles/catalogs/_models/privilege.py +++ b/python/databricks/bundles/catalogs/_models/privilege.py @@ -1,8 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam +from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam +from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam +from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam class Privilege(Enum): SELECT = "SELECT" @@ -58,61 +69,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/catalogs/_models/privilege_assignment.py b/python/databricks/bundles/catalogs/_models/privilege_assignment.py index a85471c5931..fbd6dc782cd 100644 --- a/python/databricks/bundles/catalogs/_models/privilege_assignment.py +++ b/python/databricks/bundles/catalogs/_models/privilege_assignment.py @@ -1,17 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam +from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam +from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam +from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -28,11 +35,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/__init__.py b/python/databricks/bundles/cluster_policies/__init__.py index 531fa17821a..78e8a0ea7aa 100644 --- a/python/databricks/bundles/cluster_policies/__init__.py +++ b/python/databricks/bundles/cluster_policies/__init__.py @@ -1,68 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "ClusterPolicy", - "ClusterPolicyDict", - "ClusterPolicyParam", - "ClusterPolicyPermission", - "ClusterPolicyPermissionDict", - "ClusterPolicyPermissionLevel", - "ClusterPolicyPermissionLevelParam", - "ClusterPolicyPermissionParam", - "Library", - "LibraryDict", - "LibraryParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "MavenLibrary", - "MavenLibraryDict", - "MavenLibraryParam", - "PythonPyPiLibrary", - "PythonPyPiLibraryDict", - "PythonPyPiLibraryParam", - "RCranLibrary", - "RCranLibraryDict", - "RCranLibraryParam", + 'ClusterPolicy', + 'ClusterPolicyDict', + 'ClusterPolicyParam', + 'ClusterPolicyPermission', + 'ClusterPolicyPermissionDict', + 'ClusterPolicyPermissionLevel', + 'ClusterPolicyPermissionLevelParam', + 'ClusterPolicyPermissionParam', + 'Library', + 'LibraryDict', + 'LibraryParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'MavenLibrary', + 'MavenLibraryDict', + 'MavenLibraryParam', + 'PythonPyPiLibrary', + 'PythonPyPiLibraryDict', + 'PythonPyPiLibraryParam', + 'RCranLibrary', + 'RCranLibraryDict', + 'RCranLibraryParam', ] -from databricks.bundles.cluster_policies._models.cluster_policy import ( - ClusterPolicy, - ClusterPolicyDict, - ClusterPolicyParam, -) -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ( - ClusterPolicyPermission, - ClusterPolicyPermissionDict, - ClusterPolicyPermissionParam, -) -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ( - ClusterPolicyPermissionLevel, - ClusterPolicyPermissionLevelParam, -) -from databricks.bundles.cluster_policies._models.library import ( - Library, - LibraryDict, - LibraryParam, -) -from databricks.bundles.cluster_policies._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.cluster_policies._models.maven_library import ( - MavenLibrary, - MavenLibraryDict, - MavenLibraryParam, -) -from databricks.bundles.cluster_policies._models.python_py_pi_library import ( - PythonPyPiLibrary, - PythonPyPiLibraryDict, - PythonPyPiLibraryParam, -) -from databricks.bundles.cluster_policies._models.r_cran_library import ( - RCranLibrary, - RCranLibraryDict, - RCranLibraryParam, -) +from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam +from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam + diff --git a/python/databricks/bundles/cluster_policies/_models/cluster_policy.py b/python/databricks/bundles/cluster_policies/_models/cluster_policy.py index d8033d95728..e0214a61b78 100644 --- a/python/databricks/bundles/cluster_policies/_models/cluster_policy.py +++ b/python/databricks/bundles/cluster_policies/_models/cluster_policy.py @@ -1,26 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, Any, TypedDict - -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ( - ClusterPolicyPermission, - ClusterPolicyPermissionParam, -) -from databricks.bundles.cluster_policies._models.library import Library, LibraryParam -from databricks.bundles.cluster_policies._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam +from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ClusterPolicy(Resource): """""" @@ -77,11 +77,12 @@ class ClusterPolicy(Resource): """ @classmethod - def from_dict(cls, value: "ClusterPolicyDict") -> "Self": + def from_dict(cls, value: 'ClusterPolicyDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClusterPolicyDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClusterPolicyDict': + return _transform_to_json_value(self) # type:ignore + class ClusterPolicyDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission.py b/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission.py index de9e59b93a4..e1bad34183c 100644 --- a/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission.py +++ b/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission.py @@ -1,20 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ( - ClusterPolicyPermissionLevel, - ClusterPolicyPermissionLevelParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam +from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ClusterPolicyPermission: """""" @@ -40,11 +46,12 @@ class ClusterPolicyPermission: """ @classmethod - def from_dict(cls, value: "ClusterPolicyPermissionDict") -> "Self": + def from_dict(cls, value: 'ClusterPolicyPermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClusterPolicyPermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClusterPolicyPermissionDict': + return _transform_to_json_value(self) # type:ignore + class ClusterPolicyPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission_level.py b/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission_level.py index 2773971dce1..4dc021a4252 100644 --- a/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission_level.py +++ b/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission_level.py @@ -1,15 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam +from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam class ClusterPolicyPermissionLevel(Enum): """ Permission level """ - CAN_USE = "CAN_USE" - ClusterPolicyPermissionLevelParam = Literal["CAN_USE"] | ClusterPolicyPermissionLevel diff --git a/python/databricks/bundles/cluster_policies/_models/library.py b/python/databricks/bundles/cluster_policies/_models/library.py index 42781f4ccb5..21e9a466338 100644 --- a/python/databricks/bundles/cluster_policies/_models/library.py +++ b/python/databricks/bundles/cluster_policies/_models/library.py @@ -1,28 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.cluster_policies._models.maven_library import ( - MavenLibrary, - MavenLibraryParam, -) -from databricks.bundles.cluster_policies._models.python_py_pi_library import ( - PythonPyPiLibrary, - PythonPyPiLibraryParam, -) -from databricks.bundles.cluster_policies._models.r_cran_library import ( - RCranLibrary, - RCranLibraryParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam +from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Library: """""" @@ -74,11 +72,12 @@ class Library: """ @classmethod - def from_dict(cls, value: "LibraryDict") -> "Self": + def from_dict(cls, value: 'LibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LibraryDict': + return _transform_to_json_value(self) # type:ignore + class LibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/lifecycle.py b/python/databricks/bundles/cluster_policies/_models/lifecycle.py index 697776a198e..e01071d8106 100644 --- a/python/databricks/bundles/cluster_policies/_models/lifecycle.py +++ b/python/databricks/bundles/cluster_policies/_models/lifecycle.py @@ -1,16 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +31,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/maven_library.py b/python/databricks/bundles/cluster_policies/_models/maven_library.py index 25be1599c28..7569afb3fdd 100644 --- a/python/databricks/bundles/cluster_policies/_models/maven_library.py +++ b/python/databricks/bundles/cluster_policies/_models/maven_library.py @@ -1,20 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam +from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MavenLibrary: """""" @@ -39,11 +45,12 @@ class MavenLibrary: """ @classmethod - def from_dict(cls, value: "MavenLibraryDict") -> "Self": + def from_dict(cls, value: 'MavenLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MavenLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MavenLibraryDict': + return _transform_to_json_value(self) # type:ignore + class MavenLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/python_py_pi_library.py b/python/databricks/bundles/cluster_policies/_models/python_py_pi_library.py index 037c2029f50..b4d26b3fa5b 100644 --- a/python/databricks/bundles/cluster_policies/_models/python_py_pi_library.py +++ b/python/databricks/bundles/cluster_policies/_models/python_py_pi_library.py @@ -1,16 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam +from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PythonPyPiLibrary: """""" @@ -28,11 +38,12 @@ class PythonPyPiLibrary: """ @classmethod - def from_dict(cls, value: "PythonPyPiLibraryDict") -> "Self": + def from_dict(cls, value: 'PythonPyPiLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PythonPyPiLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PythonPyPiLibraryDict': + return _transform_to_json_value(self) # type:ignore + class PythonPyPiLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/r_cran_library.py b/python/databricks/bundles/cluster_policies/_models/r_cran_library.py index 0bd2c4d5296..9d91505595b 100644 --- a/python/databricks/bundles/cluster_policies/_models/r_cran_library.py +++ b/python/databricks/bundles/cluster_policies/_models/r_cran_library.py @@ -1,16 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam +from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RCranLibrary: """""" @@ -26,11 +36,12 @@ class RCranLibrary: """ @classmethod - def from_dict(cls, value: "RCranLibraryDict") -> "Self": + def from_dict(cls, value: 'RCranLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RCranLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RCranLibraryDict': + return _transform_to_json_value(self) # type:ignore + class RCranLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/__init__.py b/python/databricks/bundles/clusters/__init__.py index 0ccecb208b0..4f19f68cb15 100644 --- a/python/databricks/bundles/clusters/__init__.py +++ b/python/databricks/bundles/clusters/__init__.py @@ -1,271 +1,150 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Adlsgen2Info", - "Adlsgen2InfoDict", - "Adlsgen2InfoParam", - "AutoScale", - "AutoScaleDict", - "AutoScaleParam", - "AwsAttributes", - "AwsAttributesDict", - "AwsAttributesParam", - "AwsAvailability", - "AwsAvailabilityParam", - "AzureAttributes", - "AzureAttributesDict", - "AzureAttributesParam", - "AzureAvailability", - "AzureAvailabilityParam", - "ClientsTypes", - "ClientsTypesDict", - "ClientsTypesParam", - "Cluster", - "ClusterDict", - "ClusterLogConf", - "ClusterLogConfDict", - "ClusterLogConfParam", - "ClusterParam", - "ClusterPermission", - "ClusterPermissionDict", - "ClusterPermissionLevel", - "ClusterPermissionLevelParam", - "ClusterPermissionParam", - "ConfidentialComputeType", - "ConfidentialComputeTypeParam", - "DataSecurityMode", - "DataSecurityModeParam", - "DbfsStorageInfo", - "DbfsStorageInfoDict", - "DbfsStorageInfoParam", - "DependencyMode", - "DependencyModeParam", - "DockerBasicAuth", - "DockerBasicAuthDict", - "DockerBasicAuthParam", - "DockerImage", - "DockerImageDict", - "DockerImageParam", - "EbsVolumeType", - "EbsVolumeTypeParam", - "GcpAttributes", - "GcpAttributesDict", - "GcpAttributesParam", - "GcpAvailability", - "GcpAvailabilityParam", - "GcsStorageInfo", - "GcsStorageInfoDict", - "GcsStorageInfoParam", - "InitScriptInfo", - "InitScriptInfoDict", - "InitScriptInfoParam", - "Kind", - "KindParam", - "Library", - "LibraryDict", - "LibraryParam", - "LifecycleWithStarted", - "LifecycleWithStartedDict", - "LifecycleWithStartedParam", - "LocalFileInfo", - "LocalFileInfoDict", - "LocalFileInfoParam", - "LogAnalyticsInfo", - "LogAnalyticsInfoDict", - "LogAnalyticsInfoParam", - "MavenLibrary", - "MavenLibraryDict", - "MavenLibraryParam", - "NodeTypeFlexibility", - "NodeTypeFlexibilityDict", - "NodeTypeFlexibilityParam", - "PythonPyPiLibrary", - "PythonPyPiLibraryDict", - "PythonPyPiLibraryParam", - "RCranLibrary", - "RCranLibraryDict", - "RCranLibraryParam", - "RuntimeEngine", - "RuntimeEngineParam", - "S3StorageInfo", - "S3StorageInfoDict", - "S3StorageInfoParam", - "VolumesStorageInfo", - "VolumesStorageInfoDict", - "VolumesStorageInfoParam", - "WorkloadType", - "WorkloadTypeDict", - "WorkloadTypeParam", - "WorkspaceStorageInfo", - "WorkspaceStorageInfoDict", - "WorkspaceStorageInfoParam", + 'Adlsgen2Info', + 'Adlsgen2InfoDict', + 'Adlsgen2InfoParam', + 'AutoScale', + 'AutoScaleDict', + 'AutoScaleParam', + 'AwsAttributes', + 'AwsAttributesDict', + 'AwsAttributesParam', + 'AwsAvailability', + 'AwsAvailabilityParam', + 'AzureAttributes', + 'AzureAttributesDict', + 'AzureAttributesParam', + 'AzureAvailability', + 'AzureAvailabilityParam', + 'ClientsTypes', + 'ClientsTypesDict', + 'ClientsTypesParam', + 'Cluster', + 'ClusterDict', + 'ClusterLogConf', + 'ClusterLogConfDict', + 'ClusterLogConfParam', + 'ClusterParam', + 'ClusterPermission', + 'ClusterPermissionDict', + 'ClusterPermissionLevel', + 'ClusterPermissionLevelParam', + 'ClusterPermissionParam', + 'ConfidentialComputeType', + 'ConfidentialComputeTypeParam', + 'DataSecurityMode', + 'DataSecurityModeParam', + 'DbfsStorageInfo', + 'DbfsStorageInfoDict', + 'DbfsStorageInfoParam', + 'DependencyMode', + 'DependencyModeParam', + 'DockerBasicAuth', + 'DockerBasicAuthDict', + 'DockerBasicAuthParam', + 'DockerImage', + 'DockerImageDict', + 'DockerImageParam', + 'EbsVolumeType', + 'EbsVolumeTypeParam', + 'GcpAttributes', + 'GcpAttributesDict', + 'GcpAttributesParam', + 'GcpAvailability', + 'GcpAvailabilityParam', + 'GcsStorageInfo', + 'GcsStorageInfoDict', + 'GcsStorageInfoParam', + 'InitScriptInfo', + 'InitScriptInfoDict', + 'InitScriptInfoParam', + 'Kind', + 'KindParam', + 'Library', + 'LibraryDict', + 'LibraryParam', + 'LifecycleWithStarted', + 'LifecycleWithStartedDict', + 'LifecycleWithStartedParam', + 'LocalFileInfo', + 'LocalFileInfoDict', + 'LocalFileInfoParam', + 'LogAnalyticsInfo', + 'LogAnalyticsInfoDict', + 'LogAnalyticsInfoParam', + 'MavenLibrary', + 'MavenLibraryDict', + 'MavenLibraryParam', + 'NodeTypeFlexibility', + 'NodeTypeFlexibilityDict', + 'NodeTypeFlexibilityParam', + 'PythonPyPiLibrary', + 'PythonPyPiLibraryDict', + 'PythonPyPiLibraryParam', + 'RCranLibrary', + 'RCranLibraryDict', + 'RCranLibraryParam', + 'RuntimeEngine', + 'RuntimeEngineParam', + 'S3StorageInfo', + 'S3StorageInfoDict', + 'S3StorageInfoParam', + 'VolumesStorageInfo', + 'VolumesStorageInfoDict', + 'VolumesStorageInfoParam', + 'WorkloadType', + 'WorkloadTypeDict', + 'WorkloadTypeParam', + 'WorkspaceStorageInfo', + 'WorkspaceStorageInfoDict', + 'WorkspaceStorageInfoParam', ] -from databricks.bundles.clusters._models.adlsgen2_info import ( - Adlsgen2Info, - Adlsgen2InfoDict, - Adlsgen2InfoParam, -) -from databricks.bundles.clusters._models.auto_scale import ( - AutoScale, - AutoScaleDict, - AutoScaleParam, -) -from databricks.bundles.clusters._models.aws_attributes import ( - AwsAttributes, - AwsAttributesDict, - AwsAttributesParam, -) -from databricks.bundles.clusters._models.aws_availability import ( - AwsAvailability, - AwsAvailabilityParam, -) -from databricks.bundles.clusters._models.azure_attributes import ( - AzureAttributes, - AzureAttributesDict, - AzureAttributesParam, -) -from databricks.bundles.clusters._models.azure_availability import ( - AzureAvailability, - AzureAvailabilityParam, -) -from databricks.bundles.clusters._models.clients_types import ( - ClientsTypes, - ClientsTypesDict, - ClientsTypesParam, -) -from databricks.bundles.clusters._models.cluster import ( - Cluster, - ClusterDict, - ClusterParam, -) -from databricks.bundles.clusters._models.cluster_log_conf import ( - ClusterLogConf, - ClusterLogConfDict, - ClusterLogConfParam, -) -from databricks.bundles.clusters._models.cluster_permission import ( - ClusterPermission, - ClusterPermissionDict, - ClusterPermissionParam, -) -from databricks.bundles.clusters._models.cluster_permission_level import ( - ClusterPermissionLevel, - ClusterPermissionLevelParam, -) -from databricks.bundles.clusters._models.confidential_compute_type import ( - ConfidentialComputeType, - ConfidentialComputeTypeParam, -) -from databricks.bundles.clusters._models.data_security_mode import ( - DataSecurityMode, - DataSecurityModeParam, -) -from databricks.bundles.clusters._models.dbfs_storage_info import ( - DbfsStorageInfo, - DbfsStorageInfoDict, - DbfsStorageInfoParam, -) -from databricks.bundles.clusters._models.dependency_mode import ( - DependencyMode, - DependencyModeParam, -) -from databricks.bundles.clusters._models.docker_basic_auth import ( - DockerBasicAuth, - DockerBasicAuthDict, - DockerBasicAuthParam, -) -from databricks.bundles.clusters._models.docker_image import ( - DockerImage, - DockerImageDict, - DockerImageParam, -) -from databricks.bundles.clusters._models.ebs_volume_type import ( - EbsVolumeType, - EbsVolumeTypeParam, -) -from databricks.bundles.clusters._models.gcp_attributes import ( - GcpAttributes, - GcpAttributesDict, - GcpAttributesParam, -) -from databricks.bundles.clusters._models.gcp_availability import ( - GcpAvailability, - GcpAvailabilityParam, -) -from databricks.bundles.clusters._models.gcs_storage_info import ( - GcsStorageInfo, - GcsStorageInfoDict, - GcsStorageInfoParam, -) -from databricks.bundles.clusters._models.init_script_info import ( - InitScriptInfo, - InitScriptInfoDict, - InitScriptInfoParam, -) +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + from databricks.bundles.clusters._models.kind import Kind, KindParam -from databricks.bundles.clusters._models.library import ( - Library, - LibraryDict, - LibraryParam, -) -from databricks.bundles.clusters._models.lifecycle_with_started import ( - LifecycleWithStarted, - LifecycleWithStartedDict, - LifecycleWithStartedParam, -) -from databricks.bundles.clusters._models.local_file_info import ( - LocalFileInfo, - LocalFileInfoDict, - LocalFileInfoParam, -) -from databricks.bundles.clusters._models.log_analytics_info import ( - LogAnalyticsInfo, - LogAnalyticsInfoDict, - LogAnalyticsInfoParam, -) -from databricks.bundles.clusters._models.maven_library import ( - MavenLibrary, - MavenLibraryDict, - MavenLibraryParam, -) -from databricks.bundles.clusters._models.node_type_flexibility import ( - NodeTypeFlexibility, - NodeTypeFlexibilityDict, - NodeTypeFlexibilityParam, -) -from databricks.bundles.clusters._models.python_py_pi_library import ( - PythonPyPiLibrary, - PythonPyPiLibraryDict, - PythonPyPiLibraryParam, -) -from databricks.bundles.clusters._models.r_cran_library import ( - RCranLibrary, - RCranLibraryDict, - RCranLibraryParam, -) -from databricks.bundles.clusters._models.runtime_engine import ( - RuntimeEngine, - RuntimeEngineParam, -) -from databricks.bundles.clusters._models.s3_storage_info import ( - S3StorageInfo, - S3StorageInfoDict, - S3StorageInfoParam, -) -from databricks.bundles.clusters._models.volumes_storage_info import ( - VolumesStorageInfo, - VolumesStorageInfoDict, - VolumesStorageInfoParam, -) -from databricks.bundles.clusters._models.workload_type import ( - WorkloadType, - WorkloadTypeDict, - WorkloadTypeParam, -) -from databricks.bundles.clusters._models.workspace_storage_info import ( - WorkspaceStorageInfo, - WorkspaceStorageInfoDict, - WorkspaceStorageInfoParam, -) + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + diff --git a/python/databricks/bundles/clusters/_models/adlsgen2_info.py b/python/databricks/bundles/clusters/_models/adlsgen2_info.py index 87c883adeb9..0f943c35844 100644 --- a/python/databricks/bundles/clusters/_models/adlsgen2_info.py +++ b/python/databricks/bundles/clusters/_models/adlsgen2_info.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Adlsgen2Info: """ A storage location in Adls Gen2 """ - destination: VariableOr[str] """ abfss destination, e.g. `abfss://@.dfs.core.windows.net/`. """ @classmethod - def from_dict(cls, value: "Adlsgen2InfoDict") -> "Self": + def from_dict(cls, value: 'Adlsgen2InfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "Adlsgen2InfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'Adlsgen2InfoDict': + return _transform_to_json_value(self) # type:ignore + class Adlsgen2InfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/auto_scale.py b/python/databricks/bundles/clusters/_models/auto_scale.py index 10ca50c26d7..1d436e99351 100644 --- a/python/databricks/bundles/clusters/_models/auto_scale.py +++ b/python/databricks/bundles/clusters/_models/auto_scale.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AutoScale: """""" @@ -28,11 +75,12 @@ class AutoScale: """ @classmethod - def from_dict(cls, value: "AutoScaleDict") -> "Self": + def from_dict(cls, value: 'AutoScaleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AutoScaleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AutoScaleDict': + return _transform_to_json_value(self) # type:ignore + class AutoScaleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/aws_attributes.py b/python/databricks/bundles/clusters/_models/aws_attributes.py index 2ff19359882..bf9e194de16 100644 --- a/python/databricks/bundles/clusters/_models/aws_attributes.py +++ b/python/databricks/bundles/clusters/_models/aws_attributes.py @@ -1,30 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.clusters._models.aws_availability import ( - AwsAvailability, - AwsAvailabilityParam, -) -from databricks.bundles.clusters._models.ebs_volume_type import ( - EbsVolumeType, - EbsVolumeTypeParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AwsAttributes: """ Attributes set during cluster creation which are related to Amazon Web Services. """ - availability: VariableOrOptional[AwsAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -123,11 +161,12 @@ class AwsAttributes: """ @classmethod - def from_dict(cls, value: "AwsAttributesDict") -> "Self": + def from_dict(cls, value: 'AwsAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AwsAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AwsAttributesDict': + return _transform_to_json_value(self) # type:ignore + class AwsAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/aws_availability.py b/python/databricks/bundles/clusters/_models/aws_availability.py index 4c3810f067e..c9b2dffc181 100644 --- a/python/databricks/bundles/clusters/_models/aws_availability.py +++ b/python/databricks/bundles/clusters/_models/aws_availability.py @@ -1,21 +1,67 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam class AwsAvailability(Enum): """ Availability type used for all subsequent nodes past the `first_on_demand` ones. - + Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ - SPOT = "SPOT" ON_DEMAND = "ON_DEMAND" SPOT_WITH_FALLBACK = "SPOT_WITH_FALLBACK" - -AwsAvailabilityParam = ( - Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability -) +AwsAvailabilityParam = Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability diff --git a/python/databricks/bundles/clusters/_models/azure_attributes.py b/python/databricks/bundles/clusters/_models/azure_attributes.py index f5c5cbee36d..0c57c96417a 100644 --- a/python/databricks/bundles/clusters/_models/azure_attributes.py +++ b/python/databricks/bundles/clusters/_models/azure_attributes.py @@ -1,30 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.clusters._models.azure_availability import ( - AzureAvailability, - AzureAvailabilityParam, -) -from databricks.bundles.clusters._models.log_analytics_info import ( - LogAnalyticsInfo, - LogAnalyticsInfoParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AzureAttributes: """ Attributes set during cluster creation which are related to Microsoft Azure. """ - availability: VariableOrOptional[AzureAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -72,11 +110,12 @@ class AzureAttributes: """ @classmethod - def from_dict(cls, value: "AzureAttributesDict") -> "Self": + def from_dict(cls, value: 'AzureAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AzureAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AzureAttributesDict': + return _transform_to_json_value(self) # type:ignore + class AzureAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/azure_availability.py b/python/databricks/bundles/clusters/_models/azure_availability.py index a03cb1fdddf..a61c7c57906 100644 --- a/python/databricks/bundles/clusters/_models/azure_availability.py +++ b/python/databricks/bundles/clusters/_models/azure_availability.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam class AzureAvailability(Enum): @@ -9,13 +59,8 @@ class AzureAvailability(Enum): Availability type used for all subsequent nodes past the `first_on_demand` ones. Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ - SPOT_AZURE = "SPOT_AZURE" ON_DEMAND_AZURE = "ON_DEMAND_AZURE" SPOT_WITH_FALLBACK_AZURE = "SPOT_WITH_FALLBACK_AZURE" - -AzureAvailabilityParam = ( - Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] - | AzureAvailability -) +AzureAvailabilityParam = Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] | AzureAvailability diff --git a/python/databricks/bundles/clusters/_models/clients_types.py b/python/databricks/bundles/clusters/_models/clients_types.py index 87f92446ba5..af4a62b9842 100644 --- a/python/databricks/bundles/clusters/_models/clients_types.py +++ b/python/databricks/bundles/clusters/_models/clients_types.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ClientsTypes: """""" @@ -26,11 +73,12 @@ class ClientsTypes: """ @classmethod - def from_dict(cls, value: "ClientsTypesDict") -> "Self": + def from_dict(cls, value: 'ClientsTypesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClientsTypesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClientsTypesDict': + return _transform_to_json_value(self) # type:ignore + class ClientsTypesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/cluster.py b/python/databricks/bundles/clusters/_models/cluster.py index cc3e9fcde04..20546fbe5b4 100644 --- a/python/databricks/bundles/clusters/_models/cluster.py +++ b/python/databricks/bundles/clusters/_models/cluster.py @@ -1,85 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.clusters._models.auto_scale import ( - AutoScale, - AutoScaleParam, -) -from databricks.bundles.clusters._models.aws_attributes import ( - AwsAttributes, - AwsAttributesParam, -) -from databricks.bundles.clusters._models.azure_attributes import ( - AzureAttributes, - AzureAttributesParam, -) -from databricks.bundles.clusters._models.cluster_log_conf import ( - ClusterLogConf, - ClusterLogConfParam, -) -from databricks.bundles.clusters._models.cluster_permission import ( - ClusterPermission, - ClusterPermissionParam, -) -from databricks.bundles.clusters._models.data_security_mode import ( - DataSecurityMode, - DataSecurityModeParam, -) -from databricks.bundles.clusters._models.dependency_mode import ( - DependencyMode, - DependencyModeParam, -) -from databricks.bundles.clusters._models.docker_image import ( - DockerImage, - DockerImageParam, -) -from databricks.bundles.clusters._models.gcp_attributes import ( - GcpAttributes, - GcpAttributesParam, -) -from databricks.bundles.clusters._models.init_script_info import ( - InitScriptInfo, - InitScriptInfoParam, -) -from databricks.bundles.clusters._models.kind import Kind, KindParam -from databricks.bundles.clusters._models.library import Library, LibraryParam -from databricks.bundles.clusters._models.lifecycle_with_started import ( - LifecycleWithStarted, - LifecycleWithStartedParam, -) -from databricks.bundles.clusters._models.node_type_flexibility import ( - NodeTypeFlexibility, - NodeTypeFlexibilityParam, -) -from databricks.bundles.clusters._models.runtime_engine import ( - RuntimeEngine, - RuntimeEngineParam, -) -from databricks.bundles.clusters._models.workload_type import ( - WorkloadType, - WorkloadTypeParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOrDict, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Cluster(Resource): """ Contains a snapshot of the latest user specified settings that were used to create/edit the cluster. """ - apply_policy_default_values: VariableOrOptional[bool] = None """ When set to true, fixed and default values from the policy will be used for fields that are omitted. When set to false, only fixed values from the policy will be applied. @@ -360,11 +343,12 @@ class Cluster(Resource): """ @classmethod - def from_dict(cls, value: "ClusterDict") -> "Self": + def from_dict(cls, value: 'ClusterDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClusterDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClusterDict': + return _transform_to_json_value(self) # type:ignore + class ClusterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/cluster_log_conf.py b/python/databricks/bundles/clusters/_models/cluster_log_conf.py index c087a2ad463..18926bd14dc 100644 --- a/python/databricks/bundles/clusters/_models/cluster_log_conf.py +++ b/python/databricks/bundles/clusters/_models/cluster_log_conf.py @@ -1,34 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.clusters._models.dbfs_storage_info import ( - DbfsStorageInfo, - DbfsStorageInfoParam, -) -from databricks.bundles.clusters._models.s3_storage_info import ( - S3StorageInfo, - S3StorageInfoParam, -) -from databricks.bundles.clusters._models.volumes_storage_info import ( - VolumesStorageInfo, - VolumesStorageInfoParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ClusterLogConf: """ Cluster log delivery config """ - dbfs: VariableOrOptional[DbfsStorageInfo] = None """ destination needs to be provided. e.g. @@ -50,11 +84,12 @@ class ClusterLogConf: """ @classmethod - def from_dict(cls, value: "ClusterLogConfDict") -> "Self": + def from_dict(cls, value: 'ClusterLogConfDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClusterLogConfDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClusterLogConfDict': + return _transform_to_json_value(self) # type:ignore + class ClusterLogConfDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/cluster_permission.py b/python/databricks/bundles/clusters/_models/cluster_permission.py index 537e136780a..c40850089e0 100644 --- a/python/databricks/bundles/clusters/_models/cluster_permission.py +++ b/python/databricks/bundles/clusters/_models/cluster_permission.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.clusters._models.cluster_permission_level import ( - ClusterPermissionLevel, - ClusterPermissionLevelParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ClusterPermission: """""" @@ -40,11 +83,12 @@ class ClusterPermission: """ @classmethod - def from_dict(cls, value: "ClusterPermissionDict") -> "Self": + def from_dict(cls, value: 'ClusterPermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClusterPermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClusterPermissionDict': + return _transform_to_json_value(self) # type:ignore + class ClusterPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/cluster_permission_level.py b/python/databricks/bundles/clusters/_models/cluster_permission_level.py index bb8b1347d92..5f9266fab7e 100644 --- a/python/databricks/bundles/clusters/_models/cluster_permission_level.py +++ b/python/databricks/bundles/clusters/_models/cluster_permission_level.py @@ -1,19 +1,65 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam class ClusterPermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" - -ClusterPermissionLevelParam = ( - Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO"] | ClusterPermissionLevel -) +ClusterPermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO"] | ClusterPermissionLevel diff --git a/python/databricks/bundles/clusters/_models/confidential_compute_type.py b/python/databricks/bundles/clusters/_models/confidential_compute_type.py index e881295bd47..beed47d71d9 100644 --- a/python/databricks/bundles/clusters/_models/confidential_compute_type.py +++ b/python/databricks/bundles/clusters/_models/confidential_compute_type.py @@ -1,23 +1,69 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam class ConfidentialComputeType(Enum): """ :meta private: [EXPERIMENTAL] - + Confidential computing technology for GCP instances. Aligns with gcloud's --confidential-compute-type flag and the REST API's confidentialInstanceConfig.confidentialInstanceType field. See: https://cloud.google.com/confidential-computing/confidential-vm/docs/create-a-confidential-vm-instance """ - CONFIDENTIAL_COMPUTE_TYPE_NONE = "CONFIDENTIAL_COMPUTE_TYPE_NONE" SEV_SNP = "SEV_SNP" - -ConfidentialComputeTypeParam = ( - Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType -) +ConfidentialComputeTypeParam = Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType diff --git a/python/databricks/bundles/clusters/_models/data_security_mode.py b/python/databricks/bundles/clusters/_models/data_security_mode.py index ae1aafcf085..88976123873 100644 --- a/python/databricks/bundles/clusters/_models/data_security_mode.py +++ b/python/databricks/bundles/clusters/_models/data_security_mode.py @@ -1,32 +1,81 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam class DataSecurityMode(Enum): """ Data security mode decides what data governance model to use when accessing data from a cluster. - + * `DATA_SECURITY_MODE_AUTO`: Databricks will choose the most appropriate access mode depending on your compute configuration. * `DATA_SECURITY_MODE_STANDARD`: A secure cluster that can be shared by multiple users. Cluster users are fully isolated so that they cannot see each other’s data and credentials. Most data governance features are supported in this mode. But programming languages and cluster features might be limited. * `DATA_SECURITY_MODE_DEDICATED`: A secure cluster that can only be exclusively used by a single user specified in `single_user_name`. Most programming languages, cluster features and data governance features are available in this mode. - + The following modes are legacy aliases for the above modes: - + * `USER_ISOLATION`: Legacy alias for `DATA_SECURITY_MODE_STANDARD`. * `SINGLE_USER`: Legacy alias for `DATA_SECURITY_MODE_DEDICATED`. - + The following modes are deprecated starting with Databricks Runtime 15.0 and will be removed for future Databricks Runtime versions: - + * `LEGACY_TABLE_ACL`: This mode is for users migrating from legacy Table ACL clusters. * `LEGACY_PASSTHROUGH`: This mode is for users migrating from legacy Passthrough on high concurrency clusters. * `LEGACY_SINGLE_USER`: This mode is for users migrating from legacy Passthrough on standard clusters. * `LEGACY_SINGLE_USER_STANDARD`: This mode provides a way that doesn’t have UC nor passthrough enabled. """ - NONE = "NONE" SINGLE_USER = "SINGLE_USER" USER_ISOLATION = "USER_ISOLATION" @@ -38,19 +87,4 @@ class DataSecurityMode(Enum): DATA_SECURITY_MODE_DEDICATED = "DATA_SECURITY_MODE_DEDICATED" DATA_SECURITY_MODE_AUTO = "DATA_SECURITY_MODE_AUTO" - -DataSecurityModeParam = ( - Literal[ - "NONE", - "SINGLE_USER", - "USER_ISOLATION", - "LEGACY_TABLE_ACL", - "LEGACY_PASSTHROUGH", - "LEGACY_SINGLE_USER", - "LEGACY_SINGLE_USER_STANDARD", - "DATA_SECURITY_MODE_STANDARD", - "DATA_SECURITY_MODE_DEDICATED", - "DATA_SECURITY_MODE_AUTO", - ] - | DataSecurityMode -) +DataSecurityModeParam = Literal["NONE", "SINGLE_USER", "USER_ISOLATION", "LEGACY_TABLE_ACL", "LEGACY_PASSTHROUGH", "LEGACY_SINGLE_USER", "LEGACY_SINGLE_USER_STANDARD", "DATA_SECURITY_MODE_STANDARD", "DATA_SECURITY_MODE_DEDICATED", "DATA_SECURITY_MODE_AUTO"] | DataSecurityMode diff --git a/python/databricks/bundles/clusters/_models/dbfs_storage_info.py b/python/databricks/bundles/clusters/_models/dbfs_storage_info.py index 29744acf26a..8d5bf7ed0bb 100644 --- a/python/databricks/bundles/clusters/_models/dbfs_storage_info.py +++ b/python/databricks/bundles/clusters/_models/dbfs_storage_info.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DbfsStorageInfo: """ A storage location in DBFS """ - destination: VariableOr[str] """ dbfs destination, e.g. `dbfs:/my/path` """ @classmethod - def from_dict(cls, value: "DbfsStorageInfoDict") -> "Self": + def from_dict(cls, value: 'DbfsStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DbfsStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DbfsStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class DbfsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/dependency_mode.py b/python/databricks/bundles/clusters/_models/dependency_mode.py index 1fdc22cfa4c..c54af29c916 100644 --- a/python/databricks/bundles/clusters/_models/dependency_mode.py +++ b/python/databricks/bundles/clusters/_models/dependency_mode.py @@ -1,28 +1,69 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam class DependencyMode(Enum): """ Controls dependency configuration for the cluster. - + * `DEPENDENCY_MODE_AUTO`: Databricks will choose the most appropriate dependency mode based on your compute configuration. * `DEPENDENCY_MODE_ENVIRONMENTS`: Enables a unified dependency management experience across classic and serverless, resulting in increased stability and performance. Supported only on DBR 19+ in Standard access mode. * `DEPENDENCY_MODE_CLUSTER_LIBRARIES`: Legacy mode: dependencies come from cluster libraries and init scripts. """ - DEPENDENCY_MODE_ENVIRONMENTS = "DEPENDENCY_MODE_ENVIRONMENTS" DEPENDENCY_MODE_CLUSTER_LIBRARIES = "DEPENDENCY_MODE_CLUSTER_LIBRARIES" DEPENDENCY_MODE_AUTO = "DEPENDENCY_MODE_AUTO" - -DependencyModeParam = ( - Literal[ - "DEPENDENCY_MODE_ENVIRONMENTS", - "DEPENDENCY_MODE_CLUSTER_LIBRARIES", - "DEPENDENCY_MODE_AUTO", - ] - | DependencyMode -) +DependencyModeParam = Literal["DEPENDENCY_MODE_ENVIRONMENTS", "DEPENDENCY_MODE_CLUSTER_LIBRARIES", "DEPENDENCY_MODE_AUTO"] | DependencyMode diff --git a/python/databricks/bundles/clusters/_models/docker_basic_auth.py b/python/databricks/bundles/clusters/_models/docker_basic_auth.py index 552ea90a83b..b235d24953b 100644 --- a/python/databricks/bundles/clusters/_models/docker_basic_auth.py +++ b/python/databricks/bundles/clusters/_models/docker_basic_auth.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DockerBasicAuth: """""" @@ -26,11 +73,12 @@ class DockerBasicAuth: """ @classmethod - def from_dict(cls, value: "DockerBasicAuthDict") -> "Self": + def from_dict(cls, value: 'DockerBasicAuthDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DockerBasicAuthDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DockerBasicAuthDict': + return _transform_to_json_value(self) # type:ignore + class DockerBasicAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/docker_image.py b/python/databricks/bundles/clusters/_models/docker_image.py index f7c47633e07..1e65b80f881 100644 --- a/python/databricks/bundles/clusters/_models/docker_image.py +++ b/python/databricks/bundles/clusters/_models/docker_image.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.clusters._models.docker_basic_auth import ( - DockerBasicAuth, - DockerBasicAuthParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DockerImage: """""" @@ -30,11 +73,12 @@ class DockerImage: """ @classmethod - def from_dict(cls, value: "DockerImageDict") -> "Self": + def from_dict(cls, value: 'DockerImageDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DockerImageDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DockerImageDict': + return _transform_to_json_value(self) # type:ignore + class DockerImageDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/ebs_volume_type.py b/python/databricks/bundles/clusters/_models/ebs_volume_type.py index d8f3ef13f89..55599e421cd 100644 --- a/python/databricks/bundles/clusters/_models/ebs_volume_type.py +++ b/python/databricks/bundles/clusters/_models/ebs_volume_type.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam class EbsVolumeType(Enum): @@ -9,11 +59,7 @@ class EbsVolumeType(Enum): All EBS volume types that Databricks supports. See https://aws.amazon.com/ebs/details/ for details. """ - GENERAL_PURPOSE_SSD = "GENERAL_PURPOSE_SSD" THROUGHPUT_OPTIMIZED_HDD = "THROUGHPUT_OPTIMIZED_HDD" - -EbsVolumeTypeParam = ( - Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType -) +EbsVolumeTypeParam = Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType diff --git a/python/databricks/bundles/clusters/_models/gcp_attributes.py b/python/databricks/bundles/clusters/_models/gcp_attributes.py index e31188d405e..ba74a57205b 100644 --- a/python/databricks/bundles/clusters/_models/gcp_attributes.py +++ b/python/databricks/bundles/clusters/_models/gcp_attributes.py @@ -1,30 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.clusters._models.confidential_compute_type import ( - ConfidentialComputeType, - ConfidentialComputeTypeParam, -) -from databricks.bundles.clusters._models.gcp_availability import ( - GcpAvailability, - GcpAvailabilityParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GcpAttributes: """ Attributes set during cluster creation which are related to GCP. """ - availability: VariableOrOptional[GcpAvailability] = None """ This field determines whether the spark executors will be scheduled to run on preemptible @@ -91,11 +129,12 @@ class GcpAttributes: """ @classmethod - def from_dict(cls, value: "GcpAttributesDict") -> "Self": + def from_dict(cls, value: 'GcpAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GcpAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GcpAttributesDict': + return _transform_to_json_value(self) # type:ignore + class GcpAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/gcp_availability.py b/python/databricks/bundles/clusters/_models/gcp_availability.py index 0d391b87fe3..2893d6d8e6a 100644 --- a/python/databricks/bundles/clusters/_models/gcp_availability.py +++ b/python/databricks/bundles/clusters/_models/gcp_availability.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam class GcpAvailability(Enum): @@ -9,13 +59,8 @@ class GcpAvailability(Enum): This field determines whether the instance pool will contain preemptible VMs, on-demand VMs, or preemptible VMs with a fallback to on-demand VMs if the former is unavailable. """ - PREEMPTIBLE_GCP = "PREEMPTIBLE_GCP" ON_DEMAND_GCP = "ON_DEMAND_GCP" PREEMPTIBLE_WITH_FALLBACK_GCP = "PREEMPTIBLE_WITH_FALLBACK_GCP" - -GcpAvailabilityParam = ( - Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] - | GcpAvailability -) +GcpAvailabilityParam = Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] | GcpAvailability diff --git a/python/databricks/bundles/clusters/_models/gcs_storage_info.py b/python/databricks/bundles/clusters/_models/gcs_storage_info.py index e8f1059373b..d8b53dd6310 100644 --- a/python/databricks/bundles/clusters/_models/gcs_storage_info.py +++ b/python/databricks/bundles/clusters/_models/gcs_storage_info.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GcsStorageInfo: """ A storage location in Google Cloud Platform's GCS """ - destination: VariableOr[str] """ GCS destination/URI, e.g. `gs://my-bucket/some-prefix` """ @classmethod - def from_dict(cls, value: "GcsStorageInfoDict") -> "Self": + def from_dict(cls, value: 'GcsStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GcsStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GcsStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class GcsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/init_script_info.py b/python/databricks/bundles/clusters/_models/init_script_info.py index 6d78d16bf92..75c8fbd8436 100644 --- a/python/databricks/bundles/clusters/_models/init_script_info.py +++ b/python/databricks/bundles/clusters/_models/init_script_info.py @@ -1,50 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.clusters._models.adlsgen2_info import ( - Adlsgen2Info, - Adlsgen2InfoParam, -) -from databricks.bundles.clusters._models.dbfs_storage_info import ( - DbfsStorageInfo, - DbfsStorageInfoParam, -) -from databricks.bundles.clusters._models.gcs_storage_info import ( - GcsStorageInfo, - GcsStorageInfoParam, -) -from databricks.bundles.clusters._models.local_file_info import ( - LocalFileInfo, - LocalFileInfoParam, -) -from databricks.bundles.clusters._models.s3_storage_info import ( - S3StorageInfo, - S3StorageInfoParam, -) -from databricks.bundles.clusters._models.volumes_storage_info import ( - VolumesStorageInfo, - VolumesStorageInfoParam, -) -from databricks.bundles.clusters._models.workspace_storage_info import ( - WorkspaceStorageInfo, - WorkspaceStorageInfoParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InitScriptInfo: """ Config for an individual init script """ - abfss: VariableOrOptional[Adlsgen2Info] = None """ Contains the Azure Data Lake Storage destination path @@ -89,11 +107,12 @@ class InitScriptInfo: """ @classmethod - def from_dict(cls, value: "InitScriptInfoDict") -> "Self": + def from_dict(cls, value: 'InitScriptInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InitScriptInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InitScriptInfoDict': + return _transform_to_json_value(self) # type:ignore + class InitScriptInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/kind.py b/python/databricks/bundles/clusters/_models/kind.py index 8614e25dd52..32672d3d1e3 100644 --- a/python/databricks/bundles/clusters/_models/kind.py +++ b/python/databricks/bundles/clusters/_models/kind.py @@ -1,23 +1,71 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam class Kind(Enum): """ The kind of compute described by this compute specification. - + Depending on `kind`, different validations and default values will be applied. - + Clusters with `kind = CLASSIC_PREVIEW` support the following fields, whereas clusters with no specified `kind` do not. * [is_single_node](/api/workspace/clusters/create#is_single_node) * [use_ml_runtime](/api/workspace/clusters/create#use_ml_runtime) - + By using the [simple form](https://docs.databricks.com/compute/simple-form.html), your clusters are automatically using `kind = CLASSIC_PREVIEW`. """ - CLASSIC_PREVIEW = "CLASSIC_PREVIEW" - KindParam = Literal["CLASSIC_PREVIEW"] | Kind diff --git a/python/databricks/bundles/clusters/_models/library.py b/python/databricks/bundles/clusters/_models/library.py index 8961f8fe744..0436500dfdb 100644 --- a/python/databricks/bundles/clusters/_models/library.py +++ b/python/databricks/bundles/clusters/_models/library.py @@ -1,28 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict - -from databricks.bundles.clusters._models.maven_library import ( - MavenLibrary, - MavenLibraryParam, -) -from databricks.bundles.clusters._models.python_py_pi_library import ( - PythonPyPiLibrary, - PythonPyPiLibraryParam, -) -from databricks.bundles.clusters._models.r_cran_library import ( - RCranLibrary, - RCranLibraryParam, -) +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Library: """""" @@ -74,11 +109,12 @@ class Library: """ @classmethod - def from_dict(cls, value: "LibraryDict") -> "Self": + def from_dict(cls, value: 'LibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LibraryDict': + return _transform_to_json_value(self) # type:ignore + class LibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/lifecycle_with_started.py b/python/databricks/bundles/clusters/_models/lifecycle_with_started.py index f9cb9b8d49c..1c947497092 100644 --- a/python/databricks/bundles/clusters/_models/lifecycle_with_started.py +++ b/python/databricks/bundles/clusters/_models/lifecycle_with_started.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LifecycleWithStarted: """""" @@ -26,11 +73,12 @@ class LifecycleWithStarted: """ @classmethod - def from_dict(cls, value: "LifecycleWithStartedDict") -> "Self": + def from_dict(cls, value: 'LifecycleWithStartedDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleWithStartedDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleWithStartedDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleWithStartedDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/local_file_info.py b/python/databricks/bundles/clusters/_models/local_file_info.py index 875f2fe8353..0b52c0d10fa 100644 --- a/python/databricks/bundles/clusters/_models/local_file_info.py +++ b/python/databricks/bundles/clusters/_models/local_file_info.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LocalFileInfo: """""" @@ -21,11 +68,12 @@ class LocalFileInfo: """ @classmethod - def from_dict(cls, value: "LocalFileInfoDict") -> "Self": + def from_dict(cls, value: 'LocalFileInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LocalFileInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LocalFileInfoDict': + return _transform_to_json_value(self) # type:ignore + class LocalFileInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/log_analytics_info.py b/python/databricks/bundles/clusters/_models/log_analytics_info.py index 67cdf85b2b9..ba78f2ca12c 100644 --- a/python/databricks/bundles/clusters/_models/log_analytics_info.py +++ b/python/databricks/bundles/clusters/_models/log_analytics_info.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LogAnalyticsInfo: """""" @@ -26,11 +73,12 @@ class LogAnalyticsInfo: """ @classmethod - def from_dict(cls, value: "LogAnalyticsInfoDict") -> "Self": + def from_dict(cls, value: 'LogAnalyticsInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LogAnalyticsInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LogAnalyticsInfoDict': + return _transform_to_json_value(self) # type:ignore + class LogAnalyticsInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/maven_library.py b/python/databricks/bundles/clusters/_models/maven_library.py index 25be1599c28..f5cad160966 100644 --- a/python/databricks/bundles/clusters/_models/maven_library.py +++ b/python/databricks/bundles/clusters/_models/maven_library.py @@ -1,20 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MavenLibrary: """""" @@ -39,11 +82,12 @@ class MavenLibrary: """ @classmethod - def from_dict(cls, value: "MavenLibraryDict") -> "Self": + def from_dict(cls, value: 'MavenLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MavenLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MavenLibraryDict': + return _transform_to_json_value(self) # type:ignore + class MavenLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/node_type_flexibility.py b/python/databricks/bundles/clusters/_models/node_type_flexibility.py index 418ea9337bd..00e00238d4e 100644 --- a/python/databricks/bundles/clusters/_models/node_type_flexibility.py +++ b/python/databricks/bundles/clusters/_models/node_type_flexibility.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class NodeTypeFlexibility: """ Configuration for flexible node types, allowing fallback to alternate node types during cluster launch and upscale. """ - alternate_node_type_ids: VariableOrList[str] = field(default_factory=list) """ A list of node type IDs to use as fallbacks when the primary node type is unavailable. @@ -29,11 +75,12 @@ class NodeTypeFlexibility: """ @classmethod - def from_dict(cls, value: "NodeTypeFlexibilityDict") -> "Self": + def from_dict(cls, value: 'NodeTypeFlexibilityDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "NodeTypeFlexibilityDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'NodeTypeFlexibilityDict': + return _transform_to_json_value(self) # type:ignore + class NodeTypeFlexibilityDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/python_py_pi_library.py b/python/databricks/bundles/clusters/_models/python_py_pi_library.py index 037c2029f50..94d29d699c6 100644 --- a/python/databricks/bundles/clusters/_models/python_py_pi_library.py +++ b/python/databricks/bundles/clusters/_models/python_py_pi_library.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PythonPyPiLibrary: """""" @@ -28,11 +75,12 @@ class PythonPyPiLibrary: """ @classmethod - def from_dict(cls, value: "PythonPyPiLibraryDict") -> "Self": + def from_dict(cls, value: 'PythonPyPiLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PythonPyPiLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PythonPyPiLibraryDict': + return _transform_to_json_value(self) # type:ignore + class PythonPyPiLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/r_cran_library.py b/python/databricks/bundles/clusters/_models/r_cran_library.py index 0bd2c4d5296..a12f8fd9db7 100644 --- a/python/databricks/bundles/clusters/_models/r_cran_library.py +++ b/python/databricks/bundles/clusters/_models/r_cran_library.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RCranLibrary: """""" @@ -26,11 +73,12 @@ class RCranLibrary: """ @classmethod - def from_dict(cls, value: "RCranLibraryDict") -> "Self": + def from_dict(cls, value: 'RCranLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RCranLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RCranLibraryDict': + return _transform_to_json_value(self) # type:ignore + class RCranLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/runtime_engine.py b/python/databricks/bundles/clusters/_models/runtime_engine.py index 2e1559fa801..e452a2537e2 100644 --- a/python/databricks/bundles/clusters/_models/runtime_engine.py +++ b/python/databricks/bundles/clusters/_models/runtime_engine.py @@ -1,7 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam class RuntimeEngine(Enum): @@ -9,5 +59,4 @@ class RuntimeEngine(Enum): STANDARD = "STANDARD" PHOTON = "PHOTON" - RuntimeEngineParam = Literal["NULL", "STANDARD", "PHOTON"] | RuntimeEngine diff --git a/python/databricks/bundles/clusters/_models/s3_storage_info.py b/python/databricks/bundles/clusters/_models/s3_storage_info.py index 071abd5b6f3..b4725668e6c 100644 --- a/python/databricks/bundles/clusters/_models/s3_storage_info.py +++ b/python/databricks/bundles/clusters/_models/s3_storage_info.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class S3StorageInfo: """ A storage location in Amazon S3 """ - destination: VariableOr[str] """ S3 destination, e.g. `s3://my-bucket/some-prefix` Note that logs will be delivered using @@ -64,11 +110,12 @@ class S3StorageInfo: """ @classmethod - def from_dict(cls, value: "S3StorageInfoDict") -> "Self": + def from_dict(cls, value: 'S3StorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "S3StorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'S3StorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class S3StorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/volumes_storage_info.py b/python/databricks/bundles/clusters/_models/volumes_storage_info.py index 1eb4fff7bc2..286fcf86728 100644 --- a/python/databricks/bundles/clusters/_models/volumes_storage_info.py +++ b/python/databricks/bundles/clusters/_models/volumes_storage_info.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class VolumesStorageInfo: """ A storage location back by UC Volumes. """ - destination: VariableOr[str] """ UC Volumes destination, e.g. `/Volumes/catalog/schema/vol1/init-scripts/setup-datadog.sh` @@ -24,11 +70,12 @@ class VolumesStorageInfo: """ @classmethod - def from_dict(cls, value: "VolumesStorageInfoDict") -> "Self": + def from_dict(cls, value: 'VolumesStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "VolumesStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'VolumesStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class VolumesStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/workload_type.py b/python/databricks/bundles/clusters/_models/workload_type.py index 6760e7d7edd..28867bb95f6 100644 --- a/python/databricks/bundles/clusters/_models/workload_type.py +++ b/python/databricks/bundles/clusters/_models/workload_type.py @@ -1,37 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field -from databricks.bundles.clusters._models.clients_types import ( - ClientsTypes, - ClientsTypesParam, -) +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class WorkloadType: """ Cluster Attributes showing for clusters workload types. """ - clients: VariableOr[ClientsTypes] """ defined what type of clients can use the cluster. E.g. Notebooks, Jobs """ @classmethod - def from_dict(cls, value: "WorkloadTypeDict") -> "Self": + def from_dict(cls, value: 'WorkloadTypeDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "WorkloadTypeDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'WorkloadTypeDict': + return _transform_to_json_value(self) # type:ignore + class WorkloadTypeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/workspace_storage_info.py b/python/databricks/bundles/clusters/_models/workspace_storage_info.py index 49aef9256c8..7d309c7a53a 100644 --- a/python/databricks/bundles/clusters/_models/workspace_storage_info.py +++ b/python/databricks/bundles/clusters/_models/workspace_storage_info.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam +from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam + +from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.clusters._models.kind import Kind, KindParam + +from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class WorkspaceStorageInfo: """ A storage location in Workspace Filesystem (WSFS) """ - destination: VariableOr[str] """ wsfs destination, e.g. `workspace:/cluster-init-scripts/setup-datadog.sh` """ @classmethod - def from_dict(cls, value: "WorkspaceStorageInfoDict") -> "Self": + def from_dict(cls, value: 'WorkspaceStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "WorkspaceStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'WorkspaceStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class WorkspaceStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/core/__init__.py b/python/databricks/bundles/core/__init__.py index b40cb9f9684..ede73a09089 100644 --- a/python/databricks/bundles/core/__init__.py +++ b/python/databricks/bundles/core/__init__.py @@ -58,6 +58,24 @@ Diagnostics, Severity, ) +from databricks.bundles.core._load import ( + load_resources_from_current_package_module, + load_resources_from_module, + load_resources_from_modules, + load_resources_from_package_module, +) +from databricks.bundles.core._location import Location +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._resource_mutator import ResourceMutator +from databricks.bundles.core._resources import Resources +from databricks.bundles.core._variable import ( + Variable, + VariableOr, + VariableOrDict, + VariableOrList, + VariableOrOptional, + variables, +) from databricks.bundles.core._generated import ( alert_mutator, app_mutator, @@ -91,21 +109,3 @@ vector_search_index_mutator, volume_mutator, ) -from databricks.bundles.core._load import ( - load_resources_from_current_package_module, - load_resources_from_module, - load_resources_from_modules, - load_resources_from_package_module, -) -from databricks.bundles.core._location import Location -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._resource_mutator import ResourceMutator -from databricks.bundles.core._resources import Resources -from databricks.bundles.core._variable import ( - Variable, - VariableOr, - VariableOrDict, - VariableOrList, - VariableOrOptional, - variables, -) diff --git a/python/databricks/bundles/core/_generated/__init__.py b/python/databricks/bundles/core/_generated/__init__.py index 6a56d2eb856..31a1548612b 100644 --- a/python/databricks/bundles/core/_generated/__init__.py +++ b/python/databricks/bundles/core/_generated/__init__.py @@ -4,109 +4,34 @@ from databricks.bundles.core._generated.alerts import _AlertResources, alert_mutator from databricks.bundles.core._generated.apps import _AppResources, app_mutator -from databricks.bundles.core._generated.catalogs import ( - _CatalogResources, - catalog_mutator, -) -from databricks.bundles.core._generated.cluster_policies import ( - _ClusterPolicyResources, - cluster_policy_mutator, -) -from databricks.bundles.core._generated.clusters import ( - _ClusterResources, - cluster_mutator, -) -from databricks.bundles.core._generated.dashboards import ( - _DashboardResources, - dashboard_mutator, -) -from databricks.bundles.core._generated.database_catalogs import ( - _DatabaseCatalogResources, - database_catalog_mutator, -) -from databricks.bundles.core._generated.database_instances import ( - _DatabaseInstanceResources, - database_instance_mutator, -) -from databricks.bundles.core._generated.experiments import ( - _MlflowExperimentResources, - mlflow_experiment_mutator, -) -from databricks.bundles.core._generated.external_locations import ( - _ExternalLocationResources, - external_location_mutator, -) -from databricks.bundles.core._generated.features import ( - _FeatureResources, - feature_mutator, -) -from databricks.bundles.core._generated.genie_spaces import ( - _GenieSpaceResources, - genie_space_mutator, -) -from databricks.bundles.core._generated.instance_pools import ( - _InstancePoolResources, - instance_pool_mutator, -) -from databricks.bundles.core._generated.job_runs import ( - _JobRunResources, - job_run_mutator, -) +from databricks.bundles.core._generated.catalogs import _CatalogResources, catalog_mutator +from databricks.bundles.core._generated.cluster_policies import _ClusterPolicyResources, cluster_policy_mutator +from databricks.bundles.core._generated.clusters import _ClusterResources, cluster_mutator +from databricks.bundles.core._generated.dashboards import _DashboardResources, dashboard_mutator +from databricks.bundles.core._generated.database_catalogs import _DatabaseCatalogResources, database_catalog_mutator +from databricks.bundles.core._generated.database_instances import _DatabaseInstanceResources, database_instance_mutator +from databricks.bundles.core._generated.experiments import _MlflowExperimentResources, mlflow_experiment_mutator +from databricks.bundles.core._generated.external_locations import _ExternalLocationResources, external_location_mutator +from databricks.bundles.core._generated.features import _FeatureResources, feature_mutator +from databricks.bundles.core._generated.genie_spaces import _GenieSpaceResources, genie_space_mutator +from databricks.bundles.core._generated.instance_pools import _InstancePoolResources, instance_pool_mutator +from databricks.bundles.core._generated.job_runs import _JobRunResources, job_run_mutator from databricks.bundles.core._generated.jobs import _JobResources, job_mutator -from databricks.bundles.core._generated.mcp_services import ( - _McpServiceResources, - mcp_service_mutator, -) -from databricks.bundles.core._generated.model_provider_services import ( - _ModelProviderServiceResources, - model_provider_service_mutator, -) -from databricks.bundles.core._generated.model_services import ( - _ModelServiceResources, - model_service_mutator, -) -from databricks.bundles.core._generated.model_serving_endpoints import ( - _ModelServingEndpointResources, - model_serving_endpoint_mutator, -) -from databricks.bundles.core._generated.models import ( - _MlflowModelResources, - mlflow_model_mutator, -) -from databricks.bundles.core._generated.pipelines import ( - _PipelineResources, - pipeline_mutator, -) -from databricks.bundles.core._generated.quality_monitors import ( - _QualityMonitorResources, - quality_monitor_mutator, -) -from databricks.bundles.core._generated.registered_models import ( - _RegisteredModelResources, - registered_model_mutator, -) +from databricks.bundles.core._generated.mcp_services import _McpServiceResources, mcp_service_mutator +from databricks.bundles.core._generated.model_provider_services import _ModelProviderServiceResources, model_provider_service_mutator +from databricks.bundles.core._generated.model_services import _ModelServiceResources, model_service_mutator +from databricks.bundles.core._generated.model_serving_endpoints import _ModelServingEndpointResources, model_serving_endpoint_mutator +from databricks.bundles.core._generated.models import _MlflowModelResources, mlflow_model_mutator +from databricks.bundles.core._generated.pipelines import _PipelineResources, pipeline_mutator +from databricks.bundles.core._generated.quality_monitors import _QualityMonitorResources, quality_monitor_mutator +from databricks.bundles.core._generated.registered_models import _RegisteredModelResources, registered_model_mutator from databricks.bundles.core._generated.schemas import _SchemaResources, schema_mutator -from databricks.bundles.core._generated.secret_scopes import ( - _SecretScopeResources, - secret_scope_mutator, -) +from databricks.bundles.core._generated.secret_scopes import _SecretScopeResources, secret_scope_mutator from databricks.bundles.core._generated.secrets import _SecretResources, secret_mutator -from databricks.bundles.core._generated.sql_warehouses import ( - _SqlWarehouseResources, - sql_warehouse_mutator, -) -from databricks.bundles.core._generated.synced_database_tables import ( - _SyncedDatabaseTableResources, - synced_database_table_mutator, -) -from databricks.bundles.core._generated.vector_search_endpoints import ( - _VectorSearchEndpointResources, - vector_search_endpoint_mutator, -) -from databricks.bundles.core._generated.vector_search_indexes import ( - _VectorSearchIndexResources, - vector_search_index_mutator, -) +from databricks.bundles.core._generated.sql_warehouses import _SqlWarehouseResources, sql_warehouse_mutator +from databricks.bundles.core._generated.synced_database_tables import _SyncedDatabaseTableResources, synced_database_table_mutator +from databricks.bundles.core._generated.vector_search_endpoints import _VectorSearchEndpointResources, vector_search_endpoint_mutator +from databricks.bundles.core._generated.vector_search_indexes import _VectorSearchIndexResources, vector_search_index_mutator from databricks.bundles.core._generated.volumes import _VolumeResources, volume_mutator if TYPE_CHECKING: diff --git a/python/databricks/bundles/core/_generated/alerts.py b/python/databricks/bundles/core/_generated/alerts.py index 94e4af9a136..95a8da2a6a2 100644 --- a/python/databricks/bundles/core/_generated/alerts.py +++ b/python/databricks/bundles/core/_generated/alerts.py @@ -9,13 +9,13 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: - from databricks.bundles.alerts._models.alert import Alert, AlertParam from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.alerts._models.alert import Alert, AlertParam def _resource_type() -> "_ResourceType": - from databricks.bundles.alerts._models.alert import Alert from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.alerts._models.alert import Alert return _ResourceType( resource_type=Alert, diff --git a/python/databricks/bundles/core/_generated/apps.py b/python/databricks/bundles/core/_generated/apps.py index 51d14e2c602..bb3c7b4e3cf 100644 --- a/python/databricks/bundles/core/_generated/apps.py +++ b/python/databricks/bundles/core/_generated/apps.py @@ -9,13 +9,13 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: - from databricks.bundles.apps._models.app import App, AppParam from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.apps._models.app import App, AppParam def _resource_type() -> "_ResourceType": - from databricks.bundles.apps._models.app import App from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.apps._models.app import App return _ResourceType( resource_type=App, diff --git a/python/databricks/bundles/core/_generated/catalogs.py b/python/databricks/bundles/core/_generated/catalogs.py index 55d6bbcd310..80869aa9b66 100644 --- a/python/databricks/bundles/core/_generated/catalogs.py +++ b/python/databricks/bundles/core/_generated/catalogs.py @@ -9,13 +9,13 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: - from databricks.bundles.catalogs._models.catalog import Catalog, CatalogParam from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.catalogs._models.catalog import Catalog, CatalogParam def _resource_type() -> "_ResourceType": - from databricks.bundles.catalogs._models.catalog import Catalog from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.catalogs._models.catalog import Catalog return _ResourceType( resource_type=Catalog, diff --git a/python/databricks/bundles/core/_generated/cluster_policies.py b/python/databricks/bundles/core/_generated/cluster_policies.py index 82da760997b..6899f2f9d4f 100644 --- a/python/databricks/bundles/core/_generated/cluster_policies.py +++ b/python/databricks/bundles/core/_generated/cluster_policies.py @@ -9,16 +9,13 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: - from databricks.bundles.cluster_policies._models.cluster_policy import ( - ClusterPolicy, - ClusterPolicyParam, - ) from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyParam def _resource_type() -> "_ResourceType": - from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy return _ResourceType( resource_type=ClusterPolicy, @@ -67,9 +64,7 @@ def add_cluster_policy( :param cluster_policy: the cluster_policy to add, can be ClusterPolicy or dict :param location: optional location of the cluster_policy in the source code """ - from databricks.bundles.cluster_policies._models.cluster_policy import ( - ClusterPolicy, - ) + from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy cluster_policy = _transform(ClusterPolicy, cluster_policy) path = ("resources", "cluster_policies", resource_name) diff --git a/python/databricks/bundles/core/_generated/clusters.py b/python/databricks/bundles/core/_generated/clusters.py index 44c94108ac4..5fa9216e6a1 100644 --- a/python/databricks/bundles/core/_generated/clusters.py +++ b/python/databricks/bundles/core/_generated/clusters.py @@ -9,13 +9,13 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: - from databricks.bundles.clusters._models.cluster import Cluster, ClusterParam from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.clusters._models.cluster import Cluster, ClusterParam def _resource_type() -> "_ResourceType": - from databricks.bundles.clusters._models.cluster import Cluster from databricks.bundles.core._resource_type import _ResourceType + from databricks.bundles.clusters._models.cluster import Cluster return _ResourceType( resource_type=Cluster, diff --git a/python/databricks/bundles/core/_generated/dashboards.py b/python/databricks/bundles/core/_generated/dashboards.py index 75cbb368354..ae185162c23 100644 --- a/python/databricks/bundles/core/_generated/dashboards.py +++ b/python/databricks/bundles/core/_generated/dashboards.py @@ -10,10 +10,7 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.dashboards._models.dashboard import ( - Dashboard, - DashboardParam, - ) + from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardParam def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/database_catalogs.py b/python/databricks/bundles/core/_generated/database_catalogs.py index a11e165b17f..6ff8d0d023b 100644 --- a/python/databricks/bundles/core/_generated/database_catalogs.py +++ b/python/databricks/bundles/core/_generated/database_catalogs.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.database_catalogs._models.database_catalog import ( - DatabaseCatalog, - DatabaseCatalogParam, - ) + from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog, DatabaseCatalogParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.database_catalogs._models.database_catalog import ( - DatabaseCatalog, - ) + from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog return _ResourceType( resource_type=DatabaseCatalog, @@ -69,9 +64,7 @@ def add_database_catalog( :param database_catalog: the database_catalog to add, can be DatabaseCatalog or dict :param location: optional location of the database_catalog in the source code """ - from databricks.bundles.database_catalogs._models.database_catalog import ( - DatabaseCatalog, - ) + from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog database_catalog = _transform(DatabaseCatalog, database_catalog) path = ("resources", "database_catalogs", resource_name) @@ -117,8 +110,6 @@ def my_database_catalog_mutator(bundle: Bundle, database_catalog: DatabaseCatalo :param function: Function that mutates database_catalogs. """ - from databricks.bundles.database_catalogs._models.database_catalog import ( - DatabaseCatalog, - ) + from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog return ResourceMutator(resource_type=DatabaseCatalog, function=function) diff --git a/python/databricks/bundles/core/_generated/database_instances.py b/python/databricks/bundles/core/_generated/database_instances.py index 9352ce5a37f..8eff3dbdb55 100644 --- a/python/databricks/bundles/core/_generated/database_instances.py +++ b/python/databricks/bundles/core/_generated/database_instances.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.database_instances._models.database_instance import ( - DatabaseInstance, - DatabaseInstanceParam, - ) + from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.database_instances._models.database_instance import ( - DatabaseInstance, - ) + from databricks.bundles.database_instances._models.database_instance import DatabaseInstance return _ResourceType( resource_type=DatabaseInstance, @@ -69,9 +64,7 @@ def add_database_instance( :param database_instance: the database_instance to add, can be DatabaseInstance or dict :param location: optional location of the database_instance in the source code """ - from databricks.bundles.database_instances._models.database_instance import ( - DatabaseInstance, - ) + from databricks.bundles.database_instances._models.database_instance import DatabaseInstance database_instance = _transform(DatabaseInstance, database_instance) path = ("resources", "database_instances", resource_name) @@ -102,9 +95,7 @@ def database_instance_mutator( ) -> ResourceMutator["DatabaseInstance"]: ... -def database_instance_mutator( - function: Callable, -) -> ResourceMutator["DatabaseInstance"]: +def database_instance_mutator(function: Callable) -> ResourceMutator["DatabaseInstance"]: """ Decorator for defining mutator for database_instances. Function should return a new instance of the database_instance with the desired changes, instead of mutating the input database_instance. @@ -119,8 +110,6 @@ def my_database_instance_mutator(bundle: Bundle, database_instance: DatabaseInst :param function: Function that mutates database_instances. """ - from databricks.bundles.database_instances._models.database_instance import ( - DatabaseInstance, - ) + from databricks.bundles.database_instances._models.database_instance import DatabaseInstance return ResourceMutator(resource_type=DatabaseInstance, function=function) diff --git a/python/databricks/bundles/core/_generated/experiments.py b/python/databricks/bundles/core/_generated/experiments.py index ca0860f3155..3b8868d51f8 100644 --- a/python/databricks/bundles/core/_generated/experiments.py +++ b/python/databricks/bundles/core/_generated/experiments.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.experiments._models.mlflow_experiment import ( - MlflowExperiment, - MlflowExperimentParam, - ) + from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.experiments._models.mlflow_experiment import ( - MlflowExperiment, - ) + from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment return _ResourceType( resource_type=MlflowExperiment, @@ -69,9 +64,7 @@ def add_mlflow_experiment( :param mlflow_experiment: the mlflow_experiment to add, can be MlflowExperiment or dict :param location: optional location of the mlflow_experiment in the source code """ - from databricks.bundles.experiments._models.mlflow_experiment import ( - MlflowExperiment, - ) + from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment mlflow_experiment = _transform(MlflowExperiment, mlflow_experiment) path = ("resources", "experiments", resource_name) @@ -102,9 +95,7 @@ def mlflow_experiment_mutator( ) -> ResourceMutator["MlflowExperiment"]: ... -def mlflow_experiment_mutator( - function: Callable, -) -> ResourceMutator["MlflowExperiment"]: +def mlflow_experiment_mutator(function: Callable) -> ResourceMutator["MlflowExperiment"]: """ Decorator for defining mutator for experiments. Function should return a new instance of the mlflow_experiment with the desired changes, instead of mutating the input mlflow_experiment. @@ -119,8 +110,6 @@ def my_mlflow_experiment_mutator(bundle: Bundle, mlflow_experiment: MlflowExperi :param function: Function that mutates experiments. """ - from databricks.bundles.experiments._models.mlflow_experiment import ( - MlflowExperiment, - ) + from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment return ResourceMutator(resource_type=MlflowExperiment, function=function) diff --git a/python/databricks/bundles/core/_generated/external_locations.py b/python/databricks/bundles/core/_generated/external_locations.py index d30243989a2..92d29e4a857 100644 --- a/python/databricks/bundles/core/_generated/external_locations.py +++ b/python/databricks/bundles/core/_generated/external_locations.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.external_locations._models.external_location import ( - ExternalLocation, - ExternalLocationParam, - ) + from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.external_locations._models.external_location import ( - ExternalLocation, - ) + from databricks.bundles.external_locations._models.external_location import ExternalLocation return _ResourceType( resource_type=ExternalLocation, @@ -69,9 +64,7 @@ def add_external_location( :param external_location: the external_location to add, can be ExternalLocation or dict :param location: optional location of the external_location in the source code """ - from databricks.bundles.external_locations._models.external_location import ( - ExternalLocation, - ) + from databricks.bundles.external_locations._models.external_location import ExternalLocation external_location = _transform(ExternalLocation, external_location) path = ("resources", "external_locations", resource_name) @@ -102,9 +95,7 @@ def external_location_mutator( ) -> ResourceMutator["ExternalLocation"]: ... -def external_location_mutator( - function: Callable, -) -> ResourceMutator["ExternalLocation"]: +def external_location_mutator(function: Callable) -> ResourceMutator["ExternalLocation"]: """ Decorator for defining mutator for external_locations. Function should return a new instance of the external_location with the desired changes, instead of mutating the input external_location. @@ -119,8 +110,6 @@ def my_external_location_mutator(bundle: Bundle, external_location: ExternalLoca :param function: Function that mutates external_locations. """ - from databricks.bundles.external_locations._models.external_location import ( - ExternalLocation, - ) + from databricks.bundles.external_locations._models.external_location import ExternalLocation return ResourceMutator(resource_type=ExternalLocation, function=function) diff --git a/python/databricks/bundles/core/_generated/genie_spaces.py b/python/databricks/bundles/core/_generated/genie_spaces.py index 37fd6cad561..7c6bea33299 100644 --- a/python/databricks/bundles/core/_generated/genie_spaces.py +++ b/python/databricks/bundles/core/_generated/genie_spaces.py @@ -10,10 +10,7 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.genie_spaces._models.genie_space import ( - GenieSpace, - GenieSpaceParam, - ) + from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceParam def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/instance_pools.py b/python/databricks/bundles/core/_generated/instance_pools.py index 0ce75fc3d2e..dd996369e5d 100644 --- a/python/databricks/bundles/core/_generated/instance_pools.py +++ b/python/databricks/bundles/core/_generated/instance_pools.py @@ -10,10 +10,7 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.instance_pools._models.instance_pool import ( - InstancePool, - InstancePoolParam, - ) + from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolParam def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/mcp_services.py b/python/databricks/bundles/core/_generated/mcp_services.py index 5e5bf32ae08..213a3768bc9 100644 --- a/python/databricks/bundles/core/_generated/mcp_services.py +++ b/python/databricks/bundles/core/_generated/mcp_services.py @@ -10,10 +10,7 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.mcp_services._models.mcp_service import ( - McpService, - McpServiceParam, - ) + from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceParam def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/model_provider_services.py b/python/databricks/bundles/core/_generated/model_provider_services.py index 89f5b631fcc..4275e42138d 100644 --- a/python/databricks/bundles/core/_generated/model_provider_services.py +++ b/python/databricks/bundles/core/_generated/model_provider_services.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_provider_services._models.model_provider_service import ( - ModelProviderService, - ModelProviderServiceParam, - ) + from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_provider_services._models.model_provider_service import ( - ModelProviderService, - ) + from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService return _ResourceType( resource_type=ModelProviderService, @@ -69,13 +64,9 @@ def add_model_provider_service( :param model_provider_service: the model_provider_service to add, can be ModelProviderService or dict :param location: optional location of the model_provider_service in the source code """ - from databricks.bundles.model_provider_services._models.model_provider_service import ( - ModelProviderService, - ) + from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService - model_provider_service = _transform( - ModelProviderService, model_provider_service - ) + model_provider_service = _transform(ModelProviderService, model_provider_service) path = ("resources", "model_provider_services", resource_name) location = location or Location.from_stack_frame(depth=1) @@ -89,9 +80,7 @@ def add_model_provider_service( if location: self.add_location(path, location) - self._resources["model_provider_services"][resource_name] = ( - model_provider_service - ) + self._resources["model_provider_services"][resource_name] = model_provider_service @overload @@ -106,9 +95,7 @@ def model_provider_service_mutator( ) -> ResourceMutator["ModelProviderService"]: ... -def model_provider_service_mutator( - function: Callable, -) -> ResourceMutator["ModelProviderService"]: +def model_provider_service_mutator(function: Callable) -> ResourceMutator["ModelProviderService"]: """ Decorator for defining mutator for model_provider_services. Function should return a new instance of the model_provider_service with the desired changes, instead of mutating the input model_provider_service. @@ -123,8 +110,6 @@ def my_model_provider_service_mutator(bundle: Bundle, model_provider_service: Mo :param function: Function that mutates model_provider_services. """ - from databricks.bundles.model_provider_services._models.model_provider_service import ( - ModelProviderService, - ) + from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService return ResourceMutator(resource_type=ModelProviderService, function=function) diff --git a/python/databricks/bundles/core/_generated/model_services.py b/python/databricks/bundles/core/_generated/model_services.py index 2b37ebe7d6b..7b09fedda52 100644 --- a/python/databricks/bundles/core/_generated/model_services.py +++ b/python/databricks/bundles/core/_generated/model_services.py @@ -10,10 +10,7 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_services._models.model_service import ( - ModelService, - ModelServiceParam, - ) + from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceParam def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/model_serving_endpoints.py b/python/databricks/bundles/core/_generated/model_serving_endpoints.py index 56a9ecde661..06d8fcc0616 100644 --- a/python/databricks/bundles/core/_generated/model_serving_endpoints.py +++ b/python/databricks/bundles/core/_generated/model_serving_endpoints.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( - ModelServingEndpoint, - ModelServingEndpointParam, - ) + from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( - ModelServingEndpoint, - ) + from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint return _ResourceType( resource_type=ModelServingEndpoint, @@ -69,13 +64,9 @@ def add_model_serving_endpoint( :param model_serving_endpoint: the model_serving_endpoint to add, can be ModelServingEndpoint or dict :param location: optional location of the model_serving_endpoint in the source code """ - from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( - ModelServingEndpoint, - ) + from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint - model_serving_endpoint = _transform( - ModelServingEndpoint, model_serving_endpoint - ) + model_serving_endpoint = _transform(ModelServingEndpoint, model_serving_endpoint) path = ("resources", "model_serving_endpoints", resource_name) location = location or Location.from_stack_frame(depth=1) @@ -89,9 +80,7 @@ def add_model_serving_endpoint( if location: self.add_location(path, location) - self._resources["model_serving_endpoints"][resource_name] = ( - model_serving_endpoint - ) + self._resources["model_serving_endpoints"][resource_name] = model_serving_endpoint @overload @@ -106,9 +95,7 @@ def model_serving_endpoint_mutator( ) -> ResourceMutator["ModelServingEndpoint"]: ... -def model_serving_endpoint_mutator( - function: Callable, -) -> ResourceMutator["ModelServingEndpoint"]: +def model_serving_endpoint_mutator(function: Callable) -> ResourceMutator["ModelServingEndpoint"]: """ Decorator for defining mutator for model_serving_endpoints. Function should return a new instance of the model_serving_endpoint with the desired changes, instead of mutating the input model_serving_endpoint. @@ -123,8 +110,6 @@ def my_model_serving_endpoint_mutator(bundle: Bundle, model_serving_endpoint: Mo :param function: Function that mutates model_serving_endpoints. """ - from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( - ModelServingEndpoint, - ) + from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint return ResourceMutator(resource_type=ModelServingEndpoint, function=function) diff --git a/python/databricks/bundles/core/_generated/models.py b/python/databricks/bundles/core/_generated/models.py index df7a4c6d6a3..ab4caad1bca 100644 --- a/python/databricks/bundles/core/_generated/models.py +++ b/python/databricks/bundles/core/_generated/models.py @@ -10,10 +10,7 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.models._models.mlflow_model import ( - MlflowModel, - MlflowModelParam, - ) + from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelParam def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/quality_monitors.py b/python/databricks/bundles/core/_generated/quality_monitors.py index ba3f52a75ae..d8c46b01e42 100644 --- a/python/databricks/bundles/core/_generated/quality_monitors.py +++ b/python/databricks/bundles/core/_generated/quality_monitors.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.quality_monitors._models.quality_monitor import ( - QualityMonitor, - QualityMonitorParam, - ) + from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.quality_monitors._models.quality_monitor import ( - QualityMonitor, - ) + from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor return _ResourceType( resource_type=QualityMonitor, @@ -69,9 +64,7 @@ def add_quality_monitor( :param quality_monitor: the quality_monitor to add, can be QualityMonitor or dict :param location: optional location of the quality_monitor in the source code """ - from databricks.bundles.quality_monitors._models.quality_monitor import ( - QualityMonitor, - ) + from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor quality_monitor = _transform(QualityMonitor, quality_monitor) path = ("resources", "quality_monitors", resource_name) @@ -117,8 +110,6 @@ def my_quality_monitor_mutator(bundle: Bundle, quality_monitor: QualityMonitor) :param function: Function that mutates quality_monitors. """ - from databricks.bundles.quality_monitors._models.quality_monitor import ( - QualityMonitor, - ) + from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor return ResourceMutator(resource_type=QualityMonitor, function=function) diff --git a/python/databricks/bundles/core/_generated/registered_models.py b/python/databricks/bundles/core/_generated/registered_models.py index cdc492beeb3..42bfb70bf88 100644 --- a/python/databricks/bundles/core/_generated/registered_models.py +++ b/python/databricks/bundles/core/_generated/registered_models.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.registered_models._models.registered_model import ( - RegisteredModel, - RegisteredModelParam, - ) + from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.registered_models._models.registered_model import ( - RegisteredModel, - ) + from databricks.bundles.registered_models._models.registered_model import RegisteredModel return _ResourceType( resource_type=RegisteredModel, @@ -69,9 +64,7 @@ def add_registered_model( :param registered_model: the registered_model to add, can be RegisteredModel or dict :param location: optional location of the registered_model in the source code """ - from databricks.bundles.registered_models._models.registered_model import ( - RegisteredModel, - ) + from databricks.bundles.registered_models._models.registered_model import RegisteredModel registered_model = _transform(RegisteredModel, registered_model) path = ("resources", "registered_models", resource_name) @@ -117,8 +110,6 @@ def my_registered_model_mutator(bundle: Bundle, registered_model: RegisteredMode :param function: Function that mutates registered_models. """ - from databricks.bundles.registered_models._models.registered_model import ( - RegisteredModel, - ) + from databricks.bundles.registered_models._models.registered_model import RegisteredModel return ResourceMutator(resource_type=RegisteredModel, function=function) diff --git a/python/databricks/bundles/core/_generated/secret_scopes.py b/python/databricks/bundles/core/_generated/secret_scopes.py index a8be3660a6a..4635888bc8f 100644 --- a/python/databricks/bundles/core/_generated/secret_scopes.py +++ b/python/databricks/bundles/core/_generated/secret_scopes.py @@ -10,10 +10,7 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.secret_scopes._models.secret_scope import ( - SecretScope, - SecretScopeParam, - ) + from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeParam def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/sql_warehouses.py b/python/databricks/bundles/core/_generated/sql_warehouses.py index 05f8bb0cf76..7630d8af559 100644 --- a/python/databricks/bundles/core/_generated/sql_warehouses.py +++ b/python/databricks/bundles/core/_generated/sql_warehouses.py @@ -10,10 +10,7 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.sql_warehouses._models.sql_warehouse import ( - SqlWarehouse, - SqlWarehouseParam, - ) + from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseParam def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/synced_database_tables.py b/python/databricks/bundles/core/_generated/synced_database_tables.py index 2d0c7e598c0..90139c00d58 100644 --- a/python/databricks/bundles/core/_generated/synced_database_tables.py +++ b/python/databricks/bundles/core/_generated/synced_database_tables.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.synced_database_tables._models.synced_database_table import ( - SyncedDatabaseTable, - SyncedDatabaseTableParam, - ) + from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.synced_database_tables._models.synced_database_table import ( - SyncedDatabaseTable, - ) + from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable return _ResourceType( resource_type=SyncedDatabaseTable, @@ -69,9 +64,7 @@ def add_synced_database_table( :param synced_database_table: the synced_database_table to add, can be SyncedDatabaseTable or dict :param location: optional location of the synced_database_table in the source code """ - from databricks.bundles.synced_database_tables._models.synced_database_table import ( - SyncedDatabaseTable, - ) + from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable synced_database_table = _transform(SyncedDatabaseTable, synced_database_table) path = ("resources", "synced_database_tables", resource_name) @@ -87,9 +80,7 @@ def add_synced_database_table( if location: self.add_location(path, location) - self._resources["synced_database_tables"][resource_name] = ( - synced_database_table - ) + self._resources["synced_database_tables"][resource_name] = synced_database_table @overload @@ -104,9 +95,7 @@ def synced_database_table_mutator( ) -> ResourceMutator["SyncedDatabaseTable"]: ... -def synced_database_table_mutator( - function: Callable, -) -> ResourceMutator["SyncedDatabaseTable"]: +def synced_database_table_mutator(function: Callable) -> ResourceMutator["SyncedDatabaseTable"]: """ Decorator for defining mutator for synced_database_tables. Function should return a new instance of the synced_database_table with the desired changes, instead of mutating the input synced_database_table. @@ -121,8 +110,6 @@ def my_synced_database_table_mutator(bundle: Bundle, synced_database_table: Sync :param function: Function that mutates synced_database_tables. """ - from databricks.bundles.synced_database_tables._models.synced_database_table import ( - SyncedDatabaseTable, - ) + from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable return ResourceMutator(resource_type=SyncedDatabaseTable, function=function) diff --git a/python/databricks/bundles/core/_generated/vector_search_endpoints.py b/python/databricks/bundles/core/_generated/vector_search_endpoints.py index 839831dafc6..54f6777c0f7 100644 --- a/python/databricks/bundles/core/_generated/vector_search_endpoints.py +++ b/python/databricks/bundles/core/_generated/vector_search_endpoints.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( - VectorSearchEndpoint, - VectorSearchEndpointParam, - ) + from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( - VectorSearchEndpoint, - ) + from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint return _ResourceType( resource_type=VectorSearchEndpoint, @@ -69,13 +64,9 @@ def add_vector_search_endpoint( :param vector_search_endpoint: the vector_search_endpoint to add, can be VectorSearchEndpoint or dict :param location: optional location of the vector_search_endpoint in the source code """ - from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( - VectorSearchEndpoint, - ) + from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint - vector_search_endpoint = _transform( - VectorSearchEndpoint, vector_search_endpoint - ) + vector_search_endpoint = _transform(VectorSearchEndpoint, vector_search_endpoint) path = ("resources", "vector_search_endpoints", resource_name) location = location or Location.from_stack_frame(depth=1) @@ -89,9 +80,7 @@ def add_vector_search_endpoint( if location: self.add_location(path, location) - self._resources["vector_search_endpoints"][resource_name] = ( - vector_search_endpoint - ) + self._resources["vector_search_endpoints"][resource_name] = vector_search_endpoint @overload @@ -106,9 +95,7 @@ def vector_search_endpoint_mutator( ) -> ResourceMutator["VectorSearchEndpoint"]: ... -def vector_search_endpoint_mutator( - function: Callable, -) -> ResourceMutator["VectorSearchEndpoint"]: +def vector_search_endpoint_mutator(function: Callable) -> ResourceMutator["VectorSearchEndpoint"]: """ Decorator for defining mutator for vector_search_endpoints. Function should return a new instance of the vector_search_endpoint with the desired changes, instead of mutating the input vector_search_endpoint. @@ -123,8 +110,6 @@ def my_vector_search_endpoint_mutator(bundle: Bundle, vector_search_endpoint: Ve :param function: Function that mutates vector_search_endpoints. """ - from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( - VectorSearchEndpoint, - ) + from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint return ResourceMutator(resource_type=VectorSearchEndpoint, function=function) diff --git a/python/databricks/bundles/core/_generated/vector_search_indexes.py b/python/databricks/bundles/core/_generated/vector_search_indexes.py index 7c84a0d00c8..d85c65ced41 100644 --- a/python/databricks/bundles/core/_generated/vector_search_indexes.py +++ b/python/databricks/bundles/core/_generated/vector_search_indexes.py @@ -10,17 +10,12 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.vector_search_indexes._models.vector_search_index import ( - VectorSearchIndex, - VectorSearchIndexParam, - ) + from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexParam def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.vector_search_indexes._models.vector_search_index import ( - VectorSearchIndex, - ) + from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex return _ResourceType( resource_type=VectorSearchIndex, @@ -69,9 +64,7 @@ def add_vector_search_index( :param vector_search_index: the vector_search_index to add, can be VectorSearchIndex or dict :param location: optional location of the vector_search_index in the source code """ - from databricks.bundles.vector_search_indexes._models.vector_search_index import ( - VectorSearchIndex, - ) + from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex vector_search_index = _transform(VectorSearchIndex, vector_search_index) path = ("resources", "vector_search_indexes", resource_name) @@ -87,9 +80,7 @@ def add_vector_search_index( if location: self.add_location(path, location) - self._resources["vector_search_indexes"][resource_name] = ( - vector_search_index - ) + self._resources["vector_search_indexes"][resource_name] = vector_search_index @overload @@ -104,9 +95,7 @@ def vector_search_index_mutator( ) -> ResourceMutator["VectorSearchIndex"]: ... -def vector_search_index_mutator( - function: Callable, -) -> ResourceMutator["VectorSearchIndex"]: +def vector_search_index_mutator(function: Callable) -> ResourceMutator["VectorSearchIndex"]: """ Decorator for defining mutator for vector_search_indexes. Function should return a new instance of the vector_search_index with the desired changes, instead of mutating the input vector_search_index. @@ -121,8 +110,6 @@ def my_vector_search_index_mutator(bundle: Bundle, vector_search_index: VectorSe :param function: Function that mutates vector_search_indexes. """ - from databricks.bundles.vector_search_indexes._models.vector_search_index import ( - VectorSearchIndex, - ) + from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex return ResourceMutator(resource_type=VectorSearchIndex, function=function) diff --git a/python/databricks/bundles/dashboards/__init__.py b/python/databricks/bundles/dashboards/__init__.py index 126427fcca6..8e8c855760d 100644 --- a/python/databricks/bundles/dashboards/__init__.py +++ b/python/databricks/bundles/dashboards/__init__.py @@ -1,42 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Dashboard", - "DashboardDict", - "DashboardParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "LifecycleState", - "LifecycleStateParam", - "Permission", - "PermissionDict", - "PermissionLevel", - "PermissionLevelParam", - "PermissionParam", + 'Dashboard', + 'DashboardDict', + 'DashboardParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'LifecycleState', + 'LifecycleStateParam', + 'Permission', + 'PermissionDict', + 'PermissionLevel', + 'PermissionLevelParam', + 'PermissionParam', ] -from databricks.bundles.dashboards._models.dashboard import ( - Dashboard, - DashboardDict, - DashboardParam, -) -from databricks.bundles.dashboards._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.dashboards._models.lifecycle_state import ( - LifecycleState, - LifecycleStateParam, -) -from databricks.bundles.dashboards._models.permission import ( - Permission, - PermissionDict, - PermissionParam, -) -from databricks.bundles.dashboards._models.permission_level import ( - PermissionLevel, - PermissionLevelParam, -) +from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam +from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam + +from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam + diff --git a/python/databricks/bundles/dashboards/_models/dashboard.py b/python/databricks/bundles/dashboards/_models/dashboard.py index 2a3e5f03364..6c1eae4c71f 100644 --- a/python/databricks/bundles/dashboards/_models/dashboard.py +++ b/python/databricks/bundles/dashboards/_models/dashboard.py @@ -1,23 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, Any, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.dashboards._models.lifecycle_state import ( - LifecycleState, - LifecycleStateParam, -) -from databricks.bundles.dashboards._models.permission import Permission, PermissionParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam + +from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Dashboard(Resource): """""" @@ -109,11 +110,12 @@ class Dashboard(Resource): """ @classmethod - def from_dict(cls, value: "DashboardDict") -> "Self": + def from_dict(cls, value: 'DashboardDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DashboardDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DashboardDict': + return _transform_to_json_value(self) # type:ignore + class DashboardDict(TypedDict, total=False): diff --git a/python/databricks/bundles/dashboards/_models/lifecycle.py b/python/databricks/bundles/dashboards/_models/lifecycle.py index 697776a198e..60ce33ace0e 100644 --- a/python/databricks/bundles/dashboards/_models/lifecycle.py +++ b/python/databricks/bundles/dashboards/_models/lifecycle.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam +from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam + +from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +29,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/dashboards/_models/lifecycle_state.py b/python/databricks/bundles/dashboards/_models/lifecycle_state.py index 15654e77a1f..ae4590aead7 100644 --- a/python/databricks/bundles/dashboards/_models/lifecycle_state.py +++ b/python/databricks/bundles/dashboards/_models/lifecycle_state.py @@ -1,12 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam +from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam class LifecycleState(Enum): ACTIVE = "ACTIVE" TRASHED = "TRASHED" - LifecycleStateParam = Literal["ACTIVE", "TRASHED"] | LifecycleState diff --git a/python/databricks/bundles/dashboards/_models/permission.py b/python/databricks/bundles/dashboards/_models/permission.py index fb4ea994f6d..9b05c19befb 100644 --- a/python/databricks/bundles/dashboards/_models/permission.py +++ b/python/databricks/bundles/dashboards/_models/permission.py @@ -1,20 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.dashboards._models.permission_level import ( - PermissionLevel, - PermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam +from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam + +from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Permission: """""" @@ -40,11 +44,12 @@ class Permission: """ @classmethod - def from_dict(cls, value: "PermissionDict") -> "Self": + def from_dict(cls, value: 'PermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PermissionDict': + return _transform_to_json_value(self) # type:ignore + class PermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/dashboards/_models/permission_level.py b/python/databricks/bundles/dashboards/_models/permission_level.py index c6111911b29..0ba31996221 100644 --- a/python/databricks/bundles/dashboards/_models/permission_level.py +++ b/python/databricks/bundles/dashboards/_models/permission_level.py @@ -1,14 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam +from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam class PermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" @@ -30,29 +40,4 @@ class PermissionLevel(Enum): CAN_MONITOR_ONLY = "CAN_MONITOR_ONLY" CAN_CREATE_APP = "CAN_CREATE_APP" - -PermissionLevelParam = ( - Literal[ - "CAN_MANAGE", - "CAN_RESTART", - "CAN_ATTACH_TO", - "IS_OWNER", - "CAN_MANAGE_RUN", - "CAN_VIEW", - "CAN_READ", - "CAN_RUN", - "CAN_EDIT", - "CAN_USE", - "CAN_MANAGE_STAGING_VERSIONS", - "CAN_MANAGE_PRODUCTION_VERSIONS", - "CAN_EDIT_METADATA", - "CAN_VIEW_METADATA", - "CAN_BIND", - "CAN_QUERY", - "CAN_MONITOR", - "CAN_CREATE", - "CAN_MONITOR_ONLY", - "CAN_CREATE_APP", - ] - | PermissionLevel -) +PermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW", "CAN_READ", "CAN_RUN", "CAN_EDIT", "CAN_USE", "CAN_MANAGE_STAGING_VERSIONS", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_EDIT_METADATA", "CAN_VIEW_METADATA", "CAN_BIND", "CAN_QUERY", "CAN_MONITOR", "CAN_CREATE", "CAN_MONITOR_ONLY", "CAN_CREATE_APP"] | PermissionLevel diff --git a/python/databricks/bundles/database_catalogs/__init__.py b/python/databricks/bundles/database_catalogs/__init__.py index fdd1d167302..e81e527d6c9 100644 --- a/python/databricks/bundles/database_catalogs/__init__.py +++ b/python/databricks/bundles/database_catalogs/__init__.py @@ -1,22 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "DatabaseCatalog", - "DatabaseCatalogDict", - "DatabaseCatalogParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", + 'DatabaseCatalog', + 'DatabaseCatalogDict', + 'DatabaseCatalogParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', ] -from databricks.bundles.database_catalogs._models.database_catalog import ( - DatabaseCatalog, - DatabaseCatalogDict, - DatabaseCatalogParam, -) -from databricks.bundles.database_catalogs._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) +from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog, DatabaseCatalogDict, DatabaseCatalogParam +from databricks.bundles.database_catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam diff --git a/python/databricks/bundles/database_catalogs/_models/database_catalog.py b/python/databricks/bundles/database_catalogs/_models/database_catalog.py index a224ac7a316..b943d184c64 100644 --- a/python/databricks/bundles/database_catalogs/_models/database_catalog.py +++ b/python/databricks/bundles/database_catalogs/_models/database_catalog.py @@ -1,21 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.database_catalogs._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.database_catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DatabaseCatalog(Resource): """""" @@ -46,11 +44,12 @@ class DatabaseCatalog(Resource): """ @classmethod - def from_dict(cls, value: "DatabaseCatalogDict") -> "Self": + def from_dict(cls, value: 'DatabaseCatalogDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DatabaseCatalogDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DatabaseCatalogDict': + return _transform_to_json_value(self) # type:ignore + class DatabaseCatalogDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_catalogs/_models/lifecycle.py b/python/databricks/bundles/database_catalogs/_models/lifecycle.py index 697776a198e..f010c457f95 100644 --- a/python/databricks/bundles/database_catalogs/_models/lifecycle.py +++ b/python/databricks/bundles/database_catalogs/_models/lifecycle.py @@ -1,16 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog, DatabaseCatalogDict, DatabaseCatalogParam if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +24,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/__init__.py b/python/databricks/bundles/database_instances/__init__.py index 13674b646a5..1b865f0ac01 100644 --- a/python/databricks/bundles/database_instances/__init__.py +++ b/python/databricks/bundles/database_instances/__init__.py @@ -1,52 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "CustomTag", - "CustomTagDict", - "CustomTagParam", - "DatabaseInstance", - "DatabaseInstanceDict", - "DatabaseInstanceParam", - "DatabaseInstanceRef", - "DatabaseInstanceRefDict", - "DatabaseInstanceRefParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "Permission", - "PermissionDict", - "PermissionLevel", - "PermissionLevelParam", - "PermissionParam", + 'CustomTag', + 'CustomTagDict', + 'CustomTagParam', + 'DatabaseInstance', + 'DatabaseInstanceDict', + 'DatabaseInstanceParam', + 'DatabaseInstanceRef', + 'DatabaseInstanceRefDict', + 'DatabaseInstanceRefParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'Permission', + 'PermissionDict', + 'PermissionLevel', + 'PermissionLevelParam', + 'PermissionParam', ] -from databricks.bundles.database_instances._models.custom_tag import ( - CustomTag, - CustomTagDict, - CustomTagParam, -) -from databricks.bundles.database_instances._models.database_instance import ( - DatabaseInstance, - DatabaseInstanceDict, - DatabaseInstanceParam, -) -from databricks.bundles.database_instances._models.database_instance_ref import ( - DatabaseInstanceRef, - DatabaseInstanceRefDict, - DatabaseInstanceRefParam, -) -from databricks.bundles.database_instances._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.database_instances._models.permission import ( - Permission, - PermissionDict, - PermissionParam, -) -from databricks.bundles.database_instances._models.permission_level import ( - PermissionLevel, - PermissionLevelParam, -) +from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam +from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam +from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam +from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam + diff --git a/python/databricks/bundles/database_instances/_models/custom_tag.py b/python/databricks/bundles/database_instances/_models/custom_tag.py index e08eb459bf7..583b3321ad3 100644 --- a/python/databricks/bundles/database_instances/_models/custom_tag.py +++ b/python/databricks/bundles/database_instances/_models/custom_tag.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam +from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam +from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class CustomTag: """""" @@ -30,11 +38,12 @@ class CustomTag: """ @classmethod - def from_dict(cls, value: "CustomTagDict") -> "Self": + def from_dict(cls, value: 'CustomTagDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CustomTagDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CustomTagDict': + return _transform_to_json_value(self) # type:ignore + class CustomTagDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/database_instance.py b/python/databricks/bundles/database_instances/_models/database_instance.py index beea60bfd1f..e0bdd84dc47 100644 --- a/python/databricks/bundles/database_instances/_models/database_instance.py +++ b/python/databricks/bundles/database_instances/_models/database_instance.py @@ -1,43 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.database_instances._models.custom_tag import ( - CustomTag, - CustomTagParam, -) -from databricks.bundles.database_instances._models.database_instance_ref import ( - DatabaseInstanceRef, - DatabaseInstanceRefParam, -) -from databricks.bundles.database_instances._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.database_instances._models.permission import ( - Permission, - PermissionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam +from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam +from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DatabaseInstance(Resource): """ A DatabaseInstance represents a logical Postgres instance, comprised of both compute and storage. """ - name: VariableOr[str] """ [Public Preview] The name of the instance. This is the unique identifier for the instance. @@ -109,11 +95,12 @@ class DatabaseInstance(Resource): """ @classmethod - def from_dict(cls, value: "DatabaseInstanceDict") -> "Self": + def from_dict(cls, value: 'DatabaseInstanceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DatabaseInstanceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DatabaseInstanceDict': + return _transform_to_json_value(self) # type:ignore + class DatabaseInstanceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/database_instance_ref.py b/python/databricks/bundles/database_instances/_models/database_instance_ref.py index 8c3e582f280..13663dd4a6a 100644 --- a/python/databricks/bundles/database_instances/_models/database_instance_ref.py +++ b/python/databricks/bundles/database_instances/_models/database_instance_ref.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam +from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam +from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DatabaseInstanceRef: """ @@ -23,7 +31,6 @@ class DatabaseInstanceRef: If both lsn and branch_time are specified, the lsn will be used to create the child instance. """ - branch_time: VariableOrOptional[str] = None """ [Public Preview] Branch time of the ref database instance. @@ -49,11 +56,12 @@ class DatabaseInstanceRef: """ @classmethod - def from_dict(cls, value: "DatabaseInstanceRefDict") -> "Self": + def from_dict(cls, value: 'DatabaseInstanceRefDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DatabaseInstanceRefDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DatabaseInstanceRefDict': + return _transform_to_json_value(self) # type:ignore + class DatabaseInstanceRefDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/lifecycle.py b/python/databricks/bundles/database_instances/_models/lifecycle.py index 697776a198e..9eda940b47c 100644 --- a/python/databricks/bundles/database_instances/_models/lifecycle.py +++ b/python/databricks/bundles/database_instances/_models/lifecycle.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam +from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam +from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam +from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +29,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/permission.py b/python/databricks/bundles/database_instances/_models/permission.py index da0f1e44ad2..b09522188ca 100644 --- a/python/databricks/bundles/database_instances/_models/permission.py +++ b/python/databricks/bundles/database_instances/_models/permission.py @@ -1,20 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.database_instances._models.permission_level import ( - PermissionLevel, - PermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam +from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam +from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam +from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Permission: """""" @@ -40,11 +44,12 @@ class Permission: """ @classmethod - def from_dict(cls, value: "PermissionDict") -> "Self": + def from_dict(cls, value: 'PermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PermissionDict': + return _transform_to_json_value(self) # type:ignore + class PermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/permission_level.py b/python/databricks/bundles/database_instances/_models/permission_level.py index c6111911b29..8295a3097cb 100644 --- a/python/databricks/bundles/database_instances/_models/permission_level.py +++ b/python/databricks/bundles/database_instances/_models/permission_level.py @@ -1,14 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam +from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam +from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam +from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam class PermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" @@ -30,29 +40,4 @@ class PermissionLevel(Enum): CAN_MONITOR_ONLY = "CAN_MONITOR_ONLY" CAN_CREATE_APP = "CAN_CREATE_APP" - -PermissionLevelParam = ( - Literal[ - "CAN_MANAGE", - "CAN_RESTART", - "CAN_ATTACH_TO", - "IS_OWNER", - "CAN_MANAGE_RUN", - "CAN_VIEW", - "CAN_READ", - "CAN_RUN", - "CAN_EDIT", - "CAN_USE", - "CAN_MANAGE_STAGING_VERSIONS", - "CAN_MANAGE_PRODUCTION_VERSIONS", - "CAN_EDIT_METADATA", - "CAN_VIEW_METADATA", - "CAN_BIND", - "CAN_QUERY", - "CAN_MONITOR", - "CAN_CREATE", - "CAN_MONITOR_ONLY", - "CAN_CREATE_APP", - ] - | PermissionLevel -) +PermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW", "CAN_READ", "CAN_RUN", "CAN_EDIT", "CAN_USE", "CAN_MANAGE_STAGING_VERSIONS", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_EDIT_METADATA", "CAN_VIEW_METADATA", "CAN_BIND", "CAN_QUERY", "CAN_MONITOR", "CAN_CREATE", "CAN_MONITOR_ONLY", "CAN_CREATE_APP"] | PermissionLevel diff --git a/python/databricks/bundles/experiments/__init__.py b/python/databricks/bundles/experiments/__init__.py index e1ae26cc521..9a4be4fc294 100644 --- a/python/databricks/bundles/experiments/__init__.py +++ b/python/databricks/bundles/experiments/__init__.py @@ -1,60 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "ExperimentPermissionLevel", - "ExperimentPermissionLevelParam", - "ExperimentTag", - "ExperimentTagDict", - "ExperimentTagParam", - "ExperimentTraceLocation", - "ExperimentTraceLocationDict", - "ExperimentTraceLocationParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "MlflowExperiment", - "MlflowExperimentDict", - "MlflowExperimentParam", - "MlflowExperimentPermission", - "MlflowExperimentPermissionDict", - "MlflowExperimentPermissionParam", - "UcTraceLocation", - "UcTraceLocationDict", - "UcTraceLocationParam", + 'ExperimentPermissionLevel', + 'ExperimentPermissionLevelParam', + 'ExperimentTag', + 'ExperimentTagDict', + 'ExperimentTagParam', + 'ExperimentTraceLocation', + 'ExperimentTraceLocationDict', + 'ExperimentTraceLocationParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'MlflowExperiment', + 'MlflowExperimentDict', + 'MlflowExperimentParam', + 'MlflowExperimentPermission', + 'MlflowExperimentPermissionDict', + 'MlflowExperimentPermissionParam', + 'UcTraceLocation', + 'UcTraceLocationDict', + 'UcTraceLocationParam', ] -from databricks.bundles.experiments._models.experiment_permission_level import ( - ExperimentPermissionLevel, - ExperimentPermissionLevelParam, -) -from databricks.bundles.experiments._models.experiment_tag import ( - ExperimentTag, - ExperimentTagDict, - ExperimentTagParam, -) -from databricks.bundles.experiments._models.experiment_trace_location import ( - ExperimentTraceLocation, - ExperimentTraceLocationDict, - ExperimentTraceLocationParam, -) -from databricks.bundles.experiments._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.experiments._models.mlflow_experiment import ( - MlflowExperiment, - MlflowExperimentDict, - MlflowExperimentParam, -) -from databricks.bundles.experiments._models.mlflow_experiment_permission import ( - MlflowExperimentPermission, - MlflowExperimentPermissionDict, - MlflowExperimentPermissionParam, -) -from databricks.bundles.experiments._models.uc_trace_location import ( - UcTraceLocation, - UcTraceLocationDict, - UcTraceLocationParam, -) +from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam +from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam +from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam +from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam +from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam +from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam + diff --git a/python/databricks/bundles/experiments/_models/experiment_permission_level.py b/python/databricks/bundles/experiments/_models/experiment_permission_level.py index 056727ac92c..275cee3f610 100644 --- a/python/databricks/bundles/experiments/_models/experiment_permission_level.py +++ b/python/databricks/bundles/experiments/_models/experiment_permission_level.py @@ -1,19 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam +from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam +from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam +from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam +from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam class ExperimentPermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_EDIT = "CAN_EDIT" CAN_READ = "CAN_READ" - -ExperimentPermissionLevelParam = ( - Literal["CAN_MANAGE", "CAN_EDIT", "CAN_READ"] | ExperimentPermissionLevel -) +ExperimentPermissionLevelParam = Literal["CAN_MANAGE", "CAN_EDIT", "CAN_READ"] | ExperimentPermissionLevel diff --git a/python/databricks/bundles/experiments/_models/experiment_tag.py b/python/databricks/bundles/experiments/_models/experiment_tag.py index 2c78a181b17..8f96768c572 100644 --- a/python/databricks/bundles/experiments/_models/experiment_tag.py +++ b/python/databricks/bundles/experiments/_models/experiment_tag.py @@ -1,22 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam +from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam +from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam +from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam +from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ExperimentTag: """ A tag for an experiment. """ - key: VariableOrOptional[str] = None """ The tag key. @@ -28,11 +36,12 @@ class ExperimentTag: """ @classmethod - def from_dict(cls, value: "ExperimentTagDict") -> "Self": + def from_dict(cls, value: 'ExperimentTagDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ExperimentTagDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ExperimentTagDict': + return _transform_to_json_value(self) # type:ignore + class ExperimentTagDict(TypedDict, total=False): diff --git a/python/databricks/bundles/experiments/_models/experiment_trace_location.py b/python/databricks/bundles/experiments/_models/experiment_trace_location.py index cbc9dffb6cf..7585d164f35 100644 --- a/python/databricks/bundles/experiments/_models/experiment_trace_location.py +++ b/python/databricks/bundles/experiments/_models/experiment_trace_location.py @@ -1,28 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.experiments._models.uc_trace_location import ( - UcTraceLocation, - UcTraceLocationParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam +from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam +from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam +from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam +from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ExperimentTraceLocation: """ :meta private: [EXPERIMENTAL] - + The storage location for an experiment's traces. """ - uc_trace_location: VariableOrOptional[UcTraceLocation] = None """ :meta private: [EXPERIMENTAL] @@ -32,11 +36,12 @@ class ExperimentTraceLocation: """ @classmethod - def from_dict(cls, value: "ExperimentTraceLocationDict") -> "Self": + def from_dict(cls, value: 'ExperimentTraceLocationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ExperimentTraceLocationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ExperimentTraceLocationDict': + return _transform_to_json_value(self) # type:ignore + class ExperimentTraceLocationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/experiments/_models/lifecycle.py b/python/databricks/bundles/experiments/_models/lifecycle.py index 697776a198e..520a7ac7b3d 100644 --- a/python/databricks/bundles/experiments/_models/lifecycle.py +++ b/python/databricks/bundles/experiments/_models/lifecycle.py @@ -1,16 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam +from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam +from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam +from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam +from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam +from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +30,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/experiments/_models/mlflow_experiment.py b/python/databricks/bundles/experiments/_models/mlflow_experiment.py index 8a9d5368e99..be5d319936d 100644 --- a/python/databricks/bundles/experiments/_models/mlflow_experiment.py +++ b/python/databricks/bundles/experiments/_models/mlflow_experiment.py @@ -1,37 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.experiments._models.experiment_tag import ( - ExperimentTag, - ExperimentTagParam, -) -from databricks.bundles.experiments._models.experiment_trace_location import ( - ExperimentTraceLocation, - ExperimentTraceLocationParam, -) -from databricks.bundles.experiments._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.experiments._models.mlflow_experiment_permission import ( - MlflowExperimentPermission, - MlflowExperimentPermissionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam +from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam +from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam +from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam +from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MlflowExperiment(Resource): """""" @@ -52,9 +40,7 @@ class MlflowExperiment(Resource): Settings that control the deployment lifecycle of the resource, such as preventing it from being destroyed. """ - permissions: VariableOrList[MlflowExperimentPermission] = field( - default_factory=list - ) + permissions: VariableOrList[MlflowExperimentPermission] = field(default_factory=list) """ The permissions to apply to this resource. """ @@ -78,11 +64,12 @@ class MlflowExperiment(Resource): """ @classmethod - def from_dict(cls, value: "MlflowExperimentDict") -> "Self": + def from_dict(cls, value: 'MlflowExperimentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MlflowExperimentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MlflowExperimentDict': + return _transform_to_json_value(self) # type:ignore + class MlflowExperimentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/experiments/_models/mlflow_experiment_permission.py b/python/databricks/bundles/experiments/_models/mlflow_experiment_permission.py index 041b02d91b3..fbaa465cccf 100644 --- a/python/databricks/bundles/experiments/_models/mlflow_experiment_permission.py +++ b/python/databricks/bundles/experiments/_models/mlflow_experiment_permission.py @@ -1,20 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.experiments._models.experiment_permission_level import ( - ExperimentPermissionLevel, - ExperimentPermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam +from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam +from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam +from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam +from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MlflowExperimentPermission: """""" @@ -40,11 +45,12 @@ class MlflowExperimentPermission: """ @classmethod - def from_dict(cls, value: "MlflowExperimentPermissionDict") -> "Self": + def from_dict(cls, value: 'MlflowExperimentPermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MlflowExperimentPermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MlflowExperimentPermissionDict': + return _transform_to_json_value(self) # type:ignore + class MlflowExperimentPermissionDict(TypedDict, total=False): @@ -71,6 +77,4 @@ class MlflowExperimentPermissionDict(TypedDict, total=False): """ -MlflowExperimentPermissionParam = ( - MlflowExperimentPermissionDict | MlflowExperimentPermission -) +MlflowExperimentPermissionParam = MlflowExperimentPermissionDict | MlflowExperimentPermission diff --git a/python/databricks/bundles/experiments/_models/uc_trace_location.py b/python/databricks/bundles/experiments/_models/uc_trace_location.py index b7b8148823e..8abb137df3e 100644 --- a/python/databricks/bundles/experiments/_models/uc_trace_location.py +++ b/python/databricks/bundles/experiments/_models/uc_trace_location.py @@ -1,25 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam +from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam +from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam +from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam +from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class UcTraceLocation: """ :meta private: [EXPERIMENTAL] - + A Unity Catalog trace storage location. Traces are stored as Delta tables in the specified catalog and schema. """ - catalog: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -47,11 +55,12 @@ class UcTraceLocation: """ @classmethod - def from_dict(cls, value: "UcTraceLocationDict") -> "Self": + def from_dict(cls, value: 'UcTraceLocationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "UcTraceLocationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'UcTraceLocationDict': + return _transform_to_json_value(self) # type:ignore + class UcTraceLocationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/__init__.py b/python/databricks/bundles/external_locations/__init__.py index d6aa3bf908c..7a69b1d9836 100644 --- a/python/databricks/bundles/external_locations/__init__.py +++ b/python/databricks/bundles/external_locations/__init__.py @@ -1,90 +1,50 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "AwsSqsQueue", - "AwsSqsQueueDict", - "AwsSqsQueueParam", - "AzureQueueStorage", - "AzureQueueStorageDict", - "AzureQueueStorageParam", - "EncryptionDetails", - "EncryptionDetailsDict", - "EncryptionDetailsParam", - "ExternalLocation", - "ExternalLocationDict", - "ExternalLocationParam", - "FileEventQueue", - "FileEventQueueDict", - "FileEventQueueParam", - "GcpPubsub", - "GcpPubsubDict", - "GcpPubsubParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", - "SseEncryptionDetails", - "SseEncryptionDetailsAlgorithm", - "SseEncryptionDetailsAlgorithmParam", - "SseEncryptionDetailsDict", - "SseEncryptionDetailsParam", + 'AwsSqsQueue', + 'AwsSqsQueueDict', + 'AwsSqsQueueParam', + 'AzureQueueStorage', + 'AzureQueueStorageDict', + 'AzureQueueStorageParam', + 'EncryptionDetails', + 'EncryptionDetailsDict', + 'EncryptionDetailsParam', + 'ExternalLocation', + 'ExternalLocationDict', + 'ExternalLocationParam', + 'FileEventQueue', + 'FileEventQueueDict', + 'FileEventQueueParam', + 'GcpPubsub', + 'GcpPubsubDict', + 'GcpPubsubParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'SseEncryptionDetails', + 'SseEncryptionDetailsAlgorithm', + 'SseEncryptionDetailsAlgorithmParam', + 'SseEncryptionDetailsDict', + 'SseEncryptionDetailsParam', ] -from databricks.bundles.external_locations._models.aws_sqs_queue import ( - AwsSqsQueue, - AwsSqsQueueDict, - AwsSqsQueueParam, -) -from databricks.bundles.external_locations._models.azure_queue_storage import ( - AzureQueueStorage, - AzureQueueStorageDict, - AzureQueueStorageParam, -) -from databricks.bundles.external_locations._models.encryption_details import ( - EncryptionDetails, - EncryptionDetailsDict, - EncryptionDetailsParam, -) -from databricks.bundles.external_locations._models.external_location import ( - ExternalLocation, - ExternalLocationDict, - ExternalLocationParam, -) -from databricks.bundles.external_locations._models.file_event_queue import ( - FileEventQueue, - FileEventQueueDict, - FileEventQueueParam, -) -from databricks.bundles.external_locations._models.gcp_pubsub import ( - GcpPubsub, - GcpPubsubDict, - GcpPubsubParam, -) -from databricks.bundles.external_locations._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.external_locations._models.privilege import ( - Privilege, - PrivilegeParam, -) -from databricks.bundles.external_locations._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) -from databricks.bundles.external_locations._models.sse_encryption_details import ( - SseEncryptionDetails, - SseEncryptionDetailsDict, - SseEncryptionDetailsParam, -) -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import ( - SseEncryptionDetailsAlgorithm, - SseEncryptionDetailsAlgorithmParam, -) +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + diff --git a/python/databricks/bundles/external_locations/_models/aws_sqs_queue.py b/python/databricks/bundles/external_locations/_models/aws_sqs_queue.py index 53c7a0b7062..437490dad63 100644 --- a/python/databricks/bundles/external_locations/_models/aws_sqs_queue.py +++ b/python/databricks/bundles/external_locations/_models/aws_sqs_queue.py @@ -1,16 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AwsSqsQueue: """""" @@ -22,11 +36,12 @@ class AwsSqsQueue: """ @classmethod - def from_dict(cls, value: "AwsSqsQueueDict") -> "Self": + def from_dict(cls, value: 'AwsSqsQueueDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AwsSqsQueueDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AwsSqsQueueDict': + return _transform_to_json_value(self) # type:ignore + class AwsSqsQueueDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/azure_queue_storage.py b/python/databricks/bundles/external_locations/_models/azure_queue_storage.py index a8d182421f7..3357af42478 100644 --- a/python/databricks/bundles/external_locations/_models/azure_queue_storage.py +++ b/python/databricks/bundles/external_locations/_models/azure_queue_storage.py @@ -1,16 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AzureQueueStorage: """""" @@ -36,11 +50,12 @@ class AzureQueueStorage: """ @classmethod - def from_dict(cls, value: "AzureQueueStorageDict") -> "Self": + def from_dict(cls, value: 'AzureQueueStorageDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AzureQueueStorageDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AzureQueueStorageDict': + return _transform_to_json_value(self) # type:ignore + class AzureQueueStorageDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/encryption_details.py b/python/databricks/bundles/external_locations/_models/encryption_details.py index d1b67fa76bd..5d0df12237e 100644 --- a/python/databricks/bundles/external_locations/_models/encryption_details.py +++ b/python/databricks/bundles/external_locations/_models/encryption_details.py @@ -1,37 +1,47 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.external_locations._models.sse_encryption_details import ( - SseEncryptionDetails, - SseEncryptionDetailsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EncryptionDetails: """ Encryption options that apply to clients connecting to cloud storage. """ - sse_encryption_details: VariableOrOptional[SseEncryptionDetails] = None """ Server-Side Encryption properties for clients communicating with AWS s3. """ @classmethod - def from_dict(cls, value: "EncryptionDetailsDict") -> "Self": + def from_dict(cls, value: 'EncryptionDetailsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EncryptionDetailsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EncryptionDetailsDict': + return _transform_to_json_value(self) # type:ignore + class EncryptionDetailsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/external_location.py b/python/databricks/bundles/external_locations/_models/external_location.py index ea72a5da3dd..9e0c99840a6 100644 --- a/python/databricks/bundles/external_locations/_models/external_location.py +++ b/python/databricks/bundles/external_locations/_models/external_location.py @@ -1,37 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.external_locations._models.encryption_details import ( - EncryptionDetails, - EncryptionDetailsParam, -) -from databricks.bundles.external_locations._models.file_event_queue import ( - FileEventQueue, - FileEventQueueParam, -) -from databricks.bundles.external_locations._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.external_locations._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ExternalLocation(Resource): """""" @@ -98,11 +91,12 @@ class ExternalLocation(Resource): """ @classmethod - def from_dict(cls, value: "ExternalLocationDict") -> "Self": + def from_dict(cls, value: 'ExternalLocationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ExternalLocationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ExternalLocationDict': + return _transform_to_json_value(self) # type:ignore + class ExternalLocationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/file_event_queue.py b/python/databricks/bundles/external_locations/_models/file_event_queue.py index 78a791766cd..bb224e885bc 100644 --- a/python/databricks/bundles/external_locations/_models/file_event_queue.py +++ b/python/databricks/bundles/external_locations/_models/file_event_queue.py @@ -1,28 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.external_locations._models.aws_sqs_queue import ( - AwsSqsQueue, - AwsSqsQueueParam, -) -from databricks.bundles.external_locations._models.azure_queue_storage import ( - AzureQueueStorage, - AzureQueueStorageParam, -) -from databricks.bundles.external_locations._models.gcp_pubsub import ( - GcpPubsub, - GcpPubsubParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FileEventQueue: """""" @@ -40,11 +42,12 @@ class FileEventQueue: provided_sqs: VariableOrOptional[AwsSqsQueue] = None @classmethod - def from_dict(cls, value: "FileEventQueueDict") -> "Self": + def from_dict(cls, value: 'FileEventQueueDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FileEventQueueDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FileEventQueueDict': + return _transform_to_json_value(self) # type:ignore + class FileEventQueueDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/gcp_pubsub.py b/python/databricks/bundles/external_locations/_models/gcp_pubsub.py index 926b27d2a55..ac2055c98d6 100644 --- a/python/databricks/bundles/external_locations/_models/gcp_pubsub.py +++ b/python/databricks/bundles/external_locations/_models/gcp_pubsub.py @@ -1,16 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GcpPubsub: """""" @@ -22,11 +36,12 @@ class GcpPubsub: """ @classmethod - def from_dict(cls, value: "GcpPubsubDict") -> "Self": + def from_dict(cls, value: 'GcpPubsubDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GcpPubsubDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GcpPubsubDict': + return _transform_to_json_value(self) # type:ignore + class GcpPubsubDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/lifecycle.py b/python/databricks/bundles/external_locations/_models/lifecycle.py index 697776a198e..842d400f831 100644 --- a/python/databricks/bundles/external_locations/_models/lifecycle.py +++ b/python/databricks/bundles/external_locations/_models/lifecycle.py @@ -1,16 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +35,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/privilege.py b/python/databricks/bundles/external_locations/_models/privilege.py index df45211a76a..41cf98001ae 100644 --- a/python/databricks/bundles/external_locations/_models/privilege.py +++ b/python/databricks/bundles/external_locations/_models/privilege.py @@ -1,7 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam class Privilege(Enum): @@ -58,61 +75,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/external_locations/_models/privilege_assignment.py b/python/databricks/bundles/external_locations/_models/privilege_assignment.py index f48f9511ac0..f6b498fea1e 100644 --- a/python/databricks/bundles/external_locations/_models/privilege_assignment.py +++ b/python/databricks/bundles/external_locations/_models/privilege_assignment.py @@ -1,20 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.external_locations._models.privilege import ( - Privilege, - PrivilegeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -31,11 +41,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/sse_encryption_details.py b/python/databricks/bundles/external_locations/_models/sse_encryption_details.py index 339b98b8b26..46c78a5774c 100644 --- a/python/databricks/bundles/external_locations/_models/sse_encryption_details.py +++ b/python/databricks/bundles/external_locations/_models/sse_encryption_details.py @@ -1,26 +1,35 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import ( - SseEncryptionDetailsAlgorithm, - SseEncryptionDetailsAlgorithmParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SseEncryptionDetails: """ Server-Side Encryption properties for clients communicating with AWS s3. """ - algorithm: VariableOrOptional[SseEncryptionDetailsAlgorithm] = None """ Sets the value of the 'x-amz-server-side-encryption' header in S3 request. @@ -33,11 +42,12 @@ class SseEncryptionDetails: """ @classmethod - def from_dict(cls, value: "SseEncryptionDetailsDict") -> "Self": + def from_dict(cls, value: 'SseEncryptionDetailsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SseEncryptionDetailsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SseEncryptionDetailsDict': + return _transform_to_json_value(self) # type:ignore + class SseEncryptionDetailsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/sse_encryption_details_algorithm.py b/python/databricks/bundles/external_locations/_models/sse_encryption_details_algorithm.py index d5d63383a74..9bc52f98294 100644 --- a/python/databricks/bundles/external_locations/_models/sse_encryption_details_algorithm.py +++ b/python/databricks/bundles/external_locations/_models/sse_encryption_details_algorithm.py @@ -1,18 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam +from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam +from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam +from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam +from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam class SseEncryptionDetailsAlgorithm(Enum): """ SSE algorithm to use for encrypting S3 objects """ - AWS_SSE_S3 = "AWS_SSE_S3" AWS_SSE_KMS = "AWS_SSE_KMS" - -SseEncryptionDetailsAlgorithmParam = ( - Literal["AWS_SSE_S3", "AWS_SSE_KMS"] | SseEncryptionDetailsAlgorithm -) +SseEncryptionDetailsAlgorithmParam = Literal["AWS_SSE_S3", "AWS_SSE_KMS"] | SseEncryptionDetailsAlgorithm diff --git a/python/databricks/bundles/features/__init__.py b/python/databricks/bundles/features/__init__.py index 24e3f143d65..e8eae18922b 100644 --- a/python/databricks/bundles/features/__init__.py +++ b/python/databricks/bundles/features/__init__.py @@ -1,356 +1,186 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "AggregationFunction", - "AggregationFunctionDict", - "AggregationFunctionParam", - "ApproxCountDistinctFunction", - "ApproxCountDistinctFunctionDict", - "ApproxCountDistinctFunctionParam", - "ApproxPercentileFunction", - "ApproxPercentileFunctionDict", - "ApproxPercentileFunctionParam", - "AvgFunction", - "AvgFunctionDict", - "AvgFunctionParam", - "ColumnSelection", - "ColumnSelectionDict", - "ColumnSelectionParam", - "CountFunction", - "CountFunctionDict", - "CountFunctionParam", - "CustomUdf", - "CustomUdfDict", - "CustomUdfParam", - "DataSource", - "DataSourceDict", - "DataSourceParam", - "DeltaTableSource", - "DeltaTableSourceDict", - "DeltaTableSourceParam", - "EntityColumn", - "EntityColumnDict", - "EntityColumnParam", - "Feature", - "FeatureDict", - "FeatureParam", - "FeatureReference", - "FeatureReferenceDict", - "FeatureReferenceParam", - "FeatureViewSource", - "FeatureViewSourceDict", - "FeatureViewSourceParam", - "FieldDefinition", - "FieldDefinitionDict", - "FieldDefinitionParam", - "FirstDistinctFunction", - "FirstDistinctFunctionDict", - "FirstDistinctFunctionParam", - "FirstFunction", - "FirstFunctionDict", - "FirstFunctionParam", - "FirstNFunction", - "FirstNFunctionDict", - "FirstNFunctionParam", - "FlatSchema", - "FlatSchemaDict", - "FlatSchemaParam", - "Function", - "FunctionDict", - "FunctionParam", - "InputBinding", - "InputBindingDict", - "InputBindingParam", - "JobContext", - "JobContextDict", - "JobContextParam", - "KafkaSource", - "KafkaSourceDict", - "KafkaSourceParam", - "LastDistinctFunction", - "LastDistinctFunctionDict", - "LastDistinctFunctionParam", - "LastFunction", - "LastFunctionDict", - "LastFunctionParam", - "LastNFunction", - "LastNFunctionDict", - "LastNFunctionParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "LineageContext", - "LineageContextDict", - "LineageContextParam", - "MaxFunction", - "MaxFunctionDict", - "MaxFunctionParam", - "MinFunction", - "MinFunctionDict", - "MinFunctionParam", - "RequestSource", - "RequestSourceDict", - "RequestSourceParam", - "RollingWindow", - "RollingWindowDict", - "RollingWindowParam", - "SawtoothWindow", - "SawtoothWindowDict", - "SawtoothWindowParam", - "ScalarDataType", - "ScalarDataTypeParam", - "SlidingWindow", - "SlidingWindowDict", - "SlidingWindowParam", - "SourceLateness", - "SourceLatenessDict", - "SourceLatenessParam", - "StddevPopFunction", - "StddevPopFunctionDict", - "StddevPopFunctionParam", - "StddevSampFunction", - "StddevSampFunctionDict", - "StddevSampFunctionParam", - "StreamSource", - "StreamSourceDict", - "StreamSourceParam", - "SumFunction", - "SumFunctionDict", - "SumFunctionParam", - "TimeWindow", - "TimeWindowDict", - "TimeWindowParam", - "TimeseriesColumn", - "TimeseriesColumnDict", - "TimeseriesColumnParam", - "TumblingWindow", - "TumblingWindowDict", - "TumblingWindowParam", - "VarPopFunction", - "VarPopFunctionDict", - "VarPopFunctionParam", - "VarSampFunction", - "VarSampFunctionDict", - "VarSampFunctionParam", + 'AggregationFunction', + 'AggregationFunctionDict', + 'AggregationFunctionParam', + 'ApproxCountDistinctFunction', + 'ApproxCountDistinctFunctionDict', + 'ApproxCountDistinctFunctionParam', + 'ApproxPercentileFunction', + 'ApproxPercentileFunctionDict', + 'ApproxPercentileFunctionParam', + 'AvgFunction', + 'AvgFunctionDict', + 'AvgFunctionParam', + 'ColumnSelection', + 'ColumnSelectionDict', + 'ColumnSelectionParam', + 'CountFunction', + 'CountFunctionDict', + 'CountFunctionParam', + 'CustomUdf', + 'CustomUdfDict', + 'CustomUdfParam', + 'DataSource', + 'DataSourceDict', + 'DataSourceParam', + 'DeltaTableSource', + 'DeltaTableSourceDict', + 'DeltaTableSourceParam', + 'EntityColumn', + 'EntityColumnDict', + 'EntityColumnParam', + 'Feature', + 'FeatureDict', + 'FeatureParam', + 'FeatureReference', + 'FeatureReferenceDict', + 'FeatureReferenceParam', + 'FeatureViewSource', + 'FeatureViewSourceDict', + 'FeatureViewSourceParam', + 'FieldDefinition', + 'FieldDefinitionDict', + 'FieldDefinitionParam', + 'FirstDistinctFunction', + 'FirstDistinctFunctionDict', + 'FirstDistinctFunctionParam', + 'FirstFunction', + 'FirstFunctionDict', + 'FirstFunctionParam', + 'FirstNFunction', + 'FirstNFunctionDict', + 'FirstNFunctionParam', + 'FlatSchema', + 'FlatSchemaDict', + 'FlatSchemaParam', + 'Function', + 'FunctionDict', + 'FunctionParam', + 'InputBinding', + 'InputBindingDict', + 'InputBindingParam', + 'JobContext', + 'JobContextDict', + 'JobContextParam', + 'KafkaSource', + 'KafkaSourceDict', + 'KafkaSourceParam', + 'LastDistinctFunction', + 'LastDistinctFunctionDict', + 'LastDistinctFunctionParam', + 'LastFunction', + 'LastFunctionDict', + 'LastFunctionParam', + 'LastNFunction', + 'LastNFunctionDict', + 'LastNFunctionParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'LineageContext', + 'LineageContextDict', + 'LineageContextParam', + 'MaxFunction', + 'MaxFunctionDict', + 'MaxFunctionParam', + 'MinFunction', + 'MinFunctionDict', + 'MinFunctionParam', + 'RequestSource', + 'RequestSourceDict', + 'RequestSourceParam', + 'RollingWindow', + 'RollingWindowDict', + 'RollingWindowParam', + 'SawtoothWindow', + 'SawtoothWindowDict', + 'SawtoothWindowParam', + 'ScalarDataType', + 'ScalarDataTypeParam', + 'SdpSource', + 'SdpSourceDict', + 'SdpSourceParam', + 'SlidingWindow', + 'SlidingWindowDict', + 'SlidingWindowParam', + 'SourceLateness', + 'SourceLatenessDict', + 'SourceLatenessParam', + 'StddevPopFunction', + 'StddevPopFunctionDict', + 'StddevPopFunctionParam', + 'StddevSampFunction', + 'StddevSampFunctionDict', + 'StddevSampFunctionParam', + 'StreamSource', + 'StreamSourceDict', + 'StreamSourceParam', + 'SumFunction', + 'SumFunctionDict', + 'SumFunctionParam', + 'TimeWindow', + 'TimeWindowDict', + 'TimeWindowParam', + 'TimeseriesColumn', + 'TimeseriesColumnDict', + 'TimeseriesColumnParam', + 'TumblingWindow', + 'TumblingWindowDict', + 'TumblingWindowParam', + 'VarPopFunction', + 'VarPopFunctionDict', + 'VarPopFunctionParam', + 'VarSampFunction', + 'VarSampFunctionDict', + 'VarSampFunctionParam', ] -from databricks.bundles.features._models.aggregation_function import ( - AggregationFunction, - AggregationFunctionDict, - AggregationFunctionParam, -) -from databricks.bundles.features._models.approx_count_distinct_function import ( - ApproxCountDistinctFunction, - ApproxCountDistinctFunctionDict, - ApproxCountDistinctFunctionParam, -) -from databricks.bundles.features._models.approx_percentile_function import ( - ApproxPercentileFunction, - ApproxPercentileFunctionDict, - ApproxPercentileFunctionParam, -) -from databricks.bundles.features._models.avg_function import ( - AvgFunction, - AvgFunctionDict, - AvgFunctionParam, -) -from databricks.bundles.features._models.column_selection import ( - ColumnSelection, - ColumnSelectionDict, - ColumnSelectionParam, -) -from databricks.bundles.features._models.count_function import ( - CountFunction, - CountFunctionDict, - CountFunctionParam, -) -from databricks.bundles.features._models.custom_udf import ( - CustomUdf, - CustomUdfDict, - CustomUdfParam, -) -from databricks.bundles.features._models.data_source import ( - DataSource, - DataSourceDict, - DataSourceParam, -) -from databricks.bundles.features._models.delta_table_source import ( - DeltaTableSource, - DeltaTableSourceDict, - DeltaTableSourceParam, -) -from databricks.bundles.features._models.entity_column import ( - EntityColumn, - EntityColumnDict, - EntityColumnParam, -) -from databricks.bundles.features._models.feature import ( - Feature, - FeatureDict, - FeatureParam, -) -from databricks.bundles.features._models.feature_reference import ( - FeatureReference, - FeatureReferenceDict, - FeatureReferenceParam, -) -from databricks.bundles.features._models.feature_view_source import ( - FeatureViewSource, - FeatureViewSourceDict, - FeatureViewSourceParam, -) -from databricks.bundles.features._models.field_definition import ( - FieldDefinition, - FieldDefinitionDict, - FieldDefinitionParam, -) -from databricks.bundles.features._models.first_distinct_function import ( - FirstDistinctFunction, - FirstDistinctFunctionDict, - FirstDistinctFunctionParam, -) -from databricks.bundles.features._models.first_function import ( - FirstFunction, - FirstFunctionDict, - FirstFunctionParam, -) -from databricks.bundles.features._models.first_n_function import ( - FirstNFunction, - FirstNFunctionDict, - FirstNFunctionParam, -) -from databricks.bundles.features._models.flat_schema import ( - FlatSchema, - FlatSchemaDict, - FlatSchemaParam, -) -from databricks.bundles.features._models.function import ( - Function, - FunctionDict, - FunctionParam, -) -from databricks.bundles.features._models.input_binding import ( - InputBinding, - InputBindingDict, - InputBindingParam, -) -from databricks.bundles.features._models.job_context import ( - JobContext, - JobContextDict, - JobContextParam, -) -from databricks.bundles.features._models.kafka_source import ( - KafkaSource, - KafkaSourceDict, - KafkaSourceParam, -) -from databricks.bundles.features._models.last_distinct_function import ( - LastDistinctFunction, - LastDistinctFunctionDict, - LastDistinctFunctionParam, -) -from databricks.bundles.features._models.last_function import ( - LastFunction, - LastFunctionDict, - LastFunctionParam, -) -from databricks.bundles.features._models.last_n_function import ( - LastNFunction, - LastNFunctionDict, - LastNFunctionParam, -) -from databricks.bundles.features._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.features._models.lineage_context import ( - LineageContext, - LineageContextDict, - LineageContextParam, -) -from databricks.bundles.features._models.max_function import ( - MaxFunction, - MaxFunctionDict, - MaxFunctionParam, -) -from databricks.bundles.features._models.min_function import ( - MinFunction, - MinFunctionDict, - MinFunctionParam, -) -from databricks.bundles.features._models.request_source import ( - RequestSource, - RequestSourceDict, - RequestSourceParam, -) -from databricks.bundles.features._models.rolling_window import ( - RollingWindow, - RollingWindowDict, - RollingWindowParam, -) -from databricks.bundles.features._models.sawtooth_window import ( - SawtoothWindow, - SawtoothWindowDict, - SawtoothWindowParam, -) -from databricks.bundles.features._models.scalar_data_type import ( - ScalarDataType, - ScalarDataTypeParam, -) -from databricks.bundles.features._models.sliding_window import ( - SlidingWindow, - SlidingWindowDict, - SlidingWindowParam, -) -from databricks.bundles.features._models.source_lateness import ( - SourceLateness, - SourceLatenessDict, - SourceLatenessParam, -) -from databricks.bundles.features._models.stddev_pop_function import ( - StddevPopFunction, - StddevPopFunctionDict, - StddevPopFunctionParam, -) -from databricks.bundles.features._models.stddev_samp_function import ( - StddevSampFunction, - StddevSampFunctionDict, - StddevSampFunctionParam, -) -from databricks.bundles.features._models.stream_source import ( - StreamSource, - StreamSourceDict, - StreamSourceParam, -) -from databricks.bundles.features._models.sum_function import ( - SumFunction, - SumFunctionDict, - SumFunctionParam, -) -from databricks.bundles.features._models.time_window import ( - TimeWindow, - TimeWindowDict, - TimeWindowParam, -) -from databricks.bundles.features._models.timeseries_column import ( - TimeseriesColumn, - TimeseriesColumnDict, - TimeseriesColumnParam, -) -from databricks.bundles.features._models.tumbling_window import ( - TumblingWindow, - TumblingWindowDict, - TumblingWindowParam, -) -from databricks.bundles.features._models.var_pop_function import ( - VarPopFunction, - VarPopFunctionDict, - VarPopFunctionParam, -) -from databricks.bundles.features._models.var_samp_function import ( - VarSampFunction, - VarSampFunctionDict, - VarSampFunctionParam, -) +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + diff --git a/python/databricks/bundles/features/_models/aggregation_function.py b/python/databricks/bundles/features/_models/aggregation_function.py index 292f03909fa..fe593e4d8ff 100644 --- a/python/databricks/bundles/features/_models/aggregation_function.py +++ b/python/databricks/bundles/features/_models/aggregation_function.py @@ -1,91 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.features._models.approx_count_distinct_function import ( - ApproxCountDistinctFunction, - ApproxCountDistinctFunctionParam, -) -from databricks.bundles.features._models.approx_percentile_function import ( - ApproxPercentileFunction, - ApproxPercentileFunctionParam, -) -from databricks.bundles.features._models.avg_function import ( - AvgFunction, - AvgFunctionParam, -) -from databricks.bundles.features._models.count_function import ( - CountFunction, - CountFunctionParam, -) -from databricks.bundles.features._models.first_distinct_function import ( - FirstDistinctFunction, - FirstDistinctFunctionParam, -) -from databricks.bundles.features._models.first_function import ( - FirstFunction, - FirstFunctionParam, -) -from databricks.bundles.features._models.first_n_function import ( - FirstNFunction, - FirstNFunctionParam, -) -from databricks.bundles.features._models.last_distinct_function import ( - LastDistinctFunction, - LastDistinctFunctionParam, -) -from databricks.bundles.features._models.last_function import ( - LastFunction, - LastFunctionParam, -) -from databricks.bundles.features._models.last_n_function import ( - LastNFunction, - LastNFunctionParam, -) -from databricks.bundles.features._models.max_function import ( - MaxFunction, - MaxFunctionParam, -) -from databricks.bundles.features._models.min_function import ( - MinFunction, - MinFunctionParam, -) -from databricks.bundles.features._models.stddev_pop_function import ( - StddevPopFunction, - StddevPopFunctionParam, -) -from databricks.bundles.features._models.stddev_samp_function import ( - StddevSampFunction, - StddevSampFunctionParam, -) -from databricks.bundles.features._models.sum_function import ( - SumFunction, - SumFunctionParam, -) -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowParam -from databricks.bundles.features._models.var_pop_function import ( - VarPopFunction, - VarPopFunctionParam, -) -from databricks.bundles.features._models.var_samp_function import ( - VarSampFunction, - VarSampFunctionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AggregationFunction: """ An aggregation function applied over a time window. """ - approx_count_distinct: VariableOrOptional[ApproxCountDistinctFunction] = None """ Computes the approximate count of distinct values. @@ -185,11 +162,12 @@ class AggregationFunction: """ @classmethod - def from_dict(cls, value: "AggregationFunctionDict") -> "Self": + def from_dict(cls, value: 'AggregationFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AggregationFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AggregationFunctionDict': + return _transform_to_json_value(self) # type:ignore + class AggregationFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/approx_count_distinct_function.py b/python/databricks/bundles/features/_models/approx_count_distinct_function.py index 6244d5a431c..6eec220608e 100644 --- a/python/databricks/bundles/features/_models/approx_count_distinct_function.py +++ b/python/databricks/bundles/features/_models/approx_count_distinct_function.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ApproxCountDistinctFunction: """ Computes the approximate count of distinct values. """ - input: VariableOr[str] """ The input column from which the approximate count of distinct values is computed. @@ -28,11 +74,12 @@ class ApproxCountDistinctFunction: """ @classmethod - def from_dict(cls, value: "ApproxCountDistinctFunctionDict") -> "Self": + def from_dict(cls, value: 'ApproxCountDistinctFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ApproxCountDistinctFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ApproxCountDistinctFunctionDict': + return _transform_to_json_value(self) # type:ignore + class ApproxCountDistinctFunctionDict(TypedDict, total=False): @@ -49,6 +96,4 @@ class ApproxCountDistinctFunctionDict(TypedDict, total=False): """ -ApproxCountDistinctFunctionParam = ( - ApproxCountDistinctFunctionDict | ApproxCountDistinctFunction -) +ApproxCountDistinctFunctionParam = ApproxCountDistinctFunctionDict | ApproxCountDistinctFunction diff --git a/python/databricks/bundles/features/_models/approx_percentile_function.py b/python/databricks/bundles/features/_models/approx_percentile_function.py index b4648e41209..235bd72fbb0 100644 --- a/python/databricks/bundles/features/_models/approx_percentile_function.py +++ b/python/databricks/bundles/features/_models/approx_percentile_function.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ApproxPercentileFunction: """ Computes the approximate percentile of values. """ - input: VariableOr[str] """ The input column from which the approximate percentile is computed. @@ -33,11 +79,12 @@ class ApproxPercentileFunction: """ @classmethod - def from_dict(cls, value: "ApproxPercentileFunctionDict") -> "Self": + def from_dict(cls, value: 'ApproxPercentileFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ApproxPercentileFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ApproxPercentileFunctionDict': + return _transform_to_json_value(self) # type:ignore + class ApproxPercentileFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/avg_function.py b/python/databricks/bundles/features/_models/avg_function.py index 5ecf0cc486b..f25ac863f21 100644 --- a/python/databricks/bundles/features/_models/avg_function.py +++ b/python/databricks/bundles/features/_models/avg_function.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AvgFunction: """ Computes the average of values. """ - input: VariableOr[str] """ The input column from which the average is computed. For Kafka sources, use dot-prefixed path @@ -26,11 +72,12 @@ class AvgFunction: """ @classmethod - def from_dict(cls, value: "AvgFunctionDict") -> "Self": + def from_dict(cls, value: 'AvgFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AvgFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AvgFunctionDict': + return _transform_to_json_value(self) # type:ignore + class AvgFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/column_selection.py b/python/databricks/bundles/features/_models/column_selection.py index d09726cd190..78b53500b33 100644 --- a/python/databricks/bundles/features/_models/column_selection.py +++ b/python/databricks/bundles/features/_models/column_selection.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ColumnSelection: """ A ColumnSelection function, equivalent to the LAST() record of an entity over a lifetime window """ - column: VariableOr[str] """ Column name from source to select as the feature value. """ @classmethod - def from_dict(cls, value: "ColumnSelectionDict") -> "Self": + def from_dict(cls, value: 'ColumnSelectionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ColumnSelectionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ColumnSelectionDict': + return _transform_to_json_value(self) # type:ignore + class ColumnSelectionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/count_function.py b/python/databricks/bundles/features/_models/count_function.py index 6cd86a68132..20c103aee19 100644 --- a/python/databricks/bundles/features/_models/count_function.py +++ b/python/databricks/bundles/features/_models/count_function.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class CountFunction: """ Computes the count of values. """ - input: VariableOr[str] """ The input column from which the count is computed. For Kafka sources, use dot-prefixed path @@ -26,11 +72,12 @@ class CountFunction: """ @classmethod - def from_dict(cls, value: "CountFunctionDict") -> "Self": + def from_dict(cls, value: 'CountFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CountFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CountFunctionDict': + return _transform_to_json_value(self) # type:ignore + class CountFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/custom_udf.py b/python/databricks/bundles/features/_models/custom_udf.py index 7130a8e59cb..2fbf69fc96d 100644 --- a/python/databricks/bundles/features/_models/custom_udf.py +++ b/python/databricks/bundles/features/_models/custom_udf.py @@ -1,29 +1,71 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrList -from databricks.bundles.features._models.input_binding import ( - InputBinding, - InputBindingParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class CustomUdf: """ :meta private: [EXPERIMENTAL] - + A CustomUdf function applies a registered Unity Catalog function row-wise to source columns, producing a single output column per row. """ - function_path: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -40,11 +82,12 @@ class CustomUdf: """ @classmethod - def from_dict(cls, value: "CustomUdfDict") -> "Self": + def from_dict(cls, value: 'CustomUdfDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CustomUdfDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CustomUdfDict': + return _transform_to_json_value(self) # type:ignore + class CustomUdfDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/data_source.py b/python/databricks/bundles/features/_models/data_source.py index b4a22c0276c..7f41d3d180b 100644 --- a/python/databricks/bundles/features/_models/data_source.py +++ b/python/databricks/bundles/features/_models/data_source.py @@ -1,46 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.features._models.delta_table_source import ( - DeltaTableSource, - DeltaTableSourceParam, -) -from databricks.bundles.features._models.feature_view_source import ( - FeatureViewSource, - FeatureViewSourceParam, -) -from databricks.bundles.features._models.kafka_source import ( - KafkaSource, - KafkaSourceParam, -) -from databricks.bundles.features._models.request_source import ( - RequestSource, - RequestSourceParam, -) -from databricks.bundles.features._models.source_lateness import ( - SourceLateness, - SourceLatenessParam, -) -from databricks.bundles.features._models.stream_source import ( - StreamSource, - StreamSourceParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DataSource: """ Specifies the data source backing a feature. Exactly one source type must be set. """ - delta_table_source: VariableOrOptional[DeltaTableSource] = None """ A Delta table data source. @@ -71,17 +93,25 @@ class DataSource: A request-time data source. """ + sdp_source: VariableOrOptional[SdpSource] = None + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] A Spark Declarative Pipeline data source. + """ + stream_source: VariableOrOptional[StreamSource] = None """ A Stream data source. """ @classmethod - def from_dict(cls, value: "DataSourceDict") -> "Self": + def from_dict(cls, value: 'DataSourceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DataSourceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DataSourceDict': + return _transform_to_json_value(self) # type:ignore + class DataSourceDict(TypedDict, total=False): @@ -117,6 +147,13 @@ class DataSourceDict(TypedDict, total=False): A request-time data source. """ + sdp_source: VariableOrOptional[SdpSourceParam] + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] A Spark Declarative Pipeline data source. + """ + stream_source: VariableOrOptional[StreamSourceParam] """ A Stream data source. diff --git a/python/databricks/bundles/features/_models/delta_table_source.py b/python/databricks/bundles/features/_models/delta_table_source.py index 3caedb40fa6..a91c4dba2cc 100644 --- a/python/databricks/bundles/features/_models/delta_table_source.py +++ b/python/databricks/bundles/features/_models/delta_table_source.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DeltaTableSource: """""" @@ -40,11 +87,12 @@ class DeltaTableSource: """ @classmethod - def from_dict(cls, value: "DeltaTableSourceDict") -> "Self": + def from_dict(cls, value: 'DeltaTableSourceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DeltaTableSourceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DeltaTableSourceDict': + return _transform_to_json_value(self) # type:ignore + class DeltaTableSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/entity_column.py b/python/databricks/bundles/features/_models/entity_column.py index d460e9ef4e2..d81033efd1e 100644 --- a/python/databricks/bundles/features/_models/entity_column.py +++ b/python/databricks/bundles/features/_models/entity_column.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EntityColumn: """""" @@ -26,11 +73,12 @@ class EntityColumn: """ @classmethod - def from_dict(cls, value: "EntityColumnDict") -> "Self": + def from_dict(cls, value: 'EntityColumnDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EntityColumnDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EntityColumnDict': + return _transform_to_json_value(self) # type:ignore + class EntityColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/feature.py b/python/databricks/bundles/features/_models/feature.py index d00ed799f92..d7e382c106e 100644 --- a/python/databricks/bundles/features/_models/feature.py +++ b/python/databricks/bundles/features/_models/feature.py @@ -1,45 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.features._models.data_source import ( - DataSource, - DataSourceParam, -) -from databricks.bundles.features._models.entity_column import ( - EntityColumn, - EntityColumnParam, -) -from databricks.bundles.features._models.function import ( - Function, - FunctionParam, -) -from databricks.bundles.features._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.features._models.lineage_context import ( - LineageContext, - LineageContextParam, -) -from databricks.bundles.features._models.timeseries_column import ( - TimeseriesColumn, - TimeseriesColumnParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Feature(Resource): """""" @@ -91,11 +109,12 @@ class Feature(Resource): """ @classmethod - def from_dict(cls, value: "FeatureDict") -> "Self": + def from_dict(cls, value: 'FeatureDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FeatureDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FeatureDict': + return _transform_to_json_value(self) # type:ignore + class FeatureDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/feature_reference.py b/python/databricks/bundles/features/_models/feature_reference.py index 21decdf1547..c7d40731e2f 100644 --- a/python/databricks/bundles/features/_models/feature_reference.py +++ b/python/databricks/bundles/features/_models/feature_reference.py @@ -1,25 +1,71 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FeatureReference: """ :meta private: [EXPERIMENTAL] - + A reference to one registered upstream Feature. A message rather than a bare name so an upstream can later be pinned more precisely (e.g. by version) without a breaking type change. """ - feature: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -28,11 +74,12 @@ class FeatureReference: """ @classmethod - def from_dict(cls, value: "FeatureReferenceDict") -> "Self": + def from_dict(cls, value: 'FeatureReferenceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FeatureReferenceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FeatureReferenceDict': + return _transform_to_json_value(self) # type:ignore + class FeatureReferenceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/feature_view_source.py b/python/databricks/bundles/features/_models/feature_view_source.py index bca40ffcafa..a23d17c8b72 100644 --- a/python/databricks/bundles/features/_models/feature_view_source.py +++ b/python/databricks/bundles/features/_models/feature_view_source.py @@ -1,28 +1,70 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList -from databricks.bundles.features._models.feature_reference import ( - FeatureReference, - FeatureReferenceParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FeatureViewSource: """ :meta private: [EXPERIMENTAL] - + A data source composed from registered upstream Features. """ - feature_references: VariableOrList[FeatureReference] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -31,11 +73,12 @@ class FeatureViewSource: """ @classmethod - def from_dict(cls, value: "FeatureViewSourceDict") -> "Self": + def from_dict(cls, value: 'FeatureViewSourceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FeatureViewSourceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FeatureViewSourceDict': + return _transform_to_json_value(self) # type:ignore + class FeatureViewSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/field_definition.py b/python/databricks/bundles/features/_models/field_definition.py index 3087c757232..ea64a48d055 100644 --- a/python/databricks/bundles/features/_models/field_definition.py +++ b/python/databricks/bundles/features/_models/field_definition.py @@ -1,27 +1,69 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr -from databricks.bundles.features._models.scalar_data_type import ( - ScalarDataType, - ScalarDataTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FieldDefinition: """ A single field definition within a FlatSchema, specifying the field name and its scalar data type. Does not support nested or complex types (arrays, maps, structs). """ - data_type: VariableOr[ScalarDataType] """ The scalar data type of the field. @@ -33,11 +75,12 @@ class FieldDefinition: """ @classmethod - def from_dict(cls, value: "FieldDefinitionDict") -> "Self": + def from_dict(cls, value: 'FieldDefinitionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FieldDefinitionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FieldDefinitionDict': + return _transform_to_json_value(self) # type:ignore + class FieldDefinitionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/first_distinct_function.py b/python/databricks/bundles/features/_models/first_distinct_function.py index 87d460fa228..801a2a4232b 100644 --- a/python/databricks/bundles/features/_models/first_distinct_function.py +++ b/python/databricks/bundles/features/_models/first_distinct_function.py @@ -1,24 +1,70 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FirstDistinctFunction: """ :meta private: [EXPERIMENTAL] - + Returns the first N distinct values, ordered by the feature's timeseries column. """ - input: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -34,11 +80,12 @@ class FirstDistinctFunction: """ @classmethod - def from_dict(cls, value: "FirstDistinctFunctionDict") -> "Self": + def from_dict(cls, value: 'FirstDistinctFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FirstDistinctFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FirstDistinctFunctionDict': + return _transform_to_json_value(self) # type:ignore + class FirstDistinctFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/first_function.py b/python/databricks/bundles/features/_models/first_function.py index eb4ea4f1d2d..feabbd4ed54 100644 --- a/python/databricks/bundles/features/_models/first_function.py +++ b/python/databricks/bundles/features/_models/first_function.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FirstFunction: """ Returns the first value. """ - input: VariableOr[str] """ The input column from which the first value is returned. """ @classmethod - def from_dict(cls, value: "FirstFunctionDict") -> "Self": + def from_dict(cls, value: 'FirstFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FirstFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FirstFunctionDict': + return _transform_to_json_value(self) # type:ignore + class FirstFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/first_n_function.py b/python/databricks/bundles/features/_models/first_n_function.py index 3368c234f68..c7158525891 100644 --- a/python/databricks/bundles/features/_models/first_n_function.py +++ b/python/databricks/bundles/features/_models/first_n_function.py @@ -1,24 +1,70 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FirstNFunction: """ :meta private: [EXPERIMENTAL] - + Returns the first N values, ordered by the feature's timeseries column. """ - input: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -34,11 +80,12 @@ class FirstNFunction: """ @classmethod - def from_dict(cls, value: "FirstNFunctionDict") -> "Self": + def from_dict(cls, value: 'FirstNFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FirstNFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FirstNFunctionDict': + return _transform_to_json_value(self) # type:ignore + class FirstNFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/flat_schema.py b/python/databricks/bundles/features/_models/flat_schema.py index 0535274cc83..603812880c7 100644 --- a/python/databricks/bundles/features/_models/flat_schema.py +++ b/python/databricks/bundles/features/_models/flat_schema.py @@ -1,38 +1,81 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList -from databricks.bundles.features._models.field_definition import ( - FieldDefinition, - FieldDefinitionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FlatSchema: """ A flat (non-nested) schema for request-time fields, defined as an ordered list of field definitions. This schema only supports scalar types. """ - fields: VariableOrList[FieldDefinition] = field(default_factory=list) """ The list of fields in this schema. """ @classmethod - def from_dict(cls, value: "FlatSchemaDict") -> "Self": + def from_dict(cls, value: 'FlatSchemaDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FlatSchemaDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FlatSchemaDict': + return _transform_to_json_value(self) # type:ignore + class FlatSchemaDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/function.py b/python/databricks/bundles/features/_models/function.py index c68a8cd0db2..a803f7e5d2f 100644 --- a/python/databricks/bundles/features/_models/function.py +++ b/python/databricks/bundles/features/_models/function.py @@ -1,28 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.features._models.aggregation_function import ( - AggregationFunction, - AggregationFunctionParam, -) -from databricks.bundles.features._models.column_selection import ( - ColumnSelection, - ColumnSelectionParam, -) -from databricks.bundles.features._models.custom_udf import ( - CustomUdf, - CustomUdfParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Function: """""" @@ -45,11 +80,12 @@ class Function: """ @classmethod - def from_dict(cls, value: "FunctionDict") -> "Self": + def from_dict(cls, value: 'FunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FunctionDict': + return _transform_to_json_value(self) # type:ignore + class FunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/input_binding.py b/python/databricks/bundles/features/_models/input_binding.py index 7ade8c9405e..88fade3d867 100644 --- a/python/databricks/bundles/features/_models/input_binding.py +++ b/python/databricks/bundles/features/_models/input_binding.py @@ -1,24 +1,70 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InputBinding: """ :meta private: [EXPERIMENTAL] - + Binds a single UC function parameter to a source column. """ - column: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -34,11 +80,12 @@ class InputBinding: """ @classmethod - def from_dict(cls, value: "InputBindingDict") -> "Self": + def from_dict(cls, value: 'InputBindingDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InputBindingDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InputBindingDict': + return _transform_to_json_value(self) # type:ignore + class InputBindingDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/job_context.py b/python/databricks/bundles/features/_models/job_context.py index e2d70d0a0bf..0f1d6880eb1 100644 --- a/python/databricks/bundles/features/_models/job_context.py +++ b/python/databricks/bundles/features/_models/job_context.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobContext: """ :meta private: [EXPERIMENTAL] """ - job_id: VariableOrOptional[int] = None """ :meta private: [EXPERIMENTAL] @@ -32,11 +78,12 @@ class JobContext: """ @classmethod - def from_dict(cls, value: "JobContextDict") -> "Self": + def from_dict(cls, value: 'JobContextDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobContextDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobContextDict': + return _transform_to_json_value(self) # type:ignore + class JobContextDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/kafka_source.py b/python/databricks/bundles/features/_models/kafka_source.py index a7142a7cacb..a7febacaadc 100644 --- a/python/databricks/bundles/features/_models/kafka_source.py +++ b/python/databricks/bundles/features/_models/kafka_source.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class KafkaSource: """""" @@ -26,11 +73,12 @@ class KafkaSource: """ @classmethod - def from_dict(cls, value: "KafkaSourceDict") -> "Self": + def from_dict(cls, value: 'KafkaSourceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "KafkaSourceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'KafkaSourceDict': + return _transform_to_json_value(self) # type:ignore + class KafkaSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/last_distinct_function.py b/python/databricks/bundles/features/_models/last_distinct_function.py index f0ed97454a1..99ae7e73635 100644 --- a/python/databricks/bundles/features/_models/last_distinct_function.py +++ b/python/databricks/bundles/features/_models/last_distinct_function.py @@ -1,24 +1,70 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LastDistinctFunction: """ :meta private: [EXPERIMENTAL] - + Returns the last N distinct values, ordered by the feature's timeseries column. """ - input: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -34,11 +80,12 @@ class LastDistinctFunction: """ @classmethod - def from_dict(cls, value: "LastDistinctFunctionDict") -> "Self": + def from_dict(cls, value: 'LastDistinctFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LastDistinctFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LastDistinctFunctionDict': + return _transform_to_json_value(self) # type:ignore + class LastDistinctFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/last_function.py b/python/databricks/bundles/features/_models/last_function.py index 56cd70a231d..5f89e093c44 100644 --- a/python/databricks/bundles/features/_models/last_function.py +++ b/python/databricks/bundles/features/_models/last_function.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LastFunction: """ Returns the last value. """ - input: VariableOr[str] """ The input column from which the last value is returned. """ @classmethod - def from_dict(cls, value: "LastFunctionDict") -> "Self": + def from_dict(cls, value: 'LastFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LastFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LastFunctionDict': + return _transform_to_json_value(self) # type:ignore + class LastFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/last_n_function.py b/python/databricks/bundles/features/_models/last_n_function.py index 03131e3182f..8105579bae3 100644 --- a/python/databricks/bundles/features/_models/last_n_function.py +++ b/python/databricks/bundles/features/_models/last_n_function.py @@ -1,24 +1,70 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LastNFunction: """ :meta private: [EXPERIMENTAL] - + Returns the last N values, ordered by the feature's timeseries column. """ - input: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -34,11 +80,12 @@ class LastNFunction: """ @classmethod - def from_dict(cls, value: "LastNFunctionDict") -> "Self": + def from_dict(cls, value: 'LastNFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LastNFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LastNFunctionDict': + return _transform_to_json_value(self) # type:ignore + class LastNFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/lifecycle.py b/python/databricks/bundles/features/_models/lifecycle.py index 697776a198e..8c910bb74c2 100644 --- a/python/databricks/bundles/features/_models/lifecycle.py +++ b/python/databricks/bundles/features/_models/lifecycle.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +68,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/lineage_context.py b/python/databricks/bundles/features/_models/lineage_context.py index d11cb8082d6..8b106bde8e9 100644 --- a/python/databricks/bundles/features/_models/lineage_context.py +++ b/python/databricks/bundles/features/_models/lineage_context.py @@ -1,28 +1,70 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.features._models.job_context import ( - JobContext, - JobContextParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LineageContext: """ :meta private: [EXPERIMENTAL] - + Lineage context information for tracking where an API was invoked. This will allow us to track lineage, which currently uses caller entity information for use across the Lineage Client and Observability in Lumberjack. """ - job_context: VariableOrOptional[JobContext] = None """ :meta private: [EXPERIMENTAL] @@ -38,11 +80,12 @@ class LineageContext: """ @classmethod - def from_dict(cls, value: "LineageContextDict") -> "Self": + def from_dict(cls, value: 'LineageContextDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LineageContextDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LineageContextDict': + return _transform_to_json_value(self) # type:ignore + class LineageContextDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/max_function.py b/python/databricks/bundles/features/_models/max_function.py index dbc7c7f8f81..90d0b0ddb82 100644 --- a/python/databricks/bundles/features/_models/max_function.py +++ b/python/databricks/bundles/features/_models/max_function.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MaxFunction: """ Computes the maximum value. """ - input: VariableOr[str] """ The input column from which the maximum is computed. """ @classmethod - def from_dict(cls, value: "MaxFunctionDict") -> "Self": + def from_dict(cls, value: 'MaxFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MaxFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MaxFunctionDict': + return _transform_to_json_value(self) # type:ignore + class MaxFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/min_function.py b/python/databricks/bundles/features/_models/min_function.py index 17bebfdbc41..cad2d927012 100644 --- a/python/databricks/bundles/features/_models/min_function.py +++ b/python/databricks/bundles/features/_models/min_function.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MinFunction: """ Computes the minimum value. """ - input: VariableOr[str] """ The input column from which the minimum is computed. """ @classmethod - def from_dict(cls, value: "MinFunctionDict") -> "Self": + def from_dict(cls, value: 'MinFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MinFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MinFunctionDict': + return _transform_to_json_value(self) # type:ignore + class MinFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/request_source.py b/python/databricks/bundles/features/_models/request_source.py index 2ad0c4a0225..4ad8edd8750 100644 --- a/python/databricks/bundles/features/_models/request_source.py +++ b/python/databricks/bundles/features/_models/request_source.py @@ -1,23 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RequestSource: """ A request-time data source whose value is provided at inference time: offline batch scoring or online serving endpoint """ - dataframe_schema: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -32,11 +77,12 @@ class RequestSource: """ @classmethod - def from_dict(cls, value: "RequestSourceDict") -> "Self": + def from_dict(cls, value: 'RequestSourceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RequestSourceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RequestSourceDict': + return _transform_to_json_value(self) # type:ignore + class RequestSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/rolling_window.py b/python/databricks/bundles/features/_models/rolling_window.py index 9cbdd0b7c2f..9604168e1c8 100644 --- a/python/databricks/bundles/features/_models/rolling_window.py +++ b/python/databricks/bundles/features/_models/rolling_window.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RollingWindow: """ A rolling time window with an optional non-negative delay. """ - delay: VariableOrOptional[str] = None """ Non-negative analytic lag that evaluates the window this far in the past. Use this for timing @@ -31,11 +77,12 @@ class RollingWindow: """ @classmethod - def from_dict(cls, value: "RollingWindowDict") -> "Self": + def from_dict(cls, value: 'RollingWindowDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RollingWindowDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RollingWindowDict': + return _transform_to_json_value(self) # type:ignore + class RollingWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/sawtooth_window.py b/python/databricks/bundles/features/_models/sawtooth_window.py index 303ab073d46..ba1de2258c4 100644 --- a/python/databricks/bundles/features/_models/sawtooth_window.py +++ b/python/databricks/bundles/features/_models/sawtooth_window.py @@ -1,28 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SawtoothWindow: """ :meta private: [EXPERIMENTAL] - + A sawtooth window served via the hybrid batch + streaming path. The batch pipeline maintains daily partial aggregates for the bulk of the window while the streaming pipeline maintains the most recent day(s), and serving merges them on read. Same field shape as RollingWindow, but a distinct type so the control plane can explicitly identify hybrid (sawtooth) features rather than inferring hybrid behavior from window_duration. """ - delay: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -41,11 +87,12 @@ class SawtoothWindow: """ @classmethod - def from_dict(cls, value: "SawtoothWindowDict") -> "Self": + def from_dict(cls, value: 'SawtoothWindowDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SawtoothWindowDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SawtoothWindowDict': + return _transform_to_json_value(self) # type:ignore + class SawtoothWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/scalar_data_type.py b/python/databricks/bundles/features/_models/scalar_data_type.py index 5c115502b0a..e673f94d9ce 100644 --- a/python/databricks/bundles/features/_models/scalar_data_type.py +++ b/python/databricks/bundles/features/_models/scalar_data_type.py @@ -1,15 +1,64 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam class ScalarDataType(Enum): """ Scalar data types for request-time field definitions. Only flat (non-nested) types are supported. """ - INTEGER = "INTEGER" FLOAT = "FLOAT" BOOLEAN = "BOOLEAN" @@ -22,20 +71,4 @@ class ScalarDataType(Enum): BINARY = "BINARY" DECIMAL = "DECIMAL" - -ScalarDataTypeParam = ( - Literal[ - "INTEGER", - "FLOAT", - "BOOLEAN", - "STRING", - "DOUBLE", - "LONG", - "TIMESTAMP", - "DATE", - "SHORT", - "BINARY", - "DECIMAL", - ] - | ScalarDataType -) +ScalarDataTypeParam = Literal["INTEGER", "FLOAT", "BOOLEAN", "STRING", "DOUBLE", "LONG", "TIMESTAMP", "DATE", "SHORT", "BINARY", "DECIMAL"] | ScalarDataType diff --git a/python/databricks/bundles/features/_models/sdp_source.py b/python/databricks/bundles/features/_models/sdp_source.py new file mode 100644 index 00000000000..0c983aa393d --- /dev/null +++ b/python/databricks/bundles/features/_models/sdp_source.py @@ -0,0 +1,189 @@ +# Code generated by pydabs-codegen. DO NOT EDIT. + +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + + +if TYPE_CHECKING: + from typing_extensions import Self + +@dataclass(kw_only=True) +class SdpSource: + """ + :meta private: [EXPERIMENTAL] + + A Spark Declarative Pipeline (SDP) data source. + """ + dataframe_schema: VariableOrOptional[str] = None + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] Schema of the resulting dataframe after transformations, in Spark StructType JSON format (from df.schema.json()). + Required if transformation_sql is specified. + Example: {"type":"struct","fields":[{"name":"col_a","type":"integer","nullable":true,"metadata":{}},{"name":"col_c","type":"integer","nullable":true,"metadata":{}}]} + """ + + filter_condition: VariableOrOptional[str] = None + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] Single WHERE clause to filter the pipeline's output table before applying transformations. Will be row-wise evaluated, so should only include conditionals and projections. + """ + + input_tables: VariableOrList[str] = field(default_factory=list) + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] The full three-part names of the tables the pipeline reads from. Required to materialize this + feature with a TableTrigger, which starts a run when these tables update. Each table must exist + when the feature is created. After the pipeline runs, Feature Store checks that it reads from + these tables, and the run fails if it doesn't. At most 100 tables. + """ + + output_table: VariableOrOptional[str] = None + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] The full three-part (catalog, schema, table) name of the pipeline's output table to read from. + The table doesn't need to exist until the pipeline first runs. Required. + """ + + pipeline_id: VariableOrOptional[str] = None + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] The ID of the Spark Declarative Pipeline that Feature Store runs to produce this source, before + each materialization or backfill. The pipeline must already exist when the feature is created, + and you need CAN_RUN, CAN_MANAGE, or owner permission on it. It doesn't need to have run yet. + Deleting the feature doesn't delete the pipeline. If the pipeline is deleted, this field reads + back empty. Required. + """ + + transformation_sql: VariableOrOptional[str] = None + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] A single SQL SELECT expression applied after filter_condition. + Should contain all the columns needed (eg. "SELECT *, col_a + col_b AS col_c FROM x.y.z WHERE col_a > 0" would have `transformation_sql` "*, col_a + col_b AS col_c") + If transformation_sql is not provided, all columns of the pipeline's output table are present in the DataSource dataframe. + """ + + @classmethod + def from_dict(cls, value: 'SdpSourceDict') -> 'Self': + return _transform(cls, value) + + def as_dict(self) -> 'SdpSourceDict': + return _transform_to_json_value(self) # type:ignore + + + +class SdpSourceDict(TypedDict, total=False): + """""" + + dataframe_schema: VariableOrOptional[str] + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] Schema of the resulting dataframe after transformations, in Spark StructType JSON format (from df.schema.json()). + Required if transformation_sql is specified. + Example: {"type":"struct","fields":[{"name":"col_a","type":"integer","nullable":true,"metadata":{}},{"name":"col_c","type":"integer","nullable":true,"metadata":{}}]} + """ + + filter_condition: VariableOrOptional[str] + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] Single WHERE clause to filter the pipeline's output table before applying transformations. Will be row-wise evaluated, so should only include conditionals and projections. + """ + + input_tables: VariableOrList[str] + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] The full three-part names of the tables the pipeline reads from. Required to materialize this + feature with a TableTrigger, which starts a run when these tables update. Each table must exist + when the feature is created. After the pipeline runs, Feature Store checks that it reads from + these tables, and the run fails if it doesn't. At most 100 tables. + """ + + output_table: VariableOrOptional[str] + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] The full three-part (catalog, schema, table) name of the pipeline's output table to read from. + The table doesn't need to exist until the pipeline first runs. Required. + """ + + pipeline_id: VariableOrOptional[str] + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] The ID of the Spark Declarative Pipeline that Feature Store runs to produce this source, before + each materialization or backfill. The pipeline must already exist when the feature is created, + and you need CAN_RUN, CAN_MANAGE, or owner permission on it. It doesn't need to have run yet. + Deleting the feature doesn't delete the pipeline. If the pipeline is deleted, this field reads + back empty. Required. + """ + + transformation_sql: VariableOrOptional[str] + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] A single SQL SELECT expression applied after filter_condition. + Should contain all the columns needed (eg. "SELECT *, col_a + col_b AS col_c FROM x.y.z WHERE col_a > 0" would have `transformation_sql` "*, col_a + col_b AS col_c") + If transformation_sql is not provided, all columns of the pipeline's output table are present in the DataSource dataframe. + """ + + +SdpSourceParam = SdpSourceDict | SdpSource diff --git a/python/databricks/bundles/features/_models/sliding_window.py b/python/databricks/bundles/features/_models/sliding_window.py index cdf67ed2af0..ea4f1736c7e 100644 --- a/python/databricks/bundles/features/_models/sliding_window.py +++ b/python/databricks/bundles/features/_models/sliding_window.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SlidingWindow: """""" @@ -46,11 +93,12 @@ class SlidingWindow: """ @classmethod - def from_dict(cls, value: "SlidingWindowDict") -> "Self": + def from_dict(cls, value: 'SlidingWindowDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SlidingWindowDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SlidingWindowDict': + return _transform_to_json_value(self) # type:ignore + class SlidingWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/source_lateness.py b/python/databricks/bundles/features/_models/source_lateness.py index 757eac320e8..a88758b38f6 100644 --- a/python/databricks/bundles/features/_models/source_lateness.py +++ b/python/databricks/bundles/features/_models/source_lateness.py @@ -1,24 +1,70 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SourceLateness: """ :meta private: [EXPERIMENTAL] - + Configures when event-time data from this source is considered complete for a Feature. """ - settling_delay: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -30,11 +76,12 @@ class SourceLateness: """ @classmethod - def from_dict(cls, value: "SourceLatenessDict") -> "Self": + def from_dict(cls, value: 'SourceLatenessDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SourceLatenessDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SourceLatenessDict': + return _transform_to_json_value(self) # type:ignore + class SourceLatenessDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/stddev_pop_function.py b/python/databricks/bundles/features/_models/stddev_pop_function.py index edd63b11b93..319eeb1d5dd 100644 --- a/python/databricks/bundles/features/_models/stddev_pop_function.py +++ b/python/databricks/bundles/features/_models/stddev_pop_function.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class StddevPopFunction: """ Computes the population standard deviation. """ - input: VariableOr[str] """ The input column from which the population standard deviation is computed. For Kafka sources, @@ -26,11 +72,12 @@ class StddevPopFunction: """ @classmethod - def from_dict(cls, value: "StddevPopFunctionDict") -> "Self": + def from_dict(cls, value: 'StddevPopFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "StddevPopFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'StddevPopFunctionDict': + return _transform_to_json_value(self) # type:ignore + class StddevPopFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/stddev_samp_function.py b/python/databricks/bundles/features/_models/stddev_samp_function.py index ca253c92315..bca039fbbaa 100644 --- a/python/databricks/bundles/features/_models/stddev_samp_function.py +++ b/python/databricks/bundles/features/_models/stddev_samp_function.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class StddevSampFunction: """ Computes the sample standard deviation. """ - input: VariableOr[str] """ The input column from which the sample standard deviation is computed. """ @classmethod - def from_dict(cls, value: "StddevSampFunctionDict") -> "Self": + def from_dict(cls, value: 'StddevSampFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "StddevSampFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'StddevSampFunctionDict': + return _transform_to_json_value(self) # type:ignore + class StddevSampFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/stream_source.py b/python/databricks/bundles/features/_models/stream_source.py index 7b591576426..e803b243641 100644 --- a/python/databricks/bundles/features/_models/stream_source.py +++ b/python/databricks/bundles/features/_models/stream_source.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class StreamSource: """ A Stream entity used as a data source for a feature. """ - full_name: VariableOr[str] """ Three-part full name of the Stream (catalog.schema.stream). @@ -45,11 +91,12 @@ class StreamSource: """ @classmethod - def from_dict(cls, value: "StreamSourceDict") -> "Self": + def from_dict(cls, value: 'StreamSourceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "StreamSourceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'StreamSourceDict': + return _transform_to_json_value(self) # type:ignore + class StreamSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/sum_function.py b/python/databricks/bundles/features/_models/sum_function.py index f74c3e8a6ac..3839fd0f349 100644 --- a/python/databricks/bundles/features/_models/sum_function.py +++ b/python/databricks/bundles/features/_models/sum_function.py @@ -1,22 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SumFunction: """ Computes the sum of values. """ - input: VariableOr[str] """ The input column from which the sum is computed. For Kafka sources, use dot-prefixed path @@ -26,11 +72,12 @@ class SumFunction: """ @classmethod - def from_dict(cls, value: "SumFunctionDict") -> "Self": + def from_dict(cls, value: 'SumFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SumFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SumFunctionDict': + return _transform_to_json_value(self) # type:ignore + class SumFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/time_window.py b/python/databricks/bundles/features/_models/time_window.py index 85bd6a488dd..16c6f51b3ba 100644 --- a/python/databricks/bundles/features/_models/time_window.py +++ b/python/databricks/bundles/features/_models/time_window.py @@ -1,32 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.features._models.rolling_window import ( - RollingWindow, - RollingWindowParam, -) -from databricks.bundles.features._models.sawtooth_window import ( - SawtoothWindow, - SawtoothWindowParam, -) -from databricks.bundles.features._models.sliding_window import ( - SlidingWindow, - SlidingWindowParam, -) -from databricks.bundles.features._models.tumbling_window import ( - TumblingWindow, - TumblingWindowParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TimeWindow: """""" @@ -62,11 +93,12 @@ class TimeWindow: tumbling: VariableOrOptional[TumblingWindow] = None @classmethod - def from_dict(cls, value: "TimeWindowDict") -> "Self": + def from_dict(cls, value: 'TimeWindowDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TimeWindowDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TimeWindowDict': + return _transform_to_json_value(self) # type:ignore + class TimeWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/timeseries_column.py b/python/databricks/bundles/features/_models/timeseries_column.py index be19cf7dad0..740439aaf52 100644 --- a/python/databricks/bundles/features/_models/timeseries_column.py +++ b/python/databricks/bundles/features/_models/timeseries_column.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TimeseriesColumn: """""" @@ -26,11 +73,12 @@ class TimeseriesColumn: """ @classmethod - def from_dict(cls, value: "TimeseriesColumnDict") -> "Self": + def from_dict(cls, value: 'TimeseriesColumnDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TimeseriesColumnDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TimeseriesColumnDict': + return _transform_to_json_value(self) # type:ignore + class TimeseriesColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/tumbling_window.py b/python/databricks/bundles/features/_models/tumbling_window.py index 38a873fa5cd..fc28ae9c5e0 100644 --- a/python/databricks/bundles/features/_models/tumbling_window.py +++ b/python/databricks/bundles/features/_models/tumbling_window.py @@ -1,16 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TumblingWindow: """""" @@ -39,11 +86,12 @@ class TumblingWindow: """ @classmethod - def from_dict(cls, value: "TumblingWindowDict") -> "Self": + def from_dict(cls, value: 'TumblingWindowDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TumblingWindowDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TumblingWindowDict': + return _transform_to_json_value(self) # type:ignore + class TumblingWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/var_pop_function.py b/python/databricks/bundles/features/_models/var_pop_function.py index 79b6dc5f425..fa68523a468 100644 --- a/python/databricks/bundles/features/_models/var_pop_function.py +++ b/python/databricks/bundles/features/_models/var_pop_function.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class VarPopFunction: """ Computes the population variance. """ - input: VariableOr[str] """ The input column from which the population variance is computed. """ @classmethod - def from_dict(cls, value: "VarPopFunctionDict") -> "Self": + def from_dict(cls, value: 'VarPopFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "VarPopFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'VarPopFunctionDict': + return _transform_to_json_value(self) # type:ignore + class VarPopFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/var_samp_function.py b/python/databricks/bundles/features/_models/var_samp_function.py index 8f0ee87ce77..4ede4549412 100644 --- a/python/databricks/bundles/features/_models/var_samp_function.py +++ b/python/databricks/bundles/features/_models/var_samp_function.py @@ -1,33 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam +from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam +from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam +from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam +from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam +from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam +from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam +from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam +from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam +from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam +from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam +from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam +from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam +from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam +from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam +from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam +from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam +from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam +from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam +from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam +from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam +from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam +from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam +from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam +from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam +from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam +from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam +from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam +from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam +from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam +from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam +from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam +from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam +from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam +from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam +from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam +from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam +from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam +from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam +from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class VarSampFunction: """ Computes the sample variance. """ - input: VariableOr[str] """ The input column from which the sample variance is computed. """ @classmethod - def from_dict(cls, value: "VarSampFunctionDict") -> "Self": + def from_dict(cls, value: 'VarSampFunctionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "VarSampFunctionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'VarSampFunctionDict': + return _transform_to_json_value(self) # type:ignore + class VarSampFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/genie_spaces/__init__.py b/python/databricks/bundles/genie_spaces/__init__.py index a56af37c716..6b12939bf92 100644 --- a/python/databricks/bundles/genie_spaces/__init__.py +++ b/python/databricks/bundles/genie_spaces/__init__.py @@ -1,36 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "GenieSpace", - "GenieSpaceDict", - "GenieSpaceParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "Permission", - "PermissionDict", - "PermissionLevel", - "PermissionLevelParam", - "PermissionParam", + 'GenieSpace', + 'GenieSpaceDict', + 'GenieSpaceParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'Permission', + 'PermissionDict', + 'PermissionLevel', + 'PermissionLevelParam', + 'PermissionParam', ] -from databricks.bundles.genie_spaces._models.genie_space import ( - GenieSpace, - GenieSpaceDict, - GenieSpaceParam, -) -from databricks.bundles.genie_spaces._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.genie_spaces._models.permission import ( - Permission, - PermissionDict, - PermissionParam, -) -from databricks.bundles.genie_spaces._models.permission_level import ( - PermissionLevel, - PermissionLevelParam, -) +from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceDict, GenieSpaceParam +from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.genie_spaces._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel, PermissionLevelParam + diff --git a/python/databricks/bundles/genie_spaces/_models/genie_space.py b/python/databricks/bundles/genie_spaces/_models/genie_space.py index bb1ec94c8df..a481f6bd59d 100644 --- a/python/databricks/bundles/genie_spaces/_models/genie_space.py +++ b/python/databricks/bundles/genie_spaces/_models/genie_space.py @@ -1,22 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, Any, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.genie_spaces._models.permission import ( - Permission, - PermissionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.genie_spaces._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GenieSpace(Resource): """""" @@ -64,11 +64,12 @@ class GenieSpace(Resource): """ @classmethod - def from_dict(cls, value: "GenieSpaceDict") -> "Self": + def from_dict(cls, value: 'GenieSpaceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GenieSpaceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GenieSpaceDict': + return _transform_to_json_value(self) # type:ignore + class GenieSpaceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/genie_spaces/_models/lifecycle.py b/python/databricks/bundles/genie_spaces/_models/lifecycle.py index 697776a198e..b5ad6018a8c 100644 --- a/python/databricks/bundles/genie_spaces/_models/lifecycle.py +++ b/python/databricks/bundles/genie_spaces/_models/lifecycle.py @@ -1,16 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceDict, GenieSpaceParam +from databricks.bundles.genie_spaces._models.permission import Permission, PermissionDict, PermissionParam +from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +27,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/genie_spaces/_models/permission.py b/python/databricks/bundles/genie_spaces/_models/permission.py index 372f063d547..3984cc95652 100644 --- a/python/databricks/bundles/genie_spaces/_models/permission.py +++ b/python/databricks/bundles/genie_spaces/_models/permission.py @@ -1,20 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.genie_spaces._models.permission_level import ( - PermissionLevel, - PermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceDict, GenieSpaceParam +from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel, PermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Permission: """""" @@ -40,11 +42,12 @@ class Permission: """ @classmethod - def from_dict(cls, value: "PermissionDict") -> "Self": + def from_dict(cls, value: 'PermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PermissionDict': + return _transform_to_json_value(self) # type:ignore + class PermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/genie_spaces/_models/permission_level.py b/python/databricks/bundles/genie_spaces/_models/permission_level.py index c6111911b29..74f0ee9d40a 100644 --- a/python/databricks/bundles/genie_spaces/_models/permission_level.py +++ b/python/databricks/bundles/genie_spaces/_models/permission_level.py @@ -1,14 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceDict, GenieSpaceParam +from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.genie_spaces._models.permission import Permission, PermissionDict, PermissionParam class PermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" @@ -30,29 +38,4 @@ class PermissionLevel(Enum): CAN_MONITOR_ONLY = "CAN_MONITOR_ONLY" CAN_CREATE_APP = "CAN_CREATE_APP" - -PermissionLevelParam = ( - Literal[ - "CAN_MANAGE", - "CAN_RESTART", - "CAN_ATTACH_TO", - "IS_OWNER", - "CAN_MANAGE_RUN", - "CAN_VIEW", - "CAN_READ", - "CAN_RUN", - "CAN_EDIT", - "CAN_USE", - "CAN_MANAGE_STAGING_VERSIONS", - "CAN_MANAGE_PRODUCTION_VERSIONS", - "CAN_EDIT_METADATA", - "CAN_VIEW_METADATA", - "CAN_BIND", - "CAN_QUERY", - "CAN_MONITOR", - "CAN_CREATE", - "CAN_MONITOR_ONLY", - "CAN_CREATE_APP", - ] - | PermissionLevel -) +PermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW", "CAN_READ", "CAN_RUN", "CAN_EDIT", "CAN_USE", "CAN_MANAGE_STAGING_VERSIONS", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_EDIT_METADATA", "CAN_VIEW_METADATA", "CAN_BIND", "CAN_QUERY", "CAN_MONITOR", "CAN_CREATE", "CAN_MONITOR_ONLY", "CAN_CREATE_APP"] | PermissionLevel diff --git a/python/databricks/bundles/instance_pools/__init__.py b/python/databricks/bundles/instance_pools/__init__.py index 2c5a0e1c4eb..5e699b4a170 100644 --- a/python/databricks/bundles/instance_pools/__init__.py +++ b/python/databricks/bundles/instance_pools/__init__.py @@ -1,130 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "DiskSpec", - "DiskSpecDict", - "DiskSpecParam", - "DiskType", - "DiskTypeAzureDiskVolumeType", - "DiskTypeAzureDiskVolumeTypeParam", - "DiskTypeDict", - "DiskTypeEbsVolumeType", - "DiskTypeEbsVolumeTypeParam", - "DiskTypeParam", - "DockerBasicAuth", - "DockerBasicAuthDict", - "DockerBasicAuthParam", - "DockerImage", - "DockerImageDict", - "DockerImageParam", - "GcpAvailability", - "GcpAvailabilityParam", - "InstancePool", - "InstancePoolAwsAttributes", - "InstancePoolAwsAttributesAvailability", - "InstancePoolAwsAttributesAvailabilityParam", - "InstancePoolAwsAttributesDict", - "InstancePoolAwsAttributesParam", - "InstancePoolAzureAttributes", - "InstancePoolAzureAttributesAvailability", - "InstancePoolAzureAttributesAvailabilityParam", - "InstancePoolAzureAttributesDict", - "InstancePoolAzureAttributesParam", - "InstancePoolDict", - "InstancePoolGcpAttributes", - "InstancePoolGcpAttributesDict", - "InstancePoolGcpAttributesParam", - "InstancePoolParam", - "InstancePoolPermission", - "InstancePoolPermissionDict", - "InstancePoolPermissionLevel", - "InstancePoolPermissionLevelParam", - "InstancePoolPermissionParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "NodeTypeFlexibility", - "NodeTypeFlexibilityDict", - "NodeTypeFlexibilityParam", + 'DiskSpec', + 'DiskSpecDict', + 'DiskSpecParam', + 'DiskType', + 'DiskTypeAzureDiskVolumeType', + 'DiskTypeAzureDiskVolumeTypeParam', + 'DiskTypeDict', + 'DiskTypeEbsVolumeType', + 'DiskTypeEbsVolumeTypeParam', + 'DiskTypeParam', + 'DockerBasicAuth', + 'DockerBasicAuthDict', + 'DockerBasicAuthParam', + 'DockerImage', + 'DockerImageDict', + 'DockerImageParam', + 'GcpAvailability', + 'GcpAvailabilityParam', + 'InstancePool', + 'InstancePoolAwsAttributes', + 'InstancePoolAwsAttributesAvailability', + 'InstancePoolAwsAttributesAvailabilityParam', + 'InstancePoolAwsAttributesDict', + 'InstancePoolAwsAttributesParam', + 'InstancePoolAzureAttributes', + 'InstancePoolAzureAttributesAvailability', + 'InstancePoolAzureAttributesAvailabilityParam', + 'InstancePoolAzureAttributesDict', + 'InstancePoolAzureAttributesParam', + 'InstancePoolDict', + 'InstancePoolGcpAttributes', + 'InstancePoolGcpAttributesDict', + 'InstancePoolGcpAttributesParam', + 'InstancePoolParam', + 'InstancePoolPermission', + 'InstancePoolPermissionDict', + 'InstancePoolPermissionLevel', + 'InstancePoolPermissionLevelParam', + 'InstancePoolPermissionParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'NodeTypeFlexibility', + 'NodeTypeFlexibilityDict', + 'NodeTypeFlexibilityParam', ] -from databricks.bundles.instance_pools._models.disk_spec import ( - DiskSpec, - DiskSpecDict, - DiskSpecParam, -) -from databricks.bundles.instance_pools._models.disk_type import ( - DiskType, - DiskTypeDict, - DiskTypeParam, -) -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import ( - DiskTypeAzureDiskVolumeType, - DiskTypeAzureDiskVolumeTypeParam, -) -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import ( - DiskTypeEbsVolumeType, - DiskTypeEbsVolumeTypeParam, -) -from databricks.bundles.instance_pools._models.docker_basic_auth import ( - DockerBasicAuth, - DockerBasicAuthDict, - DockerBasicAuthParam, -) -from databricks.bundles.instance_pools._models.docker_image import ( - DockerImage, - DockerImageDict, - DockerImageParam, -) -from databricks.bundles.instance_pools._models.gcp_availability import ( - GcpAvailability, - GcpAvailabilityParam, -) -from databricks.bundles.instance_pools._models.instance_pool import ( - InstancePool, - InstancePoolDict, - InstancePoolParam, -) -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import ( - InstancePoolAwsAttributes, - InstancePoolAwsAttributesDict, - InstancePoolAwsAttributesParam, -) -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import ( - InstancePoolAwsAttributesAvailability, - InstancePoolAwsAttributesAvailabilityParam, -) -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import ( - InstancePoolAzureAttributes, - InstancePoolAzureAttributesDict, - InstancePoolAzureAttributesParam, -) -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import ( - InstancePoolAzureAttributesAvailability, - InstancePoolAzureAttributesAvailabilityParam, -) -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import ( - InstancePoolGcpAttributes, - InstancePoolGcpAttributesDict, - InstancePoolGcpAttributesParam, -) -from databricks.bundles.instance_pools._models.instance_pool_permission import ( - InstancePoolPermission, - InstancePoolPermissionDict, - InstancePoolPermissionParam, -) -from databricks.bundles.instance_pools._models.instance_pool_permission_level import ( - InstancePoolPermissionLevel, - InstancePoolPermissionLevelParam, -) -from databricks.bundles.instance_pools._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.instance_pools._models.node_type_flexibility import ( - NodeTypeFlexibility, - NodeTypeFlexibilityDict, - NodeTypeFlexibilityParam, -) +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + diff --git a/python/databricks/bundles/instance_pools/_models/disk_spec.py b/python/databricks/bundles/instance_pools/_models/disk_spec.py index b9edbb5d4e8..335ff6bc912 100644 --- a/python/databricks/bundles/instance_pools/_models/disk_spec.py +++ b/python/databricks/bundles/instance_pools/_models/disk_spec.py @@ -1,17 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DiskSpec: """ @@ -20,7 +43,6 @@ class DiskSpec: 2 disks, 100 GiB each, then Databricks will launch a total of 6 disks, 100 GiB each, for this cluster. """ - disk_count: VariableOrOptional[int] = None """ The number of disks launched for each instance: @@ -72,11 +94,12 @@ class DiskSpec: """ @classmethod - def from_dict(cls, value: "DiskSpecDict") -> "Self": + def from_dict(cls, value: 'DiskSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DiskSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DiskSpecDict': + return _transform_to_json_value(self) # type:ignore + class DiskSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/disk_type.py b/python/databricks/bundles/instance_pools/_models/disk_type.py index 752bf75bf78..d5a3b85e516 100644 --- a/python/databricks/bundles/instance_pools/_models/disk_type.py +++ b/python/databricks/bundles/instance_pools/_models/disk_type.py @@ -1,30 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import ( - DiskTypeAzureDiskVolumeType, - DiskTypeAzureDiskVolumeTypeParam, -) -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import ( - DiskTypeEbsVolumeType, - DiskTypeEbsVolumeTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DiskType: """ Describes the disk type. """ - azure_disk_volume_type: VariableOrOptional[DiskTypeAzureDiskVolumeType] = None """ All Azure Disk types that Databricks supports. @@ -38,11 +53,12 @@ class DiskType: """ @classmethod - def from_dict(cls, value: "DiskTypeDict") -> "Self": + def from_dict(cls, value: 'DiskTypeDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DiskTypeDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DiskTypeDict': + return _transform_to_json_value(self) # type:ignore + class DiskTypeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/disk_type_azure_disk_volume_type.py b/python/databricks/bundles/instance_pools/_models/disk_type_azure_disk_volume_type.py index 740ab2dd245..4114d62c59d 100644 --- a/python/databricks/bundles/instance_pools/_models/disk_type_azure_disk_volume_type.py +++ b/python/databricks/bundles/instance_pools/_models/disk_type_azure_disk_volume_type.py @@ -1,7 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam class DiskTypeAzureDiskVolumeType(Enum): @@ -9,11 +36,7 @@ class DiskTypeAzureDiskVolumeType(Enum): All Azure Disk types that Databricks supports. See https://docs.microsoft.com/en-us/azure/storage/storage-about-disks-and-vhds-linux#types-of-disks """ - PREMIUM_LRS = "PREMIUM_LRS" STANDARD_LRS = "STANDARD_LRS" - -DiskTypeAzureDiskVolumeTypeParam = ( - Literal["PREMIUM_LRS", "STANDARD_LRS"] | DiskTypeAzureDiskVolumeType -) +DiskTypeAzureDiskVolumeTypeParam = Literal["PREMIUM_LRS", "STANDARD_LRS"] | DiskTypeAzureDiskVolumeType diff --git a/python/databricks/bundles/instance_pools/_models/disk_type_ebs_volume_type.py b/python/databricks/bundles/instance_pools/_models/disk_type_ebs_volume_type.py index cb5ae6b7a05..3d3c96487d8 100644 --- a/python/databricks/bundles/instance_pools/_models/disk_type_ebs_volume_type.py +++ b/python/databricks/bundles/instance_pools/_models/disk_type_ebs_volume_type.py @@ -1,7 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam class DiskTypeEbsVolumeType(Enum): @@ -9,11 +36,7 @@ class DiskTypeEbsVolumeType(Enum): All EBS volume types that Databricks supports. See https://aws.amazon.com/ebs/details/ for details. """ - GENERAL_PURPOSE_SSD = "GENERAL_PURPOSE_SSD" THROUGHPUT_OPTIMIZED_HDD = "THROUGHPUT_OPTIMIZED_HDD" - -DiskTypeEbsVolumeTypeParam = ( - Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | DiskTypeEbsVolumeType -) +DiskTypeEbsVolumeTypeParam = Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | DiskTypeEbsVolumeType diff --git a/python/databricks/bundles/instance_pools/_models/docker_basic_auth.py b/python/databricks/bundles/instance_pools/_models/docker_basic_auth.py index 552ea90a83b..6f165be26c1 100644 --- a/python/databricks/bundles/instance_pools/_models/docker_basic_auth.py +++ b/python/databricks/bundles/instance_pools/_models/docker_basic_auth.py @@ -1,16 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DockerBasicAuth: """""" @@ -26,11 +50,12 @@ class DockerBasicAuth: """ @classmethod - def from_dict(cls, value: "DockerBasicAuthDict") -> "Self": + def from_dict(cls, value: 'DockerBasicAuthDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DockerBasicAuthDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DockerBasicAuthDict': + return _transform_to_json_value(self) # type:ignore + class DockerBasicAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/docker_image.py b/python/databricks/bundles/instance_pools/_models/docker_image.py index 15b43d46554..857ed779ba4 100644 --- a/python/databricks/bundles/instance_pools/_models/docker_image.py +++ b/python/databricks/bundles/instance_pools/_models/docker_image.py @@ -1,20 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.instance_pools._models.docker_basic_auth import ( - DockerBasicAuth, - DockerBasicAuthParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DockerImage: """""" @@ -30,11 +50,12 @@ class DockerImage: """ @classmethod - def from_dict(cls, value: "DockerImageDict") -> "Self": + def from_dict(cls, value: 'DockerImageDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DockerImageDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DockerImageDict': + return _transform_to_json_value(self) # type:ignore + class DockerImageDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/gcp_availability.py b/python/databricks/bundles/instance_pools/_models/gcp_availability.py index 0d391b87fe3..452b66695fd 100644 --- a/python/databricks/bundles/instance_pools/_models/gcp_availability.py +++ b/python/databricks/bundles/instance_pools/_models/gcp_availability.py @@ -1,7 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam class GcpAvailability(Enum): @@ -9,13 +36,8 @@ class GcpAvailability(Enum): This field determines whether the instance pool will contain preemptible VMs, on-demand VMs, or preemptible VMs with a fallback to on-demand VMs if the former is unavailable. """ - PREEMPTIBLE_GCP = "PREEMPTIBLE_GCP" ON_DEMAND_GCP = "ON_DEMAND_GCP" PREEMPTIBLE_WITH_FALLBACK_GCP = "PREEMPTIBLE_WITH_FALLBACK_GCP" - -GcpAvailabilityParam = ( - Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] - | GcpAvailability -) +GcpAvailabilityParam = Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] | GcpAvailability diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool.py b/python/databricks/bundles/instance_pools/_models/instance_pool.py index 2b10f210284..2a61a8ae312 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool.py @@ -1,51 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecParam -from databricks.bundles.instance_pools._models.docker_image import ( - DockerImage, - DockerImageParam, -) -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import ( - InstancePoolAwsAttributes, - InstancePoolAwsAttributesParam, -) -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import ( - InstancePoolAzureAttributes, - InstancePoolAzureAttributesParam, -) -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import ( - InstancePoolGcpAttributes, - InstancePoolGcpAttributesParam, -) -from databricks.bundles.instance_pools._models.instance_pool_permission import ( - InstancePoolPermission, - InstancePoolPermissionParam, -) -from databricks.bundles.instance_pools._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.instance_pools._models.node_type_flexibility import ( - NodeTypeFlexibility, - NodeTypeFlexibilityParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InstancePool(Resource): """""" @@ -163,11 +152,12 @@ class InstancePool(Resource): """ @classmethod - def from_dict(cls, value: "InstancePoolDict") -> "Self": + def from_dict(cls, value: 'InstancePoolDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InstancePoolDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InstancePoolDict': + return _transform_to_json_value(self) # type:ignore + class InstancePoolDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes.py b/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes.py index 25fe305787e..ced88bc7faf 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes.py @@ -1,26 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import ( - InstancePoolAwsAttributesAvailability, - InstancePoolAwsAttributesAvailabilityParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InstancePoolAwsAttributes: """ Attributes set during instance pool creation which are related to Amazon Web Services. """ - availability: VariableOrOptional[InstancePoolAwsAttributesAvailability] = None """ Availability type used for the instances in the pool. Supports on-demand, spot, and @@ -80,11 +99,12 @@ class InstancePoolAwsAttributes: """ @classmethod - def from_dict(cls, value: "InstancePoolAwsAttributesDict") -> "Self": + def from_dict(cls, value: 'InstancePoolAwsAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InstancePoolAwsAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InstancePoolAwsAttributesDict': + return _transform_to_json_value(self) # type:ignore + class InstancePoolAwsAttributesDict(TypedDict, total=False): @@ -149,6 +169,4 @@ class InstancePoolAwsAttributesDict(TypedDict, total=False): """ -InstancePoolAwsAttributesParam = ( - InstancePoolAwsAttributesDict | InstancePoolAwsAttributes -) +InstancePoolAwsAttributesParam = InstancePoolAwsAttributesDict | InstancePoolAwsAttributes diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes_availability.py b/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes_availability.py index 5bc8a0350fa..903baed3b5f 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes_availability.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes_availability.py @@ -1,18 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam class InstancePoolAwsAttributesAvailability(Enum): """ The set of AWS availability types supported when setting up nodes for a cluster. """ - SPOT = "SPOT" ON_DEMAND = "ON_DEMAND" - -InstancePoolAwsAttributesAvailabilityParam = ( - Literal["SPOT", "ON_DEMAND"] | InstancePoolAwsAttributesAvailability -) +InstancePoolAwsAttributesAvailabilityParam = Literal["SPOT", "ON_DEMAND"] | InstancePoolAwsAttributesAvailability diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes.py b/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes.py index 80833993289..78ec65ef031 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes.py @@ -1,26 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import ( - InstancePoolAzureAttributesAvailability, - InstancePoolAzureAttributesAvailabilityParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InstancePoolAzureAttributes: """ Attributes set during instance pool creation which are related to Azure. """ - availability: VariableOrOptional[InstancePoolAzureAttributesAvailability] = None """ Availability type used for the instances in the pool. Supports on-demand, spot, and @@ -61,11 +80,12 @@ class InstancePoolAzureAttributes: """ @classmethod - def from_dict(cls, value: "InstancePoolAzureAttributesDict") -> "Self": + def from_dict(cls, value: 'InstancePoolAzureAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InstancePoolAzureAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InstancePoolAzureAttributesDict': + return _transform_to_json_value(self) # type:ignore + class InstancePoolAzureAttributesDict(TypedDict, total=False): @@ -111,6 +131,4 @@ class InstancePoolAzureAttributesDict(TypedDict, total=False): """ -InstancePoolAzureAttributesParam = ( - InstancePoolAzureAttributesDict | InstancePoolAzureAttributes -) +InstancePoolAzureAttributesParam = InstancePoolAzureAttributesDict | InstancePoolAzureAttributes diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes_availability.py b/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes_availability.py index 6d41fe997dd..822fb058e76 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes_availability.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes_availability.py @@ -1,18 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam class InstancePoolAzureAttributesAvailability(Enum): """ The set of Azure availability types supported when setting up nodes for a cluster. """ - SPOT_AZURE = "SPOT_AZURE" ON_DEMAND_AZURE = "ON_DEMAND_AZURE" - -InstancePoolAzureAttributesAvailabilityParam = ( - Literal["SPOT_AZURE", "ON_DEMAND_AZURE"] | InstancePoolAzureAttributesAvailability -) +InstancePoolAzureAttributesAvailabilityParam = Literal["SPOT_AZURE", "ON_DEMAND_AZURE"] | InstancePoolAzureAttributesAvailability diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_gcp_attributes.py b/python/databricks/bundles/instance_pools/_models/instance_pool_gcp_attributes.py index 244158a007d..8a93dfd8c0f 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_gcp_attributes.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_gcp_attributes.py @@ -1,26 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.instance_pools._models.gcp_availability import ( - GcpAvailability, - GcpAvailabilityParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InstancePoolGcpAttributes: """ Attributes set during instance pool creation which are related to GCP. """ - gcp_availability: VariableOrOptional[GcpAvailability] = None local_ssd_count: VariableOrOptional[int] = None @@ -50,11 +69,12 @@ class InstancePoolGcpAttributes: """ @classmethod - def from_dict(cls, value: "InstancePoolGcpAttributesDict") -> "Self": + def from_dict(cls, value: 'InstancePoolGcpAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InstancePoolGcpAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InstancePoolGcpAttributesDict': + return _transform_to_json_value(self) # type:ignore + class InstancePoolGcpAttributesDict(TypedDict, total=False): @@ -89,6 +109,4 @@ class InstancePoolGcpAttributesDict(TypedDict, total=False): """ -InstancePoolGcpAttributesParam = ( - InstancePoolGcpAttributesDict | InstancePoolGcpAttributes -) +InstancePoolGcpAttributesParam = InstancePoolGcpAttributesDict | InstancePoolGcpAttributes diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_permission.py b/python/databricks/bundles/instance_pools/_models/instance_pool_permission.py index a81c2602aa6..dc0996107ef 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_permission.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_permission.py @@ -1,20 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.instance_pools._models.instance_pool_permission_level import ( - InstancePoolPermissionLevel, - InstancePoolPermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InstancePoolPermission: """""" @@ -40,11 +60,12 @@ class InstancePoolPermission: """ @classmethod - def from_dict(cls, value: "InstancePoolPermissionDict") -> "Self": + def from_dict(cls, value: 'InstancePoolPermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InstancePoolPermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InstancePoolPermissionDict': + return _transform_to_json_value(self) # type:ignore + class InstancePoolPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_permission_level.py b/python/databricks/bundles/instance_pools/_models/instance_pool_permission_level.py index bb0468b742c..67ebf6a3989 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_permission_level.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_permission_level.py @@ -1,18 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam class InstancePoolPermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_ATTACH_TO = "CAN_ATTACH_TO" - -InstancePoolPermissionLevelParam = ( - Literal["CAN_MANAGE", "CAN_ATTACH_TO"] | InstancePoolPermissionLevel -) +InstancePoolPermissionLevelParam = Literal["CAN_MANAGE", "CAN_ATTACH_TO"] | InstancePoolPermissionLevel diff --git a/python/databricks/bundles/instance_pools/_models/lifecycle.py b/python/databricks/bundles/instance_pools/_models/lifecycle.py index 697776a198e..3b29b72c225 100644 --- a/python/databricks/bundles/instance_pools/_models/lifecycle.py +++ b/python/databricks/bundles/instance_pools/_models/lifecycle.py @@ -1,16 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +45,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/node_type_flexibility.py b/python/databricks/bundles/instance_pools/_models/node_type_flexibility.py index 418ea9337bd..8dffee2f2cd 100644 --- a/python/databricks/bundles/instance_pools/_models/node_type_flexibility.py +++ b/python/databricks/bundles/instance_pools/_models/node_type_flexibility.py @@ -1,22 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam +from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam +from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam +from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam + +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam + +from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam + +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class NodeTypeFlexibility: """ Configuration for flexible node types, allowing fallback to alternate node types during cluster launch and upscale. """ - alternate_node_type_ids: VariableOrList[str] = field(default_factory=list) """ A list of node type IDs to use as fallbacks when the primary node type is unavailable. @@ -29,11 +52,12 @@ class NodeTypeFlexibility: """ @classmethod - def from_dict(cls, value: "NodeTypeFlexibilityDict") -> "Self": + def from_dict(cls, value: 'NodeTypeFlexibilityDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "NodeTypeFlexibilityDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'NodeTypeFlexibilityDict': + return _transform_to_json_value(self) # type:ignore + class NodeTypeFlexibilityDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/__init__.py b/python/databricks/bundles/job_runs/__init__.py index 093f2201409..2c5f5e1b96c 100644 --- a/python/databricks/bundles/job_runs/__init__.py +++ b/python/databricks/bundles/job_runs/__init__.py @@ -1,48 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "JobRun", - "JobRunDict", - "JobRunLifecycle", - "JobRunLifecycleDict", - "JobRunLifecycleParam", - "JobRunParam", - "JobRunTrigger", - "JobRunTriggerDict", - "JobRunTriggerParam", - "PerformanceTarget", - "PerformanceTargetParam", - "PipelineParams", - "PipelineParamsDict", - "PipelineParamsParam", - "QueueSettings", - "QueueSettingsDict", - "QueueSettingsParam", + 'JobRun', + 'JobRunDict', + 'JobRunLifecycle', + 'JobRunLifecycleDict', + 'JobRunLifecycleParam', + 'JobRunParam', + 'JobRunTrigger', + 'JobRunTriggerDict', + 'JobRunTriggerParam', + 'PerformanceTarget', + 'PerformanceTargetParam', + 'PipelineParams', + 'PipelineParamsDict', + 'PipelineParamsParam', + 'QueueSettings', + 'QueueSettingsDict', + 'QueueSettingsParam', ] +from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam -from databricks.bundles.job_runs._models.job_run_lifecycle import ( - JobRunLifecycle, - JobRunLifecycleDict, - JobRunLifecycleParam, -) -from databricks.bundles.job_runs._models.job_run_trigger import ( - JobRunTrigger, - JobRunTriggerDict, - JobRunTriggerParam, -) -from databricks.bundles.job_runs._models.performance_target import ( - PerformanceTarget, - PerformanceTargetParam, -) -from databricks.bundles.job_runs._models.pipeline_params import ( - PipelineParams, - PipelineParamsDict, - PipelineParamsParam, -) -from databricks.bundles.job_runs._models.queue_settings import ( - QueueSettings, - QueueSettingsDict, - QueueSettingsParam, -) +from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam +from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam +from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam + diff --git a/python/databricks/bundles/job_runs/_models/job_run.py b/python/databricks/bundles/job_runs/_models/job_run.py index e7f5a0b3f71..94dc9e6758c 100644 --- a/python/databricks/bundles/job_runs/_models/job_run.py +++ b/python/databricks/bundles/job_runs/_models/job_run.py @@ -1,38 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.job_runs._models.job_run_lifecycle import ( - JobRunLifecycle, - JobRunLifecycleParam, -) -from databricks.bundles.job_runs._models.performance_target import ( - PerformanceTarget, - PerformanceTargetParam, -) -from databricks.bundles.job_runs._models.pipeline_params import ( - PipelineParams, - PipelineParamsParam, -) -from databricks.bundles.job_runs._models.queue_settings import ( - QueueSettings, - QueueSettingsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam +from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam +from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobRun(Resource): """""" @@ -91,11 +77,12 @@ class JobRun(Resource): """ @classmethod - def from_dict(cls, value: "JobRunDict") -> "Self": + def from_dict(cls, value: 'JobRunDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobRunDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobRunDict': + return _transform_to_json_value(self) # type:ignore + class JobRunDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/_models/job_run_lifecycle.py b/python/databricks/bundles/job_runs/_models/job_run_lifecycle.py index cb673b0245d..0b99f0f5d61 100644 --- a/python/databricks/bundles/job_runs/_models/job_run_lifecycle.py +++ b/python/databricks/bundles/job_runs/_models/job_run_lifecycle.py @@ -1,20 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.job_runs._models.job_run_trigger import ( - JobRunTrigger, - JobRunTriggerParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam +from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam +from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobRunLifecycle: """""" @@ -30,11 +34,12 @@ class JobRunLifecycle: """ @classmethod - def from_dict(cls, value: "JobRunLifecycleDict") -> "Self": + def from_dict(cls, value: 'JobRunLifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobRunLifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobRunLifecycleDict': + return _transform_to_json_value(self) # type:ignore + class JobRunLifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/_models/job_run_trigger.py b/python/databricks/bundles/job_runs/_models/job_run_trigger.py index 75bdf970a4e..5bac797649c 100644 --- a/python/databricks/bundles/job_runs/_models/job_run_trigger.py +++ b/python/databricks/bundles/job_runs/_models/job_run_trigger.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam +from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam +from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobRunTrigger: """""" @@ -26,11 +34,12 @@ class JobRunTrigger: """ @classmethod - def from_dict(cls, value: "JobRunTriggerDict") -> "Self": + def from_dict(cls, value: 'JobRunTriggerDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobRunTriggerDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobRunTriggerDict': + return _transform_to_json_value(self) # type:ignore + class JobRunTriggerDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/_models/performance_target.py b/python/databricks/bundles/job_runs/_models/performance_target.py index 8dbe7e4a435..43cef1b8afa 100644 --- a/python/databricks/bundles/job_runs/_models/performance_target.py +++ b/python/databricks/bundles/job_runs/_models/performance_target.py @@ -1,8 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam +from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam +from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam class PerformanceTarget(Enum): """ @@ -10,11 +21,7 @@ class PerformanceTarget(Enum): The performance mode on the job or pipeline should map to a performance setting that is passed to Cluster Manager (see cluster-common PerformanceTarget). """ - PERFORMANCE_OPTIMIZED = "PERFORMANCE_OPTIMIZED" STANDARD = "STANDARD" - -PerformanceTargetParam = ( - Literal["PERFORMANCE_OPTIMIZED", "STANDARD"] | PerformanceTarget -) +PerformanceTargetParam = Literal["PERFORMANCE_OPTIMIZED", "STANDARD"] | PerformanceTarget diff --git a/python/databricks/bundles/job_runs/_models/pipeline_params.py b/python/databricks/bundles/job_runs/_models/pipeline_params.py index 98550f271fb..fe3bf70db82 100644 --- a/python/databricks/bundles/job_runs/_models/pipeline_params.py +++ b/python/databricks/bundles/job_runs/_models/pipeline_params.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam +from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam +from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam +from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PipelineParams: """""" @@ -42,11 +50,12 @@ class PipelineParams: """ @classmethod - def from_dict(cls, value: "PipelineParamsDict") -> "Self": + def from_dict(cls, value: 'PipelineParamsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PipelineParamsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PipelineParamsDict': + return _transform_to_json_value(self) # type:ignore + class PipelineParamsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/_models/queue_settings.py b/python/databricks/bundles/job_runs/_models/queue_settings.py index a72921aed96..a6bca6701b7 100644 --- a/python/databricks/bundles/job_runs/_models/queue_settings.py +++ b/python/databricks/bundles/job_runs/_models/queue_settings.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam +from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam +from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam +from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class QueueSettings: """""" @@ -21,11 +29,12 @@ class QueueSettings: """ @classmethod - def from_dict(cls, value: "QueueSettingsDict") -> "Self": + def from_dict(cls, value: 'QueueSettingsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "QueueSettingsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'QueueSettingsDict': + return _transform_to_json_value(self) # type:ignore + class QueueSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/__init__.py b/python/databricks/bundles/jobs/__init__.py index 1df8be62977..e3fc1309b09 100644 --- a/python/databricks/bundles/jobs/__init__.py +++ b/python/databricks/bundles/jobs/__init__.py @@ -1,927 +1,524 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Adlsgen2Info", - "Adlsgen2InfoDict", - "Adlsgen2InfoParam", - "AiRuntimeTask", - "AiRuntimeTaskDict", - "AiRuntimeTaskParam", - "AiRuntimeTaskPriorityClass", - "AiRuntimeTaskPriorityClassParam", - "AlertTask", - "AlertTaskDict", - "AlertTaskParam", - "AlertTaskSubscriber", - "AlertTaskSubscriberDict", - "AlertTaskSubscriberParam", - "AuthenticationMethod", - "AuthenticationMethodParam", - "AutoScale", - "AutoScaleDict", - "AutoScaleParam", - "AwsAttributes", - "AwsAttributesDict", - "AwsAttributesParam", - "AwsAvailability", - "AwsAvailabilityParam", - "AzureAttributes", - "AzureAttributesDict", - "AzureAttributesParam", - "AzureAvailability", - "AzureAvailabilityParam", - "CleanRoomsNotebookTask", - "CleanRoomsNotebookTaskDict", - "CleanRoomsNotebookTaskParam", - "ClientsTypes", - "ClientsTypesDict", - "ClientsTypesParam", - "ClusterLogConf", - "ClusterLogConfDict", - "ClusterLogConfParam", - "ClusterSpec", - "ClusterSpecDict", - "ClusterSpecParam", - "Compute", - "ComputeConfig", - "ComputeConfigDict", - "ComputeConfigParam", - "ComputeDict", - "ComputeParam", - "ComputeSpec", - "ComputeSpecAcceleratorType", - "ComputeSpecAcceleratorTypeParam", - "ComputeSpecDict", - "ComputeSpecParam", - "Condition", - "ConditionParam", - "ConditionTask", - "ConditionTaskDict", - "ConditionTaskOp", - "ConditionTaskOpParam", - "ConditionTaskParam", - "ConfidentialComputeType", - "ConfidentialComputeTypeParam", - "Continuous", - "ContinuousDict", - "ContinuousParam", - "ContinuousTriggerConfiguration", - "ContinuousTriggerConfigurationDict", - "ContinuousTriggerConfigurationParam", - "CronSchedule", - "CronScheduleDict", - "CronScheduleParam", - "CronTriggerConfiguration", - "CronTriggerConfigurationDict", - "CronTriggerConfigurationParam", - "DashboardTask", - "DashboardTaskDict", - "DashboardTaskParam", - "DataSecurityMode", - "DataSecurityModeParam", - "DayOfWeek", - "DayOfWeekParam", - "DbfsStorageInfo", - "DbfsStorageInfoDict", - "DbfsStorageInfoParam", - "DbtPlatformTask", - "DbtPlatformTaskDict", - "DbtPlatformTaskParam", - "DbtTask", - "DbtTaskDict", - "DbtTaskParam", - "DependencyMode", - "DependencyModeParam", - "DeploymentSpec", - "DeploymentSpecDict", - "DeploymentSpecParam", - "DockerBasicAuth", - "DockerBasicAuthDict", - "DockerBasicAuthParam", - "DockerImage", - "DockerImageDict", - "DockerImageParam", - "EbsVolumeType", - "EbsVolumeTypeParam", - "Environment", - "EnvironmentDict", - "EnvironmentParam", - "FileArrivalTriggerConfiguration", - "FileArrivalTriggerConfigurationDict", - "FileArrivalTriggerConfigurationParam", - "ForEachTask", - "ForEachTaskDict", - "ForEachTaskParam", - "GcpAttributes", - "GcpAttributesDict", - "GcpAttributesParam", - "GcpAvailability", - "GcpAvailabilityParam", - "GcsStorageInfo", - "GcsStorageInfoDict", - "GcsStorageInfoParam", - "GenAiComputeTask", - "GenAiComputeTaskDict", - "GenAiComputeTaskParam", - "GitProvider", - "GitProviderParam", - "GitSource", - "GitSourceDict", - "GitSourceParam", - "HardwareAcceleratorType", - "HardwareAcceleratorTypeParam", - "InitScriptInfo", - "InitScriptInfoDict", - "InitScriptInfoParam", - "Job", - "JobCluster", - "JobClusterDict", - "JobClusterParam", - "JobDict", - "JobEmailNotifications", - "JobEmailNotificationsDict", - "JobEmailNotificationsParam", - "JobEnvironment", - "JobEnvironmentDict", - "JobEnvironmentParam", - "JobEnvironmentVariables", - "JobEnvironmentVariablesDict", - "JobEnvironmentVariablesParam", - "JobEnvironmentVariablesSpec", - "JobEnvironmentVariablesSpecDict", - "JobEnvironmentVariablesSpecParam", - "JobNotificationSettings", - "JobNotificationSettingsDict", - "JobNotificationSettingsParam", - "JobParam", - "JobParameterDefinition", - "JobParameterDefinitionDict", - "JobParameterDefinitionParam", - "JobPermission", - "JobPermissionDict", - "JobPermissionLevel", - "JobPermissionLevelParam", - "JobPermissionParam", - "JobRunAs", - "JobRunAsDict", - "JobRunAsParam", - "JobsHealthMetric", - "JobsHealthMetricParam", - "JobsHealthOperator", - "JobsHealthOperatorParam", - "JobsHealthRule", - "JobsHealthRuleDict", - "JobsHealthRuleParam", - "JobsHealthRules", - "JobsHealthRulesDict", - "JobsHealthRulesParam", - "Kind", - "KindParam", - "Library", - "LibraryDict", - "LibraryParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "LocalFileInfo", - "LocalFileInfoDict", - "LocalFileInfoParam", - "LogAnalyticsInfo", - "LogAnalyticsInfoDict", - "LogAnalyticsInfoParam", - "MaintenanceWindow", - "MaintenanceWindowDict", - "MaintenanceWindowParam", - "MavenLibrary", - "MavenLibraryDict", - "MavenLibraryParam", - "ModelTriggerConfiguration", - "ModelTriggerConfigurationCondition", - "ModelTriggerConfigurationConditionParam", - "ModelTriggerConfigurationDict", - "ModelTriggerConfigurationParam", - "NodeTypeFlexibility", - "NodeTypeFlexibilityDict", - "NodeTypeFlexibilityParam", - "NotebookTask", - "NotebookTaskDict", - "NotebookTaskParam", - "PauseStatus", - "PauseStatusParam", - "PerformanceTarget", - "PerformanceTargetParam", - "PeriodicTriggerConfiguration", - "PeriodicTriggerConfigurationDict", - "PeriodicTriggerConfigurationParam", - "PeriodicTriggerConfigurationTimeUnit", - "PeriodicTriggerConfigurationTimeUnitParam", - "PipelineParams", - "PipelineParamsDict", - "PipelineParamsParam", - "PipelineTask", - "PipelineTaskDict", - "PipelineTaskParam", - "PowerBiModel", - "PowerBiModelDict", - "PowerBiModelParam", - "PowerBiTable", - "PowerBiTableDict", - "PowerBiTableParam", - "PowerBiTask", - "PowerBiTaskDict", - "PowerBiTaskParam", - "PythonOperatorTask", - "PythonOperatorTaskDict", - "PythonOperatorTaskParam", - "PythonOperatorTaskParameter", - "PythonOperatorTaskParameterDict", - "PythonOperatorTaskParameterParam", - "PythonPyPiLibrary", - "PythonPyPiLibraryDict", - "PythonPyPiLibraryParam", - "PythonWheelTask", - "PythonWheelTaskDict", - "PythonWheelTaskParam", - "QueueSettings", - "QueueSettingsDict", - "QueueSettingsParam", - "RCranLibrary", - "RCranLibraryDict", - "RCranLibraryParam", - "RunIf", - "RunIfParam", - "RunJobTask", - "RunJobTaskDict", - "RunJobTaskParam", - "RuntimeEngine", - "RuntimeEngineParam", - "S3StorageInfo", - "S3StorageInfoDict", - "S3StorageInfoParam", - "Source", - "SourceParam", - "SparkJarTask", - "SparkJarTaskDict", - "SparkJarTaskParam", - "SparkPythonTask", - "SparkPythonTaskDict", - "SparkPythonTaskParam", - "SparkSubmitTask", - "SparkSubmitTaskDict", - "SparkSubmitTaskParam", - "SparseCheckout", - "SparseCheckoutDict", - "SparseCheckoutParam", - "SqlConditionConfiguration", - "SqlConditionConfigurationDict", - "SqlConditionConfigurationParam", - "SqlConditionTriggerMode", - "SqlConditionTriggerModeParam", - "SqlTask", - "SqlTaskAlert", - "SqlTaskAlertDict", - "SqlTaskAlertParam", - "SqlTaskDashboard", - "SqlTaskDashboardDict", - "SqlTaskDashboardParam", - "SqlTaskDict", - "SqlTaskFile", - "SqlTaskFileDict", - "SqlTaskFileParam", - "SqlTaskParam", - "SqlTaskQuery", - "SqlTaskQueryDict", - "SqlTaskQueryParam", - "SqlTaskSubscription", - "SqlTaskSubscriptionDict", - "SqlTaskSubscriptionParam", - "StorageMode", - "StorageModeParam", - "Subscription", - "SubscriptionAttachment", - "SubscriptionAttachmentDataAttachment", - "SubscriptionAttachmentDataAttachmentDict", - "SubscriptionAttachmentDataAttachmentFileFormat", - "SubscriptionAttachmentDataAttachmentFileFormatParam", - "SubscriptionAttachmentDataAttachmentParam", - "SubscriptionAttachmentDataAttachmentWidget", - "SubscriptionAttachmentDataAttachmentWidgetDict", - "SubscriptionAttachmentDataAttachmentWidgetParam", - "SubscriptionAttachmentDict", - "SubscriptionAttachmentPageAttachment", - "SubscriptionAttachmentPageAttachmentDict", - "SubscriptionAttachmentPageAttachmentPageSelection", - "SubscriptionAttachmentPageAttachmentPageSelectionParam", - "SubscriptionAttachmentPageAttachmentParam", - "SubscriptionAttachmentParam", - "SubscriptionDict", - "SubscriptionParam", - "SubscriptionSubscriber", - "SubscriptionSubscriberDict", - "SubscriptionSubscriberParam", - "TableUpdateTriggerConfiguration", - "TableUpdateTriggerConfigurationDict", - "TableUpdateTriggerConfigurationParam", - "Task", - "TaskDependency", - "TaskDependencyDict", - "TaskDependencyParam", - "TaskDict", - "TaskEmailNotifications", - "TaskEmailNotificationsDict", - "TaskEmailNotificationsParam", - "TaskNotificationSettings", - "TaskNotificationSettingsDict", - "TaskNotificationSettingsParam", - "TaskParam", - "TaskRetryMode", - "TaskRetryModeParam", - "TriggerConfiguration", - "TriggerConfigurationDict", - "TriggerConfigurationParam", - "TriggerSettings", - "TriggerSettingsDict", - "TriggerSettingsParam", - "VolumesStorageInfo", - "VolumesStorageInfoDict", - "VolumesStorageInfoParam", - "Webhook", - "WebhookDict", - "WebhookNotifications", - "WebhookNotificationsDict", - "WebhookNotificationsParam", - "WebhookParam", - "WorkloadType", - "WorkloadTypeDict", - "WorkloadTypeParam", - "WorkspaceStorageInfo", - "WorkspaceStorageInfoDict", - "WorkspaceStorageInfoParam", + 'Adlsgen2Info', + 'Adlsgen2InfoDict', + 'Adlsgen2InfoParam', + 'AiRuntimeTask', + 'AiRuntimeTaskDict', + 'AiRuntimeTaskParam', + 'AiRuntimeTaskPriorityClass', + 'AiRuntimeTaskPriorityClassParam', + 'AlertTask', + 'AlertTaskDict', + 'AlertTaskParam', + 'AlertTaskSubscriber', + 'AlertTaskSubscriberDict', + 'AlertTaskSubscriberParam', + 'AuthenticationMethod', + 'AuthenticationMethodParam', + 'AutoScale', + 'AutoScaleDict', + 'AutoScaleParam', + 'AwsAttributes', + 'AwsAttributesDict', + 'AwsAttributesParam', + 'AwsAvailability', + 'AwsAvailabilityParam', + 'AzureAttributes', + 'AzureAttributesDict', + 'AzureAttributesParam', + 'AzureAvailability', + 'AzureAvailabilityParam', + 'CleanRoomsNotebookTask', + 'CleanRoomsNotebookTaskDict', + 'CleanRoomsNotebookTaskParam', + 'ClientsTypes', + 'ClientsTypesDict', + 'ClientsTypesParam', + 'ClusterLogConf', + 'ClusterLogConfDict', + 'ClusterLogConfParam', + 'ClusterSpec', + 'ClusterSpecDict', + 'ClusterSpecParam', + 'Compute', + 'ComputeConfig', + 'ComputeConfigDict', + 'ComputeConfigParam', + 'ComputeDict', + 'ComputeParam', + 'ComputeSpec', + 'ComputeSpecAcceleratorType', + 'ComputeSpecAcceleratorTypeParam', + 'ComputeSpecDict', + 'ComputeSpecParam', + 'Condition', + 'ConditionParam', + 'ConditionTask', + 'ConditionTaskDict', + 'ConditionTaskOp', + 'ConditionTaskOpParam', + 'ConditionTaskParam', + 'ConfidentialComputeType', + 'ConfidentialComputeTypeParam', + 'Continuous', + 'ContinuousDict', + 'ContinuousParam', + 'ContinuousTriggerConfiguration', + 'ContinuousTriggerConfigurationDict', + 'ContinuousTriggerConfigurationParam', + 'CronSchedule', + 'CronScheduleDict', + 'CronScheduleParam', + 'CronTriggerConfiguration', + 'CronTriggerConfigurationDict', + 'CronTriggerConfigurationParam', + 'DashboardTask', + 'DashboardTaskDict', + 'DashboardTaskParam', + 'DataSecurityMode', + 'DataSecurityModeParam', + 'DayOfWeek', + 'DayOfWeekParam', + 'DbfsStorageInfo', + 'DbfsStorageInfoDict', + 'DbfsStorageInfoParam', + 'DbtPlatformTask', + 'DbtPlatformTaskDict', + 'DbtPlatformTaskParam', + 'DbtTask', + 'DbtTaskDict', + 'DbtTaskParam', + 'DependencyMode', + 'DependencyModeParam', + 'DeploymentSpec', + 'DeploymentSpecDict', + 'DeploymentSpecParam', + 'DockerBasicAuth', + 'DockerBasicAuthDict', + 'DockerBasicAuthParam', + 'DockerImage', + 'DockerImageDict', + 'DockerImageParam', + 'EbsVolumeType', + 'EbsVolumeTypeParam', + 'Environment', + 'EnvironmentDict', + 'EnvironmentParam', + 'FileArrivalTriggerConfiguration', + 'FileArrivalTriggerConfigurationDict', + 'FileArrivalTriggerConfigurationParam', + 'ForEachTask', + 'ForEachTaskDict', + 'ForEachTaskParam', + 'GcpAttributes', + 'GcpAttributesDict', + 'GcpAttributesParam', + 'GcpAvailability', + 'GcpAvailabilityParam', + 'GcsStorageInfo', + 'GcsStorageInfoDict', + 'GcsStorageInfoParam', + 'GenAiComputeTask', + 'GenAiComputeTaskDict', + 'GenAiComputeTaskParam', + 'GitProvider', + 'GitProviderParam', + 'GitSource', + 'GitSourceDict', + 'GitSourceParam', + 'HardwareAcceleratorType', + 'HardwareAcceleratorTypeParam', + 'InitScriptInfo', + 'InitScriptInfoDict', + 'InitScriptInfoParam', + 'Job', + 'JobCluster', + 'JobClusterDict', + 'JobClusterParam', + 'JobDict', + 'JobEmailNotifications', + 'JobEmailNotificationsDict', + 'JobEmailNotificationsParam', + 'JobEnvironment', + 'JobEnvironmentDict', + 'JobEnvironmentParam', + 'JobEnvironmentVariables', + 'JobEnvironmentVariablesDict', + 'JobEnvironmentVariablesParam', + 'JobEnvironmentVariablesSpec', + 'JobEnvironmentVariablesSpecDict', + 'JobEnvironmentVariablesSpecParam', + 'JobNotificationSettings', + 'JobNotificationSettingsDict', + 'JobNotificationSettingsParam', + 'JobParam', + 'JobParameterDefinition', + 'JobParameterDefinitionDict', + 'JobParameterDefinitionParam', + 'JobPermission', + 'JobPermissionDict', + 'JobPermissionLevel', + 'JobPermissionLevelParam', + 'JobPermissionParam', + 'JobRunAs', + 'JobRunAsDict', + 'JobRunAsParam', + 'JobsHealthMetric', + 'JobsHealthMetricParam', + 'JobsHealthOperator', + 'JobsHealthOperatorParam', + 'JobsHealthRule', + 'JobsHealthRuleDict', + 'JobsHealthRuleParam', + 'JobsHealthRules', + 'JobsHealthRulesDict', + 'JobsHealthRulesParam', + 'Kind', + 'KindParam', + 'Library', + 'LibraryDict', + 'LibraryParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'LocalFileInfo', + 'LocalFileInfoDict', + 'LocalFileInfoParam', + 'LogAnalyticsInfo', + 'LogAnalyticsInfoDict', + 'LogAnalyticsInfoParam', + 'MaintenanceWindow', + 'MaintenanceWindowDict', + 'MaintenanceWindowParam', + 'MavenLibrary', + 'MavenLibraryDict', + 'MavenLibraryParam', + 'ModelTriggerConfiguration', + 'ModelTriggerConfigurationCondition', + 'ModelTriggerConfigurationConditionParam', + 'ModelTriggerConfigurationDict', + 'ModelTriggerConfigurationParam', + 'NodeTypeFlexibility', + 'NodeTypeFlexibilityDict', + 'NodeTypeFlexibilityParam', + 'NotebookTask', + 'NotebookTaskDict', + 'NotebookTaskParam', + 'PauseStatus', + 'PauseStatusParam', + 'PerformanceTarget', + 'PerformanceTargetParam', + 'PeriodicTriggerConfiguration', + 'PeriodicTriggerConfigurationDict', + 'PeriodicTriggerConfigurationParam', + 'PeriodicTriggerConfigurationTimeUnit', + 'PeriodicTriggerConfigurationTimeUnitParam', + 'PipelineParams', + 'PipelineParamsDict', + 'PipelineParamsParam', + 'PipelineTask', + 'PipelineTaskDict', + 'PipelineTaskParam', + 'PowerBiModel', + 'PowerBiModelDict', + 'PowerBiModelParam', + 'PowerBiTable', + 'PowerBiTableDict', + 'PowerBiTableParam', + 'PowerBiTask', + 'PowerBiTaskDict', + 'PowerBiTaskParam', + 'PythonOperatorTask', + 'PythonOperatorTaskDict', + 'PythonOperatorTaskParam', + 'PythonOperatorTaskParameter', + 'PythonOperatorTaskParameterDict', + 'PythonOperatorTaskParameterParam', + 'PythonPyPiLibrary', + 'PythonPyPiLibraryDict', + 'PythonPyPiLibraryParam', + 'PythonWheelTask', + 'PythonWheelTaskDict', + 'PythonWheelTaskParam', + 'QueueSettings', + 'QueueSettingsDict', + 'QueueSettingsParam', + 'RCranLibrary', + 'RCranLibraryDict', + 'RCranLibraryParam', + 'RunIf', + 'RunIfParam', + 'RunJobTask', + 'RunJobTaskDict', + 'RunJobTaskParam', + 'RuntimeEngine', + 'RuntimeEngineParam', + 'S3StorageInfo', + 'S3StorageInfoDict', + 'S3StorageInfoParam', + 'Source', + 'SourceParam', + 'SparkJarTask', + 'SparkJarTaskDict', + 'SparkJarTaskParam', + 'SparkPythonTask', + 'SparkPythonTaskDict', + 'SparkPythonTaskParam', + 'SparkSubmitTask', + 'SparkSubmitTaskDict', + 'SparkSubmitTaskParam', + 'SparseCheckout', + 'SparseCheckoutDict', + 'SparseCheckoutParam', + 'SqlConditionConfiguration', + 'SqlConditionConfigurationDict', + 'SqlConditionConfigurationParam', + 'SqlConditionTriggerMode', + 'SqlConditionTriggerModeParam', + 'SqlTask', + 'SqlTaskAlert', + 'SqlTaskAlertDict', + 'SqlTaskAlertParam', + 'SqlTaskDashboard', + 'SqlTaskDashboardDict', + 'SqlTaskDashboardParam', + 'SqlTaskDict', + 'SqlTaskFile', + 'SqlTaskFileDict', + 'SqlTaskFileParam', + 'SqlTaskParam', + 'SqlTaskQuery', + 'SqlTaskQueryDict', + 'SqlTaskQueryParam', + 'SqlTaskSubscription', + 'SqlTaskSubscriptionDict', + 'SqlTaskSubscriptionParam', + 'StorageMode', + 'StorageModeParam', + 'Subscription', + 'SubscriptionAttachment', + 'SubscriptionAttachmentDataAttachment', + 'SubscriptionAttachmentDataAttachmentDict', + 'SubscriptionAttachmentDataAttachmentFileFormat', + 'SubscriptionAttachmentDataAttachmentFileFormatParam', + 'SubscriptionAttachmentDataAttachmentParam', + 'SubscriptionAttachmentDataAttachmentWidget', + 'SubscriptionAttachmentDataAttachmentWidgetDict', + 'SubscriptionAttachmentDataAttachmentWidgetParam', + 'SubscriptionAttachmentDict', + 'SubscriptionAttachmentPageAttachment', + 'SubscriptionAttachmentPageAttachmentDict', + 'SubscriptionAttachmentPageAttachmentPageSelection', + 'SubscriptionAttachmentPageAttachmentPageSelectionParam', + 'SubscriptionAttachmentPageAttachmentParam', + 'SubscriptionAttachmentParam', + 'SubscriptionDict', + 'SubscriptionParam', + 'SubscriptionSubscriber', + 'SubscriptionSubscriberDict', + 'SubscriptionSubscriberParam', + 'TableUpdateTriggerConfiguration', + 'TableUpdateTriggerConfigurationDict', + 'TableUpdateTriggerConfigurationParam', + 'Task', + 'TaskDependency', + 'TaskDependencyDict', + 'TaskDependencyParam', + 'TaskDict', + 'TaskEmailNotifications', + 'TaskEmailNotificationsDict', + 'TaskEmailNotificationsParam', + 'TaskNotificationSettings', + 'TaskNotificationSettingsDict', + 'TaskNotificationSettingsParam', + 'TaskParam', + 'TaskRetryMode', + 'TaskRetryModeParam', + 'TriggerConfiguration', + 'TriggerConfigurationDict', + 'TriggerConfigurationParam', + 'TriggerSettings', + 'TriggerSettingsDict', + 'TriggerSettingsParam', + 'VolumesStorageInfo', + 'VolumesStorageInfoDict', + 'VolumesStorageInfoParam', + 'Webhook', + 'WebhookDict', + 'WebhookNotifications', + 'WebhookNotificationsDict', + 'WebhookNotificationsParam', + 'WebhookParam', + 'WorkloadType', + 'WorkloadTypeDict', + 'WorkloadTypeParam', + 'WorkspaceStorageInfo', + 'WorkspaceStorageInfoDict', + 'WorkspaceStorageInfoParam', ] -from databricks.bundles.jobs._models.adlsgen2_info import ( - Adlsgen2Info, - Adlsgen2InfoDict, - Adlsgen2InfoParam, -) -from databricks.bundles.jobs._models.ai_runtime_task import ( - AiRuntimeTask, - AiRuntimeTaskDict, - AiRuntimeTaskParam, -) -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import ( - AiRuntimeTaskPriorityClass, - AiRuntimeTaskPriorityClassParam, -) -from databricks.bundles.jobs._models.alert_task import ( - AlertTask, - AlertTaskDict, - AlertTaskParam, -) -from databricks.bundles.jobs._models.alert_task_subscriber import ( - AlertTaskSubscriber, - AlertTaskSubscriberDict, - AlertTaskSubscriberParam, -) -from databricks.bundles.jobs._models.authentication_method import ( - AuthenticationMethod, - AuthenticationMethodParam, -) -from databricks.bundles.jobs._models.auto_scale import ( - AutoScale, - AutoScaleDict, - AutoScaleParam, -) -from databricks.bundles.jobs._models.aws_attributes import ( - AwsAttributes, - AwsAttributesDict, - AwsAttributesParam, -) -from databricks.bundles.jobs._models.aws_availability import ( - AwsAvailability, - AwsAvailabilityParam, -) -from databricks.bundles.jobs._models.azure_attributes import ( - AzureAttributes, - AzureAttributesDict, - AzureAttributesParam, -) -from databricks.bundles.jobs._models.azure_availability import ( - AzureAvailability, - AzureAvailabilityParam, -) -from databricks.bundles.jobs._models.clean_rooms_notebook_task import ( - CleanRoomsNotebookTask, - CleanRoomsNotebookTaskDict, - CleanRoomsNotebookTaskParam, -) -from databricks.bundles.jobs._models.clients_types import ( - ClientsTypes, - ClientsTypesDict, - ClientsTypesParam, -) -from databricks.bundles.jobs._models.cluster_log_conf import ( - ClusterLogConf, - ClusterLogConfDict, - ClusterLogConfParam, -) -from databricks.bundles.jobs._models.cluster_spec import ( - ClusterSpec, - ClusterSpecDict, - ClusterSpecParam, -) +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ( - ComputeConfig, - ComputeConfigDict, - ComputeConfigParam, -) -from databricks.bundles.jobs._models.compute_spec import ( - ComputeSpec, - ComputeSpecDict, - ComputeSpecParam, -) -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ( - ComputeSpecAcceleratorType, - ComputeSpecAcceleratorTypeParam, -) -from databricks.bundles.jobs._models.condition import Condition, ConditionParam -from databricks.bundles.jobs._models.condition_task import ( - ConditionTask, - ConditionTaskDict, - ConditionTaskParam, -) -from databricks.bundles.jobs._models.condition_task_op import ( - ConditionTaskOp, - ConditionTaskOpParam, -) -from databricks.bundles.jobs._models.confidential_compute_type import ( - ConfidentialComputeType, - ConfidentialComputeTypeParam, -) -from databricks.bundles.jobs._models.continuous import ( - Continuous, - ContinuousDict, - ContinuousParam, -) -from databricks.bundles.jobs._models.continuous_trigger_configuration import ( - ContinuousTriggerConfiguration, - ContinuousTriggerConfigurationDict, - ContinuousTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.cron_schedule import ( - CronSchedule, - CronScheduleDict, - CronScheduleParam, -) -from databricks.bundles.jobs._models.cron_trigger_configuration import ( - CronTriggerConfiguration, - CronTriggerConfigurationDict, - CronTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.dashboard_task import ( - DashboardTask, - DashboardTaskDict, - DashboardTaskParam, -) -from databricks.bundles.jobs._models.data_security_mode import ( - DataSecurityMode, - DataSecurityModeParam, -) -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam -from databricks.bundles.jobs._models.dbfs_storage_info import ( - DbfsStorageInfo, - DbfsStorageInfoDict, - DbfsStorageInfoParam, -) -from databricks.bundles.jobs._models.dbt_platform_task import ( - DbtPlatformTask, - DbtPlatformTaskDict, - DbtPlatformTaskParam, -) +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.dependency_mode import ( - DependencyMode, - DependencyModeParam, -) -from databricks.bundles.jobs._models.deployment_spec import ( - DeploymentSpec, - DeploymentSpecDict, - DeploymentSpecParam, -) -from databricks.bundles.jobs._models.docker_basic_auth import ( - DockerBasicAuth, - DockerBasicAuthDict, - DockerBasicAuthParam, -) -from databricks.bundles.jobs._models.docker_image import ( - DockerImage, - DockerImageDict, - DockerImageParam, -) -from databricks.bundles.jobs._models.ebs_volume_type import ( - EbsVolumeType, - EbsVolumeTypeParam, -) -from databricks.bundles.jobs._models.environment import ( - Environment, - EnvironmentDict, - EnvironmentParam, -) -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import ( - FileArrivalTriggerConfiguration, - FileArrivalTriggerConfigurationDict, - FileArrivalTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.for_each_task import ( - ForEachTask, - ForEachTaskDict, - ForEachTaskParam, -) -from databricks.bundles.jobs._models.gcp_attributes import ( - GcpAttributes, - GcpAttributesDict, - GcpAttributesParam, -) -from databricks.bundles.jobs._models.gcp_availability import ( - GcpAvailability, - GcpAvailabilityParam, -) -from databricks.bundles.jobs._models.gcs_storage_info import ( - GcsStorageInfo, - GcsStorageInfoDict, - GcsStorageInfoParam, -) -from databricks.bundles.jobs._models.gen_ai_compute_task import ( - GenAiComputeTask, - GenAiComputeTaskDict, - GenAiComputeTaskParam, -) -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam -from databricks.bundles.jobs._models.git_source import ( - GitSource, - GitSourceDict, - GitSourceParam, -) -from databricks.bundles.jobs._models.hardware_accelerator_type import ( - HardwareAcceleratorType, - HardwareAcceleratorTypeParam, -) -from databricks.bundles.jobs._models.init_script_info import ( - InitScriptInfo, - InitScriptInfoDict, - InitScriptInfoParam, -) +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_cluster import ( - JobCluster, - JobClusterDict, - JobClusterParam, -) -from databricks.bundles.jobs._models.job_email_notifications import ( - JobEmailNotifications, - JobEmailNotificationsDict, - JobEmailNotificationsParam, -) -from databricks.bundles.jobs._models.job_environment import ( - JobEnvironment, - JobEnvironmentDict, - JobEnvironmentParam, -) -from databricks.bundles.jobs._models.job_environment_variables import ( - JobEnvironmentVariables, - JobEnvironmentVariablesDict, - JobEnvironmentVariablesParam, -) -from databricks.bundles.jobs._models.job_environment_variables_spec import ( - JobEnvironmentVariablesSpec, - JobEnvironmentVariablesSpecDict, - JobEnvironmentVariablesSpecParam, -) -from databricks.bundles.jobs._models.job_notification_settings import ( - JobNotificationSettings, - JobNotificationSettingsDict, - JobNotificationSettingsParam, -) -from databricks.bundles.jobs._models.job_parameter_definition import ( - JobParameterDefinition, - JobParameterDefinitionDict, - JobParameterDefinitionParam, -) -from databricks.bundles.jobs._models.job_permission import ( - JobPermission, - JobPermissionDict, - JobPermissionParam, -) -from databricks.bundles.jobs._models.job_permission_level import ( - JobPermissionLevel, - JobPermissionLevelParam, -) -from databricks.bundles.jobs._models.job_run_as import ( - JobRunAs, - JobRunAsDict, - JobRunAsParam, -) -from databricks.bundles.jobs._models.jobs_health_metric import ( - JobsHealthMetric, - JobsHealthMetricParam, -) -from databricks.bundles.jobs._models.jobs_health_operator import ( - JobsHealthOperator, - JobsHealthOperatorParam, -) -from databricks.bundles.jobs._models.jobs_health_rule import ( - JobsHealthRule, - JobsHealthRuleDict, - JobsHealthRuleParam, -) -from databricks.bundles.jobs._models.jobs_health_rules import ( - JobsHealthRules, - JobsHealthRulesDict, - JobsHealthRulesParam, -) +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + from databricks.bundles.jobs._models.kind import Kind, KindParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.jobs._models.local_file_info import ( - LocalFileInfo, - LocalFileInfoDict, - LocalFileInfoParam, -) -from databricks.bundles.jobs._models.log_analytics_info import ( - LogAnalyticsInfo, - LogAnalyticsInfoDict, - LogAnalyticsInfoParam, -) -from databricks.bundles.jobs._models.maintenance_window import ( - MaintenanceWindow, - MaintenanceWindowDict, - MaintenanceWindowParam, -) -from databricks.bundles.jobs._models.maven_library import ( - MavenLibrary, - MavenLibraryDict, - MavenLibraryParam, -) -from databricks.bundles.jobs._models.model_trigger_configuration import ( - ModelTriggerConfiguration, - ModelTriggerConfigurationDict, - ModelTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ( - ModelTriggerConfigurationCondition, - ModelTriggerConfigurationConditionParam, -) -from databricks.bundles.jobs._models.node_type_flexibility import ( - NodeTypeFlexibility, - NodeTypeFlexibilityDict, - NodeTypeFlexibilityParam, -) -from databricks.bundles.jobs._models.notebook_task import ( - NotebookTask, - NotebookTaskDict, - NotebookTaskParam, -) + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam -from databricks.bundles.jobs._models.performance_target import ( - PerformanceTarget, - PerformanceTargetParam, -) -from databricks.bundles.jobs._models.periodic_trigger_configuration import ( - PeriodicTriggerConfiguration, - PeriodicTriggerConfigurationDict, - PeriodicTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import ( - PeriodicTriggerConfigurationTimeUnit, - PeriodicTriggerConfigurationTimeUnitParam, -) -from databricks.bundles.jobs._models.pipeline_params import ( - PipelineParams, - PipelineParamsDict, - PipelineParamsParam, -) -from databricks.bundles.jobs._models.pipeline_task import ( - PipelineTask, - PipelineTaskDict, - PipelineTaskParam, -) -from databricks.bundles.jobs._models.power_bi_model import ( - PowerBiModel, - PowerBiModelDict, - PowerBiModelParam, -) -from databricks.bundles.jobs._models.power_bi_table import ( - PowerBiTable, - PowerBiTableDict, - PowerBiTableParam, -) -from databricks.bundles.jobs._models.power_bi_task import ( - PowerBiTask, - PowerBiTaskDict, - PowerBiTaskParam, -) -from databricks.bundles.jobs._models.python_operator_task import ( - PythonOperatorTask, - PythonOperatorTaskDict, - PythonOperatorTaskParam, -) -from databricks.bundles.jobs._models.python_operator_task_parameter import ( - PythonOperatorTaskParameter, - PythonOperatorTaskParameterDict, - PythonOperatorTaskParameterParam, -) -from databricks.bundles.jobs._models.python_py_pi_library import ( - PythonPyPiLibrary, - PythonPyPiLibraryDict, - PythonPyPiLibraryParam, -) -from databricks.bundles.jobs._models.python_wheel_task import ( - PythonWheelTask, - PythonWheelTaskDict, - PythonWheelTaskParam, -) -from databricks.bundles.jobs._models.queue_settings import ( - QueueSettings, - QueueSettingsDict, - QueueSettingsParam, -) -from databricks.bundles.jobs._models.r_cran_library import ( - RCranLibrary, - RCranLibraryDict, - RCranLibraryParam, -) + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam -from databricks.bundles.jobs._models.run_job_task import ( - RunJobTask, - RunJobTaskDict, - RunJobTaskParam, -) -from databricks.bundles.jobs._models.runtime_engine import ( - RuntimeEngine, - RuntimeEngineParam, -) -from databricks.bundles.jobs._models.s3_storage_info import ( - S3StorageInfo, - S3StorageInfoDict, - S3StorageInfoParam, -) + from databricks.bundles.jobs._models.source import Source, SourceParam -from databricks.bundles.jobs._models.spark_jar_task import ( - SparkJarTask, - SparkJarTaskDict, - SparkJarTaskParam, -) -from databricks.bundles.jobs._models.spark_python_task import ( - SparkPythonTask, - SparkPythonTaskDict, - SparkPythonTaskParam, -) -from databricks.bundles.jobs._models.spark_submit_task import ( - SparkSubmitTask, - SparkSubmitTaskDict, - SparkSubmitTaskParam, -) -from databricks.bundles.jobs._models.sparse_checkout import ( - SparseCheckout, - SparseCheckoutDict, - SparseCheckoutParam, -) -from databricks.bundles.jobs._models.sql_condition_configuration import ( - SqlConditionConfiguration, - SqlConditionConfigurationDict, - SqlConditionConfigurationParam, -) -from databricks.bundles.jobs._models.sql_condition_trigger_mode import ( - SqlConditionTriggerMode, - SqlConditionTriggerModeParam, -) -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import ( - SqlTaskAlert, - SqlTaskAlertDict, - SqlTaskAlertParam, -) -from databricks.bundles.jobs._models.sql_task_dashboard import ( - SqlTaskDashboard, - SqlTaskDashboardDict, - SqlTaskDashboardParam, -) -from databricks.bundles.jobs._models.sql_task_file import ( - SqlTaskFile, - SqlTaskFileDict, - SqlTaskFileParam, -) -from databricks.bundles.jobs._models.sql_task_query import ( - SqlTaskQuery, - SqlTaskQueryDict, - SqlTaskQueryParam, -) -from databricks.bundles.jobs._models.sql_task_subscription import ( - SqlTaskSubscription, - SqlTaskSubscriptionDict, - SqlTaskSubscriptionParam, -) + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam -from databricks.bundles.jobs._models.subscription import ( - Subscription, - SubscriptionDict, - SubscriptionParam, -) -from databricks.bundles.jobs._models.subscription_attachment import ( - SubscriptionAttachment, - SubscriptionAttachmentDict, - SubscriptionAttachmentParam, -) -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import ( - SubscriptionAttachmentDataAttachment, - SubscriptionAttachmentDataAttachmentDict, - SubscriptionAttachmentDataAttachmentParam, -) -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import ( - SubscriptionAttachmentDataAttachmentFileFormat, - SubscriptionAttachmentDataAttachmentFileFormatParam, -) -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import ( - SubscriptionAttachmentDataAttachmentWidget, - SubscriptionAttachmentDataAttachmentWidgetDict, - SubscriptionAttachmentDataAttachmentWidgetParam, -) -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import ( - SubscriptionAttachmentPageAttachment, - SubscriptionAttachmentPageAttachmentDict, - SubscriptionAttachmentPageAttachmentParam, -) -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import ( - SubscriptionAttachmentPageAttachmentPageSelection, - SubscriptionAttachmentPageAttachmentPageSelectionParam, -) -from databricks.bundles.jobs._models.subscription_subscriber import ( - SubscriptionSubscriber, - SubscriptionSubscriberDict, - SubscriptionSubscriberParam, -) -from databricks.bundles.jobs._models.table_update_trigger_configuration import ( - TableUpdateTriggerConfiguration, - TableUpdateTriggerConfigurationDict, - TableUpdateTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import ( - TaskDependency, - TaskDependencyDict, - TaskDependencyParam, -) -from databricks.bundles.jobs._models.task_email_notifications import ( - TaskEmailNotifications, - TaskEmailNotificationsDict, - TaskEmailNotificationsParam, -) -from databricks.bundles.jobs._models.task_notification_settings import ( - TaskNotificationSettings, - TaskNotificationSettingsDict, - TaskNotificationSettingsParam, -) -from databricks.bundles.jobs._models.task_retry_mode import ( - TaskRetryMode, - TaskRetryModeParam, -) -from databricks.bundles.jobs._models.trigger_configuration import ( - TriggerConfiguration, - TriggerConfigurationDict, - TriggerConfigurationParam, -) -from databricks.bundles.jobs._models.trigger_settings import ( - TriggerSettings, - TriggerSettingsDict, - TriggerSettingsParam, -) -from databricks.bundles.jobs._models.volumes_storage_info import ( - VolumesStorageInfo, - VolumesStorageInfoDict, - VolumesStorageInfoParam, -) -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import ( - WebhookNotifications, - WebhookNotificationsDict, - WebhookNotificationsParam, -) -from databricks.bundles.jobs._models.workload_type import ( - WorkloadType, - WorkloadTypeDict, - WorkloadTypeParam, -) -from databricks.bundles.jobs._models.workspace_storage_info import ( - WorkspaceStorageInfo, - WorkspaceStorageInfoDict, - WorkspaceStorageInfoParam, -) + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam def _resolve_recursive_imports(): @@ -935,5 +532,4 @@ def _resolve_recursive_imports(): globalns={"Task": Task, "VariableOr": VariableOr}, ) - _resolve_recursive_imports() diff --git a/python/databricks/bundles/jobs/_models/adlsgen2_info.py b/python/databricks/bundles/jobs/_models/adlsgen2_info.py index 87c883adeb9..82d71e373c4 100644 --- a/python/databricks/bundles/jobs/_models/adlsgen2_info.py +++ b/python/databricks/bundles/jobs/_models/adlsgen2_info.py @@ -1,33 +1,194 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Adlsgen2Info: """ A storage location in Adls Gen2 """ - destination: VariableOr[str] """ abfss destination, e.g. `abfss://@.dfs.core.windows.net/`. """ @classmethod - def from_dict(cls, value: "Adlsgen2InfoDict") -> "Self": + def from_dict(cls, value: 'Adlsgen2InfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "Adlsgen2InfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'Adlsgen2InfoDict': + return _transform_to_json_value(self) # type:ignore + class Adlsgen2InfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/ai_runtime_task.py b/python/databricks/bundles/jobs/_models/ai_runtime_task.py index 09144f23c83..bee1cc5a45c 100644 --- a/python/databricks/bundles/jobs/_models/ai_runtime_task.py +++ b/python/databricks/bundles/jobs/_models/ai_runtime_task.py @@ -1,41 +1,189 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import ( - AiRuntimeTaskPriorityClass, - AiRuntimeTaskPriorityClassParam, -) -from databricks.bundles.jobs._models.deployment_spec import ( - DeploymentSpec, - DeploymentSpecParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AiRuntimeTask: """ AiRuntimeTask: multi-node GPU compute task definition for Databricks AI Runtime workloads. - + Jobs-framework-level concepts (retries, per-task timeout, idempotency token, usage/budget policy, permissions) live on the surrounding TaskSettings / run-submit request and are intentionally NOT duplicated here. Users compose `ai_runtime_task` with the standard Jobs/DABs task wrapper to get those. """ - experiment: VariableOr[str] """ [Public Preview] MLflow experiment name for this run. If an experiment with this name @@ -120,11 +268,12 @@ class AiRuntimeTask: """ @classmethod - def from_dict(cls, value: "AiRuntimeTaskDict") -> "Self": + def from_dict(cls, value: 'AiRuntimeTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AiRuntimeTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AiRuntimeTaskDict': + return _transform_to_json_value(self) # type:ignore + class AiRuntimeTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/ai_runtime_task_priority_class.py b/python/databricks/bundles/jobs/_models/ai_runtime_task_priority_class.py index 1dd09afd131..2db44c93491 100644 --- a/python/databricks/bundles/jobs/_models/ai_runtime_task_priority_class.py +++ b/python/databricks/bundles/jobs/_models/ai_runtime_task_priority_class.py @@ -1,22 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class AiRuntimeTaskPriorityClass(Enum): """ :meta private: [EXPERIMENTAL] - + Scheduling priority class for a workload — its priority and preemptability when the scheduler ranks pending work. """ - BEST_EFFORT = "BEST_EFFORT" NORMAL = "NORMAL" CRITICAL = "CRITICAL" - -AiRuntimeTaskPriorityClassParam = ( - Literal["BEST_EFFORT", "NORMAL", "CRITICAL"] | AiRuntimeTaskPriorityClass -) +AiRuntimeTaskPriorityClassParam = Literal["BEST_EFFORT", "NORMAL", "CRITICAL"] | AiRuntimeTaskPriorityClass diff --git a/python/databricks/bundles/jobs/_models/alert_task.py b/python/databricks/bundles/jobs/_models/alert_task.py index 4f3b0d97611..eb139db7056 100644 --- a/python/databricks/bundles/jobs/_models/alert_task.py +++ b/python/databricks/bundles/jobs/_models/alert_task.py @@ -1,24 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.jobs._models.alert_task_subscriber import ( - AlertTaskSubscriber, - AlertTaskSubscriberParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertTask: """""" @@ -60,11 +213,12 @@ class AlertTask: """ @classmethod - def from_dict(cls, value: "AlertTaskDict") -> "Self": + def from_dict(cls, value: 'AlertTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertTaskDict': + return _transform_to_json_value(self) # type:ignore + class AlertTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/alert_task_subscriber.py b/python/databricks/bundles/jobs/_models/alert_task_subscriber.py index 3805069edcf..5fd6c9b9b2a 100644 --- a/python/databricks/bundles/jobs/_models/alert_task_subscriber.py +++ b/python/databricks/bundles/jobs/_models/alert_task_subscriber.py @@ -1,23 +1,183 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AlertTaskSubscriber: """ Represents a subscriber that will receive alert notifications. A subscriber can be either a user (via email) or a notification destination (via destination_id). """ - destination_id: VariableOrOptional[str] = None """ [Public Preview] @@ -29,11 +189,12 @@ class AlertTaskSubscriber: """ @classmethod - def from_dict(cls, value: "AlertTaskSubscriberDict") -> "Self": + def from_dict(cls, value: 'AlertTaskSubscriberDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AlertTaskSubscriberDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AlertTaskSubscriberDict': + return _transform_to_json_value(self) # type:ignore + class AlertTaskSubscriberDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/authentication_method.py b/python/databricks/bundles/jobs/_models/authentication_method.py index ba7b603d8d5..55ad8f3e192 100644 --- a/python/databricks/bundles/jobs/_models/authentication_method.py +++ b/python/databricks/bundles/jobs/_models/authentication_method.py @@ -1,12 +1,175 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class AuthenticationMethod(Enum): OAUTH = "OAUTH" PAT = "PAT" - AuthenticationMethodParam = Literal["OAUTH", "PAT"] | AuthenticationMethod diff --git a/python/databricks/bundles/jobs/_models/auto_scale.py b/python/databricks/bundles/jobs/_models/auto_scale.py index 10ca50c26d7..031836d4fbc 100644 --- a/python/databricks/bundles/jobs/_models/auto_scale.py +++ b/python/databricks/bundles/jobs/_models/auto_scale.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AutoScale: """""" @@ -28,11 +189,12 @@ class AutoScale: """ @classmethod - def from_dict(cls, value: "AutoScaleDict") -> "Self": + def from_dict(cls, value: 'AutoScaleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AutoScaleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AutoScaleDict': + return _transform_to_json_value(self) # type:ignore + class AutoScaleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/aws_attributes.py b/python/databricks/bundles/jobs/_models/aws_attributes.py index 5eaf7acbed8..081ac942051 100644 --- a/python/databricks/bundles/jobs/_models/aws_attributes.py +++ b/python/databricks/bundles/jobs/_models/aws_attributes.py @@ -1,30 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.aws_availability import ( - AwsAvailability, - AwsAvailabilityParam, -) -from databricks.bundles.jobs._models.ebs_volume_type import ( - EbsVolumeType, - EbsVolumeTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AwsAttributes: """ Attributes set during cluster creation which are related to Amazon Web Services. """ - availability: VariableOrOptional[AwsAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -123,11 +275,12 @@ class AwsAttributes: """ @classmethod - def from_dict(cls, value: "AwsAttributesDict") -> "Self": + def from_dict(cls, value: 'AwsAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AwsAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AwsAttributesDict': + return _transform_to_json_value(self) # type:ignore + class AwsAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/aws_availability.py b/python/databricks/bundles/jobs/_models/aws_availability.py index 4c3810f067e..8188487c483 100644 --- a/python/databricks/bundles/jobs/_models/aws_availability.py +++ b/python/databricks/bundles/jobs/_models/aws_availability.py @@ -1,21 +1,181 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class AwsAvailability(Enum): """ Availability type used for all subsequent nodes past the `first_on_demand` ones. - + Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ - SPOT = "SPOT" ON_DEMAND = "ON_DEMAND" SPOT_WITH_FALLBACK = "SPOT_WITH_FALLBACK" - -AwsAvailabilityParam = ( - Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability -) +AwsAvailabilityParam = Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability diff --git a/python/databricks/bundles/jobs/_models/azure_attributes.py b/python/databricks/bundles/jobs/_models/azure_attributes.py index 64509dcd040..969d3abdb89 100644 --- a/python/databricks/bundles/jobs/_models/azure_attributes.py +++ b/python/databricks/bundles/jobs/_models/azure_attributes.py @@ -1,30 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.azure_availability import ( - AzureAvailability, - AzureAvailabilityParam, -) -from databricks.bundles.jobs._models.log_analytics_info import ( - LogAnalyticsInfo, - LogAnalyticsInfoParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AzureAttributes: """ Attributes set during cluster creation which are related to Microsoft Azure. """ - availability: VariableOrOptional[AzureAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -72,11 +224,12 @@ class AzureAttributes: """ @classmethod - def from_dict(cls, value: "AzureAttributesDict") -> "Self": + def from_dict(cls, value: 'AzureAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AzureAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AzureAttributesDict': + return _transform_to_json_value(self) # type:ignore + class AzureAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/azure_availability.py b/python/databricks/bundles/jobs/_models/azure_availability.py index a03cb1fdddf..dfc68e708f0 100644 --- a/python/databricks/bundles/jobs/_models/azure_availability.py +++ b/python/databricks/bundles/jobs/_models/azure_availability.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class AzureAvailability(Enum): @@ -9,13 +173,8 @@ class AzureAvailability(Enum): Availability type used for all subsequent nodes past the `first_on_demand` ones. Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ - SPOT_AZURE = "SPOT_AZURE" ON_DEMAND_AZURE = "ON_DEMAND_AZURE" SPOT_WITH_FALLBACK_AZURE = "SPOT_WITH_FALLBACK_AZURE" - -AzureAvailabilityParam = ( - Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] - | AzureAvailability -) +AzureAvailabilityParam = Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] | AzureAvailability diff --git a/python/databricks/bundles/jobs/_models/clean_rooms_notebook_task.py b/python/databricks/bundles/jobs/_models/clean_rooms_notebook_task.py index 52845928fe1..507f5485c2d 100644 --- a/python/databricks/bundles/jobs/_models/clean_rooms_notebook_task.py +++ b/python/databricks/bundles/jobs/_models/clean_rooms_notebook_task.py @@ -1,27 +1,183 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class CleanRoomsNotebookTask: """ Clean Rooms notebook task for V1 Clean Room service (GA). Replaces the deprecated CleanRoomNotebookTask (defined above) which was for V0 service. """ - clean_room_name: VariableOr[str] """ The clean room that the notebook belongs to. @@ -44,11 +200,12 @@ class CleanRoomsNotebookTask: """ @classmethod - def from_dict(cls, value: "CleanRoomsNotebookTaskDict") -> "Self": + def from_dict(cls, value: 'CleanRoomsNotebookTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CleanRoomsNotebookTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CleanRoomsNotebookTaskDict': + return _transform_to_json_value(self) # type:ignore + class CleanRoomsNotebookTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/clients_types.py b/python/databricks/bundles/jobs/_models/clients_types.py index 87f92446ba5..601bc4a4ba1 100644 --- a/python/databricks/bundles/jobs/_models/clients_types.py +++ b/python/databricks/bundles/jobs/_models/clients_types.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ClientsTypes: """""" @@ -26,11 +187,12 @@ class ClientsTypes: """ @classmethod - def from_dict(cls, value: "ClientsTypesDict") -> "Self": + def from_dict(cls, value: 'ClientsTypesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClientsTypesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClientsTypesDict': + return _transform_to_json_value(self) # type:ignore + class ClientsTypesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/cluster_log_conf.py b/python/databricks/bundles/jobs/_models/cluster_log_conf.py index a5ff3dd34f8..496995fabd8 100644 --- a/python/databricks/bundles/jobs/_models/cluster_log_conf.py +++ b/python/databricks/bundles/jobs/_models/cluster_log_conf.py @@ -1,34 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.dbfs_storage_info import ( - DbfsStorageInfo, - DbfsStorageInfoParam, -) -from databricks.bundles.jobs._models.s3_storage_info import ( - S3StorageInfo, - S3StorageInfoParam, -) -from databricks.bundles.jobs._models.volumes_storage_info import ( - VolumesStorageInfo, - VolumesStorageInfoParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ClusterLogConf: """ Cluster log delivery config """ - dbfs: VariableOrOptional[DbfsStorageInfo] = None """ destination needs to be provided. e.g. @@ -50,11 +198,12 @@ class ClusterLogConf: """ @classmethod - def from_dict(cls, value: "ClusterLogConfDict") -> "Self": + def from_dict(cls, value: 'ClusterLogConfDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClusterLogConfDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClusterLogConfDict': + return _transform_to_json_value(self) # type:ignore + class ClusterLogConfDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/cluster_spec.py b/python/databricks/bundles/jobs/_models/cluster_spec.py index 2afee23b3ca..2ebec63c4c6 100644 --- a/python/databricks/bundles/jobs/_models/cluster_spec.py +++ b/python/databricks/bundles/jobs/_models/cluster_spec.py @@ -1,72 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.jobs._models.auto_scale import ( - AutoScale, - AutoScaleParam, -) -from databricks.bundles.jobs._models.aws_attributes import ( - AwsAttributes, - AwsAttributesParam, -) -from databricks.bundles.jobs._models.azure_attributes import ( - AzureAttributes, - AzureAttributesParam, -) -from databricks.bundles.jobs._models.cluster_log_conf import ( - ClusterLogConf, - ClusterLogConfParam, -) -from databricks.bundles.jobs._models.data_security_mode import ( - DataSecurityMode, - DataSecurityModeParam, -) -from databricks.bundles.jobs._models.dependency_mode import ( - DependencyMode, - DependencyModeParam, -) -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageParam -from databricks.bundles.jobs._models.gcp_attributes import ( - GcpAttributes, - GcpAttributesParam, -) -from databricks.bundles.jobs._models.init_script_info import ( - InitScriptInfo, - InitScriptInfoParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + from databricks.bundles.jobs._models.kind import Kind, KindParam -from databricks.bundles.jobs._models.node_type_flexibility import ( - NodeTypeFlexibility, - NodeTypeFlexibilityParam, -) -from databricks.bundles.jobs._models.runtime_engine import ( - RuntimeEngine, - RuntimeEngineParam, -) -from databricks.bundles.jobs._models.workload_type import ( - WorkloadType, - WorkloadTypeParam, -) + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ClusterSpec: """ Contains a snapshot of the latest user specified settings that were used to create/edit the cluster. """ - apply_policy_default_values: VariableOrOptional[bool] = None """ When set to true, fixed and default values from the policy will be used for fields that are omitted. When set to false, only fixed values from the policy will be applied. @@ -332,11 +442,12 @@ class ClusterSpec: """ @classmethod - def from_dict(cls, value: "ClusterSpecDict") -> "Self": + def from_dict(cls, value: 'ClusterSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClusterSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClusterSpecDict': + return _transform_to_json_value(self) # type:ignore + class ClusterSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/compute.py b/python/databricks/bundles/jobs/_models/compute.py index 51cfac4566c..8fe082b5b51 100644 --- a/python/databricks/bundles/jobs/_models/compute.py +++ b/python/databricks/bundles/jobs/_models/compute.py @@ -1,20 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.hardware_accelerator_type import ( - HardwareAcceleratorType, - HardwareAcceleratorTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Compute: """""" @@ -25,11 +182,12 @@ class Compute: """ @classmethod - def from_dict(cls, value: "ComputeDict") -> "Self": + def from_dict(cls, value: 'ComputeDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ComputeDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ComputeDict': + return _transform_to_json_value(self) # type:ignore + class ComputeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/compute_config.py b/python/databricks/bundles/jobs/_models/compute_config.py index cac29578ce0..b3e9e45f235 100644 --- a/python/databricks/bundles/jobs/_models/compute_config.py +++ b/python/databricks/bundles/jobs/_models/compute_config.py @@ -1,22 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ComputeConfig: """ :meta private: [EXPERIMENTAL] """ - num_gpus: VariableOr[int] """ :meta private: [EXPERIMENTAL] @@ -39,11 +199,12 @@ class ComputeConfig: """ @classmethod - def from_dict(cls, value: "ComputeConfigDict") -> "Self": + def from_dict(cls, value: 'ComputeConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ComputeConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ComputeConfigDict': + return _transform_to_json_value(self) # type:ignore + class ComputeConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/compute_spec.py b/python/databricks/bundles/jobs/_models/compute_spec.py index 404c4eb6b37..2786c8f7f20 100644 --- a/python/databricks/bundles/jobs/_models/compute_spec.py +++ b/python/databricks/bundles/jobs/_models/compute_spec.py @@ -1,27 +1,183 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ( - ComputeSpecAcceleratorType, - ComputeSpecAcceleratorTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ComputeSpec: """ ComputeSpec: compute configuration — accelerator type and total accelerator count across all nodes. """ - accelerator_count: VariableOr[int] """ [Public Preview] Total number of accelerators across all nodes. Must be a positive @@ -48,11 +204,12 @@ class ComputeSpec: """ @classmethod - def from_dict(cls, value: "ComputeSpecDict") -> "Self": + def from_dict(cls, value: 'ComputeSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ComputeSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ComputeSpecDict': + return _transform_to_json_value(self) # type:ignore + class ComputeSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/compute_spec_accelerator_type.py b/python/databricks/bundles/jobs/_models/compute_spec_accelerator_type.py index 218cce4766f..5abf5de3d41 100644 --- a/python/databricks/bundles/jobs/_models/compute_spec_accelerator_type.py +++ b/python/databricks/bundles/jobs/_models/compute_spec_accelerator_type.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class ComputeSpecAcceleratorType(Enum): @@ -10,14 +174,9 @@ class ComputeSpecAcceleratorType(Enum): accelerator count is encoded in the value name (e.g. `GPU_8xH100` means 8 H100s per node). """ - GPU_1X_A10 = "GPU_1xA10" GPU_1X_H100 = "GPU_1xH100" GPU_8X_H100 = "GPU_8xH100" GPU_8X_B300 = "GPU_8xB300" - -ComputeSpecAcceleratorTypeParam = ( - Literal["GPU_1xA10", "GPU_1xH100", "GPU_8xH100", "GPU_8xB300"] - | ComputeSpecAcceleratorType -) +ComputeSpecAcceleratorTypeParam = Literal["GPU_1xA10", "GPU_1xH100", "GPU_8xH100", "GPU_8xB300"] | ComputeSpecAcceleratorType diff --git a/python/databricks/bundles/jobs/_models/condition.py b/python/databricks/bundles/jobs/_models/condition.py index 7c242ae9dbf..195d285f9d1 100644 --- a/python/databricks/bundles/jobs/_models/condition.py +++ b/python/databricks/bundles/jobs/_models/condition.py @@ -1,12 +1,175 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class Condition(Enum): ANY_UPDATED = "ANY_UPDATED" ALL_UPDATED = "ALL_UPDATED" - ConditionParam = Literal["ANY_UPDATED", "ALL_UPDATED"] | Condition diff --git a/python/databricks/bundles/jobs/_models/condition_task.py b/python/databricks/bundles/jobs/_models/condition_task.py index 478ca88de11..7dd85d12143 100644 --- a/python/databricks/bundles/jobs/_models/condition_task.py +++ b/python/databricks/bundles/jobs/_models/condition_task.py @@ -1,20 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr -from databricks.bundles.jobs._models.condition_task_op import ( - ConditionTaskOp, - ConditionTaskOpParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ConditionTask: """""" @@ -38,11 +195,12 @@ class ConditionTask: """ @classmethod - def from_dict(cls, value: "ConditionTaskDict") -> "Self": + def from_dict(cls, value: 'ConditionTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ConditionTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ConditionTaskDict': + return _transform_to_json_value(self) # type:ignore + class ConditionTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/condition_task_op.py b/python/databricks/bundles/jobs/_models/condition_task_op.py index 615a7ea995c..e77ecd5d7b2 100644 --- a/python/databricks/bundles/jobs/_models/condition_task_op.py +++ b/python/databricks/bundles/jobs/_models/condition_task_op.py @@ -1,17 +1,180 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class ConditionTaskOp(Enum): """ * `EQUAL_TO`, `NOT_EQUAL` operators perform string comparison of their operands. This means that `“12.0” == “12”` will evaluate to `false`. * `GREATER_THAN`, `GREATER_THAN_OR_EQUAL`, `LESS_THAN`, `LESS_THAN_OR_EQUAL` operators perform numeric comparison of their operands. `“12.0” >= “12”` will evaluate to `true`, `“10.0” >= “12”` will evaluate to `false`. - + The boolean comparison to task values can be implemented with operators `EQUAL_TO`, `NOT_EQUAL`. If a task value was set to a boolean value, it will be serialized to `“true”` or `“false”` for the comparison. """ - EQUAL_TO = "EQUAL_TO" GREATER_THAN = "GREATER_THAN" GREATER_THAN_OR_EQUAL = "GREATER_THAN_OR_EQUAL" @@ -19,15 +182,4 @@ class ConditionTaskOp(Enum): LESS_THAN_OR_EQUAL = "LESS_THAN_OR_EQUAL" NOT_EQUAL = "NOT_EQUAL" - -ConditionTaskOpParam = ( - Literal[ - "EQUAL_TO", - "GREATER_THAN", - "GREATER_THAN_OR_EQUAL", - "LESS_THAN", - "LESS_THAN_OR_EQUAL", - "NOT_EQUAL", - ] - | ConditionTaskOp -) +ConditionTaskOpParam = Literal["EQUAL_TO", "GREATER_THAN", "GREATER_THAN_OR_EQUAL", "LESS_THAN", "LESS_THAN_OR_EQUAL", "NOT_EQUAL"] | ConditionTaskOp diff --git a/python/databricks/bundles/jobs/_models/confidential_compute_type.py b/python/databricks/bundles/jobs/_models/confidential_compute_type.py index e881295bd47..bee78ddb3aa 100644 --- a/python/databricks/bundles/jobs/_models/confidential_compute_type.py +++ b/python/databricks/bundles/jobs/_models/confidential_compute_type.py @@ -1,23 +1,183 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class ConfidentialComputeType(Enum): """ :meta private: [EXPERIMENTAL] - + Confidential computing technology for GCP instances. Aligns with gcloud's --confidential-compute-type flag and the REST API's confidentialInstanceConfig.confidentialInstanceType field. See: https://cloud.google.com/confidential-computing/confidential-vm/docs/create-a-confidential-vm-instance """ - CONFIDENTIAL_COMPUTE_TYPE_NONE = "CONFIDENTIAL_COMPUTE_TYPE_NONE" SEV_SNP = "SEV_SNP" - -ConfidentialComputeTypeParam = ( - Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType -) +ConfidentialComputeTypeParam = Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType diff --git a/python/databricks/bundles/jobs/_models/continuous.py b/python/databricks/bundles/jobs/_models/continuous.py index e707bb377f7..72ef2fb5357 100644 --- a/python/databricks/bundles/jobs/_models/continuous.py +++ b/python/databricks/bundles/jobs/_models/continuous.py @@ -1,25 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.maintenance_window import ( - MaintenanceWindow, - MaintenanceWindowParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam -from databricks.bundles.jobs._models.task_retry_mode import ( - TaskRetryMode, - TaskRetryModeParam, -) + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Continuous: """""" @@ -40,11 +192,12 @@ class Continuous: """ @classmethod - def from_dict(cls, value: "ContinuousDict") -> "Self": + def from_dict(cls, value: 'ContinuousDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ContinuousDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ContinuousDict': + return _transform_to_json_value(self) # type:ignore + class ContinuousDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/continuous_trigger_configuration.py b/python/databricks/bundles/jobs/_models/continuous_trigger_configuration.py index 421e630a548..3b500e591c5 100644 --- a/python/databricks/bundles/jobs/_models/continuous_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/continuous_trigger_configuration.py @@ -1,31 +1,183 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.maintenance_window import ( - MaintenanceWindow, - MaintenanceWindowParam, -) -from databricks.bundles.jobs._models.task_retry_mode import ( - TaskRetryMode, - TaskRetryModeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ContinuousTriggerConfiguration: """ Continuous trigger. Stripped-down counterpart to `ContinuousSettings`: `pause_status` is owned by the enclosing `TriggerConfiguration` and intentionally omitted here. """ - maintenance_window: VariableOrOptional[MaintenanceWindow] = None """ :meta private: [EXPERIMENTAL] @@ -42,11 +194,12 @@ class ContinuousTriggerConfiguration: """ @classmethod - def from_dict(cls, value: "ContinuousTriggerConfigurationDict") -> "Self": + def from_dict(cls, value: 'ContinuousTriggerConfigurationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ContinuousTriggerConfigurationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ContinuousTriggerConfigurationDict': + return _transform_to_json_value(self) # type:ignore + class ContinuousTriggerConfigurationDict(TypedDict, total=False): @@ -68,6 +221,4 @@ class ContinuousTriggerConfigurationDict(TypedDict, total=False): """ -ContinuousTriggerConfigurationParam = ( - ContinuousTriggerConfigurationDict | ContinuousTriggerConfiguration -) +ContinuousTriggerConfigurationParam = ContinuousTriggerConfigurationDict | ContinuousTriggerConfiguration diff --git a/python/databricks/bundles/jobs/_models/cron_schedule.py b/python/databricks/bundles/jobs/_models/cron_schedule.py index 3c8a37992af..570e42beb07 100644 --- a/python/databricks/bundles/jobs/_models/cron_schedule.py +++ b/python/databricks/bundles/jobs/_models/cron_schedule.py @@ -1,21 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam -from databricks.bundles.jobs._models.sql_condition_configuration import ( - SqlConditionConfiguration, - SqlConditionConfigurationParam, -) + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class CronSchedule: """""" @@ -44,11 +200,12 @@ class CronSchedule: """ @classmethod - def from_dict(cls, value: "CronScheduleDict") -> "Self": + def from_dict(cls, value: 'CronScheduleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CronScheduleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CronScheduleDict': + return _transform_to_json_value(self) # type:ignore + class CronScheduleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/cron_trigger_configuration.py b/python/databricks/bundles/jobs/_models/cron_trigger_configuration.py index 9e083514b15..98cd9ab6dd8 100644 --- a/python/databricks/bundles/jobs/_models/cron_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/cron_trigger_configuration.py @@ -1,23 +1,183 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class CronTriggerConfiguration: """ Cron schedule trigger. Stripped-down counterpart to `CronSchedule`: `pause_status` and `sql_condition` are owned by the enclosing `TriggerConfiguration` and intentionally omitted here. """ - quartz_cron_expression: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -35,11 +195,12 @@ class CronTriggerConfiguration: """ @classmethod - def from_dict(cls, value: "CronTriggerConfigurationDict") -> "Self": + def from_dict(cls, value: 'CronTriggerConfigurationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CronTriggerConfigurationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CronTriggerConfigurationDict': + return _transform_to_json_value(self) # type:ignore + class CronTriggerConfigurationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/dashboard_task.py b/python/databricks/bundles/jobs/_models/dashboard_task.py index 7d07ea50175..ccb28da7381 100644 --- a/python/databricks/bundles/jobs/_models/dashboard_task.py +++ b/python/databricks/bundles/jobs/_models/dashboard_task.py @@ -1,26 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrDict, VariableOrOptional -from databricks.bundles.jobs._models.subscription import ( - Subscription, - SubscriptionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DashboardTask: """ Configures the Lakeview Dashboard job task type. """ - dashboard_id: VariableOrOptional[str] = None """ The identifier of the dashboard to refresh. @@ -48,11 +204,12 @@ class DashboardTask: """ @classmethod - def from_dict(cls, value: "DashboardTaskDict") -> "Self": + def from_dict(cls, value: 'DashboardTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DashboardTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DashboardTaskDict': + return _transform_to_json_value(self) # type:ignore + class DashboardTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/data_security_mode.py b/python/databricks/bundles/jobs/_models/data_security_mode.py index ae1aafcf085..b91c86dccd6 100644 --- a/python/databricks/bundles/jobs/_models/data_security_mode.py +++ b/python/databricks/bundles/jobs/_models/data_security_mode.py @@ -1,32 +1,195 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class DataSecurityMode(Enum): """ Data security mode decides what data governance model to use when accessing data from a cluster. - + * `DATA_SECURITY_MODE_AUTO`: Databricks will choose the most appropriate access mode depending on your compute configuration. * `DATA_SECURITY_MODE_STANDARD`: A secure cluster that can be shared by multiple users. Cluster users are fully isolated so that they cannot see each other’s data and credentials. Most data governance features are supported in this mode. But programming languages and cluster features might be limited. * `DATA_SECURITY_MODE_DEDICATED`: A secure cluster that can only be exclusively used by a single user specified in `single_user_name`. Most programming languages, cluster features and data governance features are available in this mode. - + The following modes are legacy aliases for the above modes: - + * `USER_ISOLATION`: Legacy alias for `DATA_SECURITY_MODE_STANDARD`. * `SINGLE_USER`: Legacy alias for `DATA_SECURITY_MODE_DEDICATED`. - + The following modes are deprecated starting with Databricks Runtime 15.0 and will be removed for future Databricks Runtime versions: - + * `LEGACY_TABLE_ACL`: This mode is for users migrating from legacy Table ACL clusters. * `LEGACY_PASSTHROUGH`: This mode is for users migrating from legacy Passthrough on high concurrency clusters. * `LEGACY_SINGLE_USER`: This mode is for users migrating from legacy Passthrough on standard clusters. * `LEGACY_SINGLE_USER_STANDARD`: This mode provides a way that doesn’t have UC nor passthrough enabled. """ - NONE = "NONE" SINGLE_USER = "SINGLE_USER" USER_ISOLATION = "USER_ISOLATION" @@ -38,19 +201,4 @@ class DataSecurityMode(Enum): DATA_SECURITY_MODE_DEDICATED = "DATA_SECURITY_MODE_DEDICATED" DATA_SECURITY_MODE_AUTO = "DATA_SECURITY_MODE_AUTO" - -DataSecurityModeParam = ( - Literal[ - "NONE", - "SINGLE_USER", - "USER_ISOLATION", - "LEGACY_TABLE_ACL", - "LEGACY_PASSTHROUGH", - "LEGACY_SINGLE_USER", - "LEGACY_SINGLE_USER_STANDARD", - "DATA_SECURITY_MODE_STANDARD", - "DATA_SECURITY_MODE_DEDICATED", - "DATA_SECURITY_MODE_AUTO", - ] - | DataSecurityMode -) +DataSecurityModeParam = Literal["NONE", "SINGLE_USER", "USER_ISOLATION", "LEGACY_TABLE_ACL", "LEGACY_PASSTHROUGH", "LEGACY_SINGLE_USER", "LEGACY_SINGLE_USER_STANDARD", "DATA_SECURITY_MODE_STANDARD", "DATA_SECURITY_MODE_DEDICATED", "DATA_SECURITY_MODE_AUTO"] | DataSecurityMode diff --git a/python/databricks/bundles/jobs/_models/day_of_week.py b/python/databricks/bundles/jobs/_models/day_of_week.py index e5874108dd6..39e0e7263bb 100644 --- a/python/databricks/bundles/jobs/_models/day_of_week.py +++ b/python/databricks/bundles/jobs/_models/day_of_week.py @@ -1,14 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class DayOfWeek(Enum): """ Days of week that can be referenced by Jobs scheduling settings. """ - MONDAY = "MONDAY" TUESDAY = "TUESDAY" WEDNESDAY = "WEDNESDAY" @@ -17,10 +180,4 @@ class DayOfWeek(Enum): SATURDAY = "SATURDAY" SUNDAY = "SUNDAY" - -DayOfWeekParam = ( - Literal[ - "MONDAY", "TUESDAY", "WEDNESDAY", "THURSDAY", "FRIDAY", "SATURDAY", "SUNDAY" - ] - | DayOfWeek -) +DayOfWeekParam = Literal["MONDAY", "TUESDAY", "WEDNESDAY", "THURSDAY", "FRIDAY", "SATURDAY", "SUNDAY"] | DayOfWeek diff --git a/python/databricks/bundles/jobs/_models/dbfs_storage_info.py b/python/databricks/bundles/jobs/_models/dbfs_storage_info.py index 29744acf26a..02ea147094a 100644 --- a/python/databricks/bundles/jobs/_models/dbfs_storage_info.py +++ b/python/databricks/bundles/jobs/_models/dbfs_storage_info.py @@ -1,33 +1,194 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DbfsStorageInfo: """ A storage location in DBFS """ - destination: VariableOr[str] """ dbfs destination, e.g. `dbfs:/my/path` """ @classmethod - def from_dict(cls, value: "DbfsStorageInfoDict") -> "Self": + def from_dict(cls, value: 'DbfsStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DbfsStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DbfsStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class DbfsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/dbt_platform_task.py b/python/databricks/bundles/jobs/_models/dbt_platform_task.py index 61128b71634..1c32f08969f 100644 --- a/python/databricks/bundles/jobs/_models/dbt_platform_task.py +++ b/python/databricks/bundles/jobs/_models/dbt_platform_task.py @@ -1,22 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DbtPlatformTask: """ :meta private: [EXPERIMENTAL] """ - connection_resource_name: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -32,11 +192,12 @@ class DbtPlatformTask: """ @classmethod - def from_dict(cls, value: "DbtPlatformTaskDict") -> "Self": + def from_dict(cls, value: 'DbtPlatformTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DbtPlatformTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DbtPlatformTaskDict': + return _transform_to_json_value(self) # type:ignore + class DbtPlatformTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/dbt_task.py b/python/databricks/bundles/jobs/_models/dbt_task.py index d8c92391e30..10ef2fd70f7 100644 --- a/python/databricks/bundles/jobs/_models/dbt_task.py +++ b/python/databricks/bundles/jobs/_models/dbt_task.py @@ -1,17 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + from databricks.bundles.jobs._models.source import Source, SourceParam +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DbtTask: """""" @@ -58,11 +218,12 @@ class DbtTask: """ @classmethod - def from_dict(cls, value: "DbtTaskDict") -> "Self": + def from_dict(cls, value: 'DbtTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DbtTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DbtTaskDict': + return _transform_to_json_value(self) # type:ignore + class DbtTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/dependency_mode.py b/python/databricks/bundles/jobs/_models/dependency_mode.py index 1fdc22cfa4c..90de274cfd6 100644 --- a/python/databricks/bundles/jobs/_models/dependency_mode.py +++ b/python/databricks/bundles/jobs/_models/dependency_mode.py @@ -1,28 +1,183 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class DependencyMode(Enum): """ Controls dependency configuration for the cluster. - + * `DEPENDENCY_MODE_AUTO`: Databricks will choose the most appropriate dependency mode based on your compute configuration. * `DEPENDENCY_MODE_ENVIRONMENTS`: Enables a unified dependency management experience across classic and serverless, resulting in increased stability and performance. Supported only on DBR 19+ in Standard access mode. * `DEPENDENCY_MODE_CLUSTER_LIBRARIES`: Legacy mode: dependencies come from cluster libraries and init scripts. """ - DEPENDENCY_MODE_ENVIRONMENTS = "DEPENDENCY_MODE_ENVIRONMENTS" DEPENDENCY_MODE_CLUSTER_LIBRARIES = "DEPENDENCY_MODE_CLUSTER_LIBRARIES" DEPENDENCY_MODE_AUTO = "DEPENDENCY_MODE_AUTO" - -DependencyModeParam = ( - Literal[ - "DEPENDENCY_MODE_ENVIRONMENTS", - "DEPENDENCY_MODE_CLUSTER_LIBRARIES", - "DEPENDENCY_MODE_AUTO", - ] - | DependencyMode -) +DependencyModeParam = Literal["DEPENDENCY_MODE_ENVIRONMENTS", "DEPENDENCY_MODE_CLUSTER_LIBRARIES", "DEPENDENCY_MODE_AUTO"] | DependencyMode diff --git a/python/databricks/bundles/jobs/_models/deployment_spec.py b/python/databricks/bundles/jobs/_models/deployment_spec.py index 3c0d8d77a6f..48a01c16135 100644 --- a/python/databricks/bundles/jobs/_models/deployment_spec.py +++ b/python/databricks/bundles/jobs/_models/deployment_spec.py @@ -1,20 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.jobs._models.compute_spec import ( - ComputeSpec, - ComputeSpecParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DeploymentSpec: """ @@ -25,7 +182,6 @@ class DeploymentSpec: role-split workloads (driver + worker, parameter server, separate eval node, etc.) use multiple entries. """ - compute: VariableOr[ComputeSpec] """ [Public Preview] Compute resources allocated to each node in this deployment. @@ -42,11 +198,12 @@ class DeploymentSpec: """ @classmethod - def from_dict(cls, value: "DeploymentSpecDict") -> "Self": + def from_dict(cls, value: 'DeploymentSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DeploymentSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DeploymentSpecDict': + return _transform_to_json_value(self) # type:ignore + class DeploymentSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/docker_basic_auth.py b/python/databricks/bundles/jobs/_models/docker_basic_auth.py index 552ea90a83b..93f10ecfa99 100644 --- a/python/databricks/bundles/jobs/_models/docker_basic_auth.py +++ b/python/databricks/bundles/jobs/_models/docker_basic_auth.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DockerBasicAuth: """""" @@ -26,11 +187,12 @@ class DockerBasicAuth: """ @classmethod - def from_dict(cls, value: "DockerBasicAuthDict") -> "Self": + def from_dict(cls, value: 'DockerBasicAuthDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DockerBasicAuthDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DockerBasicAuthDict': + return _transform_to_json_value(self) # type:ignore + class DockerBasicAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/docker_image.py b/python/databricks/bundles/jobs/_models/docker_image.py index 9dca00df1e2..5f542b48795 100644 --- a/python/databricks/bundles/jobs/_models/docker_image.py +++ b/python/databricks/bundles/jobs/_models/docker_image.py @@ -1,20 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.docker_basic_auth import ( - DockerBasicAuth, - DockerBasicAuthParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DockerImage: """""" @@ -30,11 +187,12 @@ class DockerImage: """ @classmethod - def from_dict(cls, value: "DockerImageDict") -> "Self": + def from_dict(cls, value: 'DockerImageDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DockerImageDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DockerImageDict': + return _transform_to_json_value(self) # type:ignore + class DockerImageDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/ebs_volume_type.py b/python/databricks/bundles/jobs/_models/ebs_volume_type.py index d8f3ef13f89..93277238be3 100644 --- a/python/databricks/bundles/jobs/_models/ebs_volume_type.py +++ b/python/databricks/bundles/jobs/_models/ebs_volume_type.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class EbsVolumeType(Enum): @@ -9,11 +173,7 @@ class EbsVolumeType(Enum): All EBS volume types that Databricks supports. See https://aws.amazon.com/ebs/details/ for details. """ - GENERAL_PURPOSE_SSD = "GENERAL_PURPOSE_SSD" THROUGHPUT_OPTIMIZED_HDD = "THROUGHPUT_OPTIMIZED_HDD" - -EbsVolumeTypeParam = ( - Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType -) +EbsVolumeTypeParam = Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType diff --git a/python/databricks/bundles/jobs/_models/environment.py b/python/databricks/bundles/jobs/_models/environment.py index 850a1cf3098..0b1e2077a18 100644 --- a/python/databricks/bundles/jobs/_models/environment.py +++ b/python/databricks/bundles/jobs/_models/environment.py @@ -1,23 +1,183 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Environment: """ The environment entity used to preserve serverless environment side panel, jobs' environment for non-notebook task, and SDP's environment for classic and serverless pipelines. In this minimal environment spec, only pip and java dependencies are supported. """ - base_environment: VariableOrOptional[str] = None """ The base environment this environment is built on top of. A base environment defines the environment version and a @@ -61,11 +221,12 @@ class Environment: """ @classmethod - def from_dict(cls, value: "EnvironmentDict") -> "Self": + def from_dict(cls, value: 'EnvironmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EnvironmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EnvironmentDict': + return _transform_to_json_value(self) # type:ignore + class EnvironmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/file_arrival_trigger_configuration.py b/python/databricks/bundles/jobs/_models/file_arrival_trigger_configuration.py index 7e31a4b6946..aba1a82fab5 100644 --- a/python/databricks/bundles/jobs/_models/file_arrival_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/file_arrival_trigger_configuration.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FileArrivalTriggerConfiguration: """""" @@ -34,11 +195,12 @@ class FileArrivalTriggerConfiguration: """ @classmethod - def from_dict(cls, value: "FileArrivalTriggerConfigurationDict") -> "Self": + def from_dict(cls, value: 'FileArrivalTriggerConfigurationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FileArrivalTriggerConfigurationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FileArrivalTriggerConfigurationDict': + return _transform_to_json_value(self) # type:ignore + class FileArrivalTriggerConfigurationDict(TypedDict, total=False): @@ -63,6 +225,4 @@ class FileArrivalTriggerConfigurationDict(TypedDict, total=False): """ -FileArrivalTriggerConfigurationParam = ( - FileArrivalTriggerConfigurationDict | FileArrivalTriggerConfiguration -) +FileArrivalTriggerConfigurationParam = FileArrivalTriggerConfigurationDict | FileArrivalTriggerConfiguration diff --git a/python/databricks/bundles/jobs/_models/for_each_task.py b/python/databricks/bundles/jobs/_models/for_each_task.py index c8ac1625ff6..b52421df3d8 100644 --- a/python/databricks/bundles/jobs/_models/for_each_task.py +++ b/python/databricks/bundles/jobs/_models/for_each_task.py @@ -1,17 +1,176 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam -if TYPE_CHECKING: - from typing_extensions import Self +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - from databricks.bundles.jobs._models.task import Task, TaskParam +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + +if TYPE_CHECKING: + from databricks.bundles.jobs._models.task import Task, TaskParam + from typing_extensions import Self @dataclass(kw_only=True) class ForEachTask: @@ -35,11 +194,12 @@ class ForEachTask: """ @classmethod - def from_dict(cls, value: "ForEachTaskDict") -> "Self": + def from_dict(cls, value: 'ForEachTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ForEachTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ForEachTaskDict': + return _transform_to_json_value(self) # type:ignore + class ForEachTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/gcp_attributes.py b/python/databricks/bundles/jobs/_models/gcp_attributes.py index 2874f9da8e4..1e4e47efb17 100644 --- a/python/databricks/bundles/jobs/_models/gcp_attributes.py +++ b/python/databricks/bundles/jobs/_models/gcp_attributes.py @@ -1,30 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.confidential_compute_type import ( - ConfidentialComputeType, - ConfidentialComputeTypeParam, -) -from databricks.bundles.jobs._models.gcp_availability import ( - GcpAvailability, - GcpAvailabilityParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GcpAttributes: """ Attributes set during cluster creation which are related to GCP. """ - availability: VariableOrOptional[GcpAvailability] = None """ This field determines whether the spark executors will be scheduled to run on preemptible @@ -91,11 +243,12 @@ class GcpAttributes: """ @classmethod - def from_dict(cls, value: "GcpAttributesDict") -> "Self": + def from_dict(cls, value: 'GcpAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GcpAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GcpAttributesDict': + return _transform_to_json_value(self) # type:ignore + class GcpAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/gcp_availability.py b/python/databricks/bundles/jobs/_models/gcp_availability.py index 0d391b87fe3..2f1cdb6dea9 100644 --- a/python/databricks/bundles/jobs/_models/gcp_availability.py +++ b/python/databricks/bundles/jobs/_models/gcp_availability.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class GcpAvailability(Enum): @@ -9,13 +173,8 @@ class GcpAvailability(Enum): This field determines whether the instance pool will contain preemptible VMs, on-demand VMs, or preemptible VMs with a fallback to on-demand VMs if the former is unavailable. """ - PREEMPTIBLE_GCP = "PREEMPTIBLE_GCP" ON_DEMAND_GCP = "ON_DEMAND_GCP" PREEMPTIBLE_WITH_FALLBACK_GCP = "PREEMPTIBLE_WITH_FALLBACK_GCP" - -GcpAvailabilityParam = ( - Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] - | GcpAvailability -) +GcpAvailabilityParam = Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] | GcpAvailability diff --git a/python/databricks/bundles/jobs/_models/gcs_storage_info.py b/python/databricks/bundles/jobs/_models/gcs_storage_info.py index e8f1059373b..7dc8748f2fa 100644 --- a/python/databricks/bundles/jobs/_models/gcs_storage_info.py +++ b/python/databricks/bundles/jobs/_models/gcs_storage_info.py @@ -1,33 +1,194 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GcsStorageInfo: """ A storage location in Google Cloud Platform's GCS """ - destination: VariableOr[str] """ GCS destination/URI, e.g. `gs://my-bucket/some-prefix` """ @classmethod - def from_dict(cls, value: "GcsStorageInfoDict") -> "Self": + def from_dict(cls, value: 'GcsStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GcsStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GcsStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class GcsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/gen_ai_compute_task.py b/python/databricks/bundles/jobs/_models/gen_ai_compute_task.py index 4e42c50f81d..4787305d82b 100644 --- a/python/databricks/bundles/jobs/_models/gen_ai_compute_task.py +++ b/python/databricks/bundles/jobs/_models/gen_ai_compute_task.py @@ -1,26 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.jobs._models.compute_config import ( - ComputeConfig, - ComputeConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + from databricks.bundles.jobs._models.source import Source, SourceParam +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GenAiComputeTask: """ :meta private: [EXPERIMENTAL] - + DEPRECATED — use `AiRuntimeTask` for all new BYOT multi-node GPU workloads (see ai_runtime_task.proto). `AiRuntimeTask` is the only supported BYOT task type for new workloads; this proto is retained only @@ -28,7 +184,6 @@ class GenAiComputeTask: removed once the pywheel → databricks-cli migration completes (post- PuPr). """ - dl_runtime_image: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -91,11 +246,12 @@ class GenAiComputeTask: """ @classmethod - def from_dict(cls, value: "GenAiComputeTaskDict") -> "Self": + def from_dict(cls, value: 'GenAiComputeTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GenAiComputeTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GenAiComputeTaskDict': + return _transform_to_json_value(self) # type:ignore + class GenAiComputeTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/git_provider.py b/python/databricks/bundles/jobs/_models/git_provider.py index 165f77de5cb..d69bfbdfe9d 100644 --- a/python/databricks/bundles/jobs/_models/git_provider.py +++ b/python/databricks/bundles/jobs/_models/git_provider.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class GitProvider(Enum): @@ -14,17 +178,4 @@ class GitProvider(Enum): GIT_LAB_ENTERPRISE_EDITION = "gitLabEnterpriseEdition" AWS_CODE_COMMIT = "awsCodeCommit" - -GitProviderParam = ( - Literal[ - "gitHub", - "bitbucketCloud", - "azureDevOpsServices", - "gitHubEnterprise", - "bitbucketServer", - "gitLab", - "gitLabEnterpriseEdition", - "awsCodeCommit", - ] - | GitProvider -) +GitProviderParam = Literal["gitHub", "bitbucketCloud", "azureDevOpsServices", "gitHubEnterprise", "bitbucketServer", "gitLab", "gitLabEnterpriseEdition", "awsCodeCommit"] | GitProvider diff --git a/python/databricks/bundles/jobs/_models/git_source.py b/python/databricks/bundles/jobs/_models/git_source.py index 4ddd1d83764..bee4198df05 100644 --- a/python/databricks/bundles/jobs/_models/git_source.py +++ b/python/databricks/bundles/jobs/_models/git_source.py @@ -1,31 +1,186 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam -from databricks.bundles.jobs._models.sparse_checkout import ( - SparseCheckout, - SparseCheckoutParam, -) + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GitSource: """ An optional specification for a remote Git repository containing the source code used by tasks. Version-controlled source code is supported by notebook, dbt, Python script, and SQL File tasks. - + If `git_source` is set, these tasks retrieve the file from the remote repository by default. However, this behavior can be overridden by setting `source` to `WORKSPACE` on the task. - + Note: dbt and SQL File tasks support only version-controlled sources. If dbt or SQL File tasks are used, `git_source` must be defined on the job. """ - git_provider: VariableOr[GitProvider] """ Unique identifier of the service used to host the Git repository. The value is case insensitive. @@ -54,11 +209,12 @@ class GitSource: sparse_checkout: VariableOrOptional[SparseCheckout] = None @classmethod - def from_dict(cls, value: "GitSourceDict") -> "Self": + def from_dict(cls, value: 'GitSourceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GitSourceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GitSourceDict': + return _transform_to_json_value(self) # type:ignore + class GitSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/hardware_accelerator_type.py b/python/databricks/bundles/jobs/_models/hardware_accelerator_type.py index 3d77c62b727..6aa2466d4ad 100644 --- a/python/databricks/bundles/jobs/_models/hardware_accelerator_type.py +++ b/python/databricks/bundles/jobs/_models/hardware_accelerator_type.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class HardwareAcceleratorType(Enum): @@ -10,14 +174,9 @@ class HardwareAcceleratorType(Enum): NOTE: This enum is referenced and is intended to be used by other Databricks services that need to specify hardware accelerator requirements for AI compute workloads. """ - GPU_1X_A10 = "GPU_1xA10" GPU_8X_H100 = "GPU_8xH100" GPU_1X_H100 = "GPU_1xH100" GPU_8X_B300 = "GPU_8xB300" - -HardwareAcceleratorTypeParam = ( - Literal["GPU_1xA10", "GPU_8xH100", "GPU_1xH100", "GPU_8xB300"] - | HardwareAcceleratorType -) +HardwareAcceleratorTypeParam = Literal["GPU_1xA10", "GPU_8xH100", "GPU_1xH100", "GPU_8xB300"] | HardwareAcceleratorType diff --git a/python/databricks/bundles/jobs/_models/init_script_info.py b/python/databricks/bundles/jobs/_models/init_script_info.py index a7c5800a594..854d00dc6f8 100644 --- a/python/databricks/bundles/jobs/_models/init_script_info.py +++ b/python/databricks/bundles/jobs/_models/init_script_info.py @@ -1,50 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.adlsgen2_info import ( - Adlsgen2Info, - Adlsgen2InfoParam, -) -from databricks.bundles.jobs._models.dbfs_storage_info import ( - DbfsStorageInfo, - DbfsStorageInfoParam, -) -from databricks.bundles.jobs._models.gcs_storage_info import ( - GcsStorageInfo, - GcsStorageInfoParam, -) -from databricks.bundles.jobs._models.local_file_info import ( - LocalFileInfo, - LocalFileInfoParam, -) -from databricks.bundles.jobs._models.s3_storage_info import ( - S3StorageInfo, - S3StorageInfoParam, -) -from databricks.bundles.jobs._models.volumes_storage_info import ( - VolumesStorageInfo, - VolumesStorageInfoParam, -) -from databricks.bundles.jobs._models.workspace_storage_info import ( - WorkspaceStorageInfo, - WorkspaceStorageInfoParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InitScriptInfo: """ Config for an individual init script """ - abfss: VariableOrOptional[Adlsgen2Info] = None """ Contains the Azure Data Lake Storage destination path @@ -89,11 +221,12 @@ class InitScriptInfo: """ @classmethod - def from_dict(cls, value: "InitScriptInfoDict") -> "Self": + def from_dict(cls, value: 'InitScriptInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InitScriptInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InitScriptInfoDict': + return _transform_to_json_value(self) # type:ignore + class InitScriptInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job.py b/python/databricks/bundles/jobs/_models/job.py index e38305c7d02..e3df4756349 100644 --- a/python/databricks/bundles/jobs/_models/job.py +++ b/python/databricks/bundles/jobs/_models/job.py @@ -1,82 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.jobs._models.continuous import ( - Continuous, - ContinuousParam, -) -from databricks.bundles.jobs._models.cron_schedule import ( - CronSchedule, - CronScheduleParam, -) -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import ( - JobEmailNotifications, - JobEmailNotificationsParam, -) -from databricks.bundles.jobs._models.job_environment import ( - JobEnvironment, - JobEnvironmentParam, -) -from databricks.bundles.jobs._models.job_environment_variables import ( - JobEnvironmentVariables, - JobEnvironmentVariablesParam, -) -from databricks.bundles.jobs._models.job_notification_settings import ( - JobNotificationSettings, - JobNotificationSettingsParam, -) -from databricks.bundles.jobs._models.job_parameter_definition import ( - JobParameterDefinition, - JobParameterDefinitionParam, -) -from databricks.bundles.jobs._models.job_permission import ( - JobPermission, - JobPermissionParam, -) -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rules import ( - JobsHealthRules, - JobsHealthRulesParam, -) -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.jobs._models.performance_target import ( - PerformanceTarget, - PerformanceTargetParam, -) -from databricks.bundles.jobs._models.queue_settings import ( - QueueSettings, - QueueSettingsParam, -) -from databricks.bundles.jobs._models.task import Task, TaskParam -from databricks.bundles.jobs._models.trigger_configuration import ( - TriggerConfiguration, - TriggerConfigurationParam, -) -from databricks.bundles.jobs._models.trigger_settings import ( - TriggerSettings, - TriggerSettingsParam, -) -from databricks.bundles.jobs._models.webhook_notifications import ( - WebhookNotifications, - WebhookNotificationsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Job(Resource): """""" @@ -115,9 +210,7 @@ class Job(Resource): An optional set of email addresses that is notified when runs of this job begin or complete as well as when this job is deleted. """ - environment_variables: VariableOrList[JobEnvironmentVariables] = field( - default_factory=list - ) + environment_variables: VariableOrList[JobEnvironmentVariables] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -268,11 +361,12 @@ class Job(Resource): """ @classmethod - def from_dict(cls, value: "JobDict") -> "Self": + def from_dict(cls, value: 'JobDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobDict': + return _transform_to_json_value(self) # type:ignore + class JobDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_cluster.py b/python/databricks/bundles/jobs/_models/job_cluster.py index 8c3390db857..57cffd54587 100644 --- a/python/databricks/bundles/jobs/_models/job_cluster.py +++ b/python/databricks/bundles/jobs/_models/job_cluster.py @@ -1,17 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobCluster: """""" @@ -37,11 +197,12 @@ class JobCluster: """ @classmethod - def from_dict(cls, value: "JobClusterDict") -> "Self": + def from_dict(cls, value: 'JobClusterDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobClusterDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobClusterDict': + return _transform_to_json_value(self) # type:ignore + class JobClusterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_email_notifications.py b/python/databricks/bundles/jobs/_models/job_email_notifications.py index f6bb6fdd548..89d7472177e 100644 --- a/python/databricks/bundles/jobs/_models/job_email_notifications.py +++ b/python/databricks/bundles/jobs/_models/job_email_notifications.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobEmailNotifications: """""" @@ -21,9 +182,7 @@ class JobEmailNotifications: This field is `deprecated`. Please use the `notification_settings.no_alert_for_skipped_runs` field. """ - on_duration_warning_threshold_exceeded: VariableOrList[str] = field( - default_factory=list - ) + on_duration_warning_threshold_exceeded: VariableOrList[str] = field(default_factory=list) """ A list of email addresses to be notified when the duration of a run exceeds the threshold specified for the `RUN_DURATION_SECONDS` metric in the `health` field. If no rule for the `RUN_DURATION_SECONDS` metric is specified in the `health` field for the job, notifications are not sent. """ @@ -61,11 +220,12 @@ class JobEmailNotifications: """ @classmethod - def from_dict(cls, value: "JobEmailNotificationsDict") -> "Self": + def from_dict(cls, value: 'JobEmailNotificationsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobEmailNotificationsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobEmailNotificationsDict': + return _transform_to_json_value(self) # type:ignore + class JobEmailNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_environment.py b/python/databricks/bundles/jobs/_models/job_environment.py index c71ee45dbd6..a698e361c5d 100644 --- a/python/databricks/bundles/jobs/_models/job_environment.py +++ b/python/databricks/bundles/jobs/_models/job_environment.py @@ -1,17 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.jobs._models.environment import Environment, EnvironmentParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobEnvironment: """""" @@ -28,11 +188,12 @@ class JobEnvironment: """ @classmethod - def from_dict(cls, value: "JobEnvironmentDict") -> "Self": + def from_dict(cls, value: 'JobEnvironmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobEnvironmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobEnvironmentDict': + return _transform_to_json_value(self) # type:ignore + class JobEnvironmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_environment_variables.py b/python/databricks/bundles/jobs/_models/job_environment_variables.py index aca6fd61d7d..f4bf0a58cb4 100644 --- a/python/databricks/bundles/jobs/_models/job_environment_variables.py +++ b/python/databricks/bundles/jobs/_models/job_environment_variables.py @@ -1,20 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.jobs._models.job_environment_variables_spec import ( - JobEnvironmentVariablesSpec, - JobEnvironmentVariablesSpecParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobEnvironmentVariables: """ @@ -23,7 +180,6 @@ class JobEnvironmentVariables: `JobSettings.environment_variables`, and tasks select one via `TaskSettings.environment_variables_key`. """ - environment_variables_key: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -41,11 +197,12 @@ class JobEnvironmentVariables: """ @classmethod - def from_dict(cls, value: "JobEnvironmentVariablesDict") -> "Self": + def from_dict(cls, value: 'JobEnvironmentVariablesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobEnvironmentVariablesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobEnvironmentVariablesDict': + return _transform_to_json_value(self) # type:ignore + class JobEnvironmentVariablesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_environment_variables_spec.py b/python/databricks/bundles/jobs/_models/job_environment_variables_spec.py index 907d2914d57..1cbdfb3c906 100644 --- a/python/databricks/bundles/jobs/_models/job_environment_variables_spec.py +++ b/python/databricks/bundles/jobs/_models/job_environment_variables_spec.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrDict, VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobEnvironmentVariablesSpec: """ @@ -18,7 +179,6 @@ class JobEnvironmentVariablesSpec: variable entry. Runtime environment variables override inline `variables`, which override values from `files`, on duplicate keys. """ - files: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -58,11 +218,12 @@ class JobEnvironmentVariablesSpec: """ @classmethod - def from_dict(cls, value: "JobEnvironmentVariablesSpecDict") -> "Self": + def from_dict(cls, value: 'JobEnvironmentVariablesSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobEnvironmentVariablesSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobEnvironmentVariablesSpecDict': + return _transform_to_json_value(self) # type:ignore + class JobEnvironmentVariablesSpecDict(TypedDict, total=False): @@ -107,6 +268,4 @@ class JobEnvironmentVariablesSpecDict(TypedDict, total=False): """ -JobEnvironmentVariablesSpecParam = ( - JobEnvironmentVariablesSpecDict | JobEnvironmentVariablesSpec -) +JobEnvironmentVariablesSpecParam = JobEnvironmentVariablesSpecDict | JobEnvironmentVariablesSpec diff --git a/python/databricks/bundles/jobs/_models/job_notification_settings.py b/python/databricks/bundles/jobs/_models/job_notification_settings.py index eb6ca454186..a54982135d3 100644 --- a/python/databricks/bundles/jobs/_models/job_notification_settings.py +++ b/python/databricks/bundles/jobs/_models/job_notification_settings.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobNotificationSettings: """""" @@ -26,11 +187,12 @@ class JobNotificationSettings: """ @classmethod - def from_dict(cls, value: "JobNotificationSettingsDict") -> "Self": + def from_dict(cls, value: 'JobNotificationSettingsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobNotificationSettingsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobNotificationSettingsDict': + return _transform_to_json_value(self) # type:ignore + class JobNotificationSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_parameter_definition.py b/python/databricks/bundles/jobs/_models/job_parameter_definition.py index 3e81a538218..211b052d70a 100644 --- a/python/databricks/bundles/jobs/_models/job_parameter_definition.py +++ b/python/databricks/bundles/jobs/_models/job_parameter_definition.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobParameterDefinition: """""" @@ -26,11 +187,12 @@ class JobParameterDefinition: """ @classmethod - def from_dict(cls, value: "JobParameterDefinitionDict") -> "Self": + def from_dict(cls, value: 'JobParameterDefinitionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobParameterDefinitionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobParameterDefinitionDict': + return _transform_to_json_value(self) # type:ignore + class JobParameterDefinitionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_permission.py b/python/databricks/bundles/jobs/_models/job_permission.py index 6864bf99e23..b5e1a1b8e35 100644 --- a/python/databricks/bundles/jobs/_models/job_permission.py +++ b/python/databricks/bundles/jobs/_models/job_permission.py @@ -1,20 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.jobs._models.job_permission_level import ( - JobPermissionLevel, - JobPermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobPermission: """""" @@ -40,11 +197,12 @@ class JobPermission: """ @classmethod - def from_dict(cls, value: "JobPermissionDict") -> "Self": + def from_dict(cls, value: 'JobPermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobPermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobPermissionDict': + return _transform_to_json_value(self) # type:ignore + class JobPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_permission_level.py b/python/databricks/bundles/jobs/_models/job_permission_level.py index e56c96b09d2..88a4481b22b 100644 --- a/python/databricks/bundles/jobs/_models/job_permission_level.py +++ b/python/databricks/bundles/jobs/_models/job_permission_level.py @@ -1,20 +1,180 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class JobPermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" IS_OWNER = "IS_OWNER" CAN_MANAGE_RUN = "CAN_MANAGE_RUN" CAN_VIEW = "CAN_VIEW" - -JobPermissionLevelParam = ( - Literal["CAN_MANAGE", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW"] | JobPermissionLevel -) +JobPermissionLevelParam = Literal["CAN_MANAGE", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW"] | JobPermissionLevel diff --git a/python/databricks/bundles/jobs/_models/job_run_as.py b/python/databricks/bundles/jobs/_models/job_run_as.py index 6242eafad59..b1df2878fba 100644 --- a/python/databricks/bundles/jobs/_models/job_run_as.py +++ b/python/databricks/bundles/jobs/_models/job_run_as.py @@ -1,24 +1,184 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobRunAs: """ Write-only setting. Specifies the user, service principal, or group that the job runs as. If not specified, the job runs as the user who created the job. - + One of `user_name`, `service_principal_name`, or `group_name` should be specified. If not, an error is thrown. """ - group_name: VariableOrOptional[str] = None """ Group name of an account group assigned to the workspace. When set, all tasks run as the group and the group's permissions are used for data access. Setting this field requires being a member of the group, or having the `Assume` permission on the group. @@ -35,11 +195,12 @@ class JobRunAs: """ @classmethod - def from_dict(cls, value: "JobRunAsDict") -> "Self": + def from_dict(cls, value: 'JobRunAsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobRunAsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobRunAsDict': + return _transform_to_json_value(self) # type:ignore + class JobRunAsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/jobs_health_metric.py b/python/databricks/bundles/jobs/_models/jobs_health_metric.py index b662439d3ef..3adecf417cf 100644 --- a/python/databricks/bundles/jobs/_models/jobs_health_metric.py +++ b/python/databricks/bundles/jobs/_models/jobs_health_metric.py @@ -1,34 +1,187 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class JobsHealthMetric(Enum): """ Specifies the health metric that is being evaluated for a particular health rule. - + * `RUN_DURATION_SECONDS`: Expected total time for a run in seconds. * `STREAMING_BACKLOG_BYTES`: An estimate of the maximum bytes of data waiting to be consumed across all streams. This metric is in Public Preview. * `STREAMING_BACKLOG_RECORDS`: An estimate of the maximum offset lag across all streams. This metric is in Public Preview. * `STREAMING_BACKLOG_SECONDS`: An estimate of the maximum consumer delay across all streams. This metric is in Public Preview. * `STREAMING_BACKLOG_FILES`: An estimate of the maximum number of outstanding files across all streams. This metric is in Public Preview. """ - RUN_DURATION_SECONDS = "RUN_DURATION_SECONDS" STREAMING_BACKLOG_BYTES = "STREAMING_BACKLOG_BYTES" STREAMING_BACKLOG_RECORDS = "STREAMING_BACKLOG_RECORDS" STREAMING_BACKLOG_SECONDS = "STREAMING_BACKLOG_SECONDS" STREAMING_BACKLOG_FILES = "STREAMING_BACKLOG_FILES" - -JobsHealthMetricParam = ( - Literal[ - "RUN_DURATION_SECONDS", - "STREAMING_BACKLOG_BYTES", - "STREAMING_BACKLOG_RECORDS", - "STREAMING_BACKLOG_SECONDS", - "STREAMING_BACKLOG_FILES", - ] - | JobsHealthMetric -) +JobsHealthMetricParam = Literal["RUN_DURATION_SECONDS", "STREAMING_BACKLOG_BYTES", "STREAMING_BACKLOG_RECORDS", "STREAMING_BACKLOG_SECONDS", "STREAMING_BACKLOG_FILES"] | JobsHealthMetric diff --git a/python/databricks/bundles/jobs/_models/jobs_health_operator.py b/python/databricks/bundles/jobs/_models/jobs_health_operator.py index 3de343a8b6a..478e48bb561 100644 --- a/python/databricks/bundles/jobs/_models/jobs_health_operator.py +++ b/python/databricks/bundles/jobs/_models/jobs_health_operator.py @@ -1,15 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class JobsHealthOperator(Enum): """ Specifies the operator used to compare the health metric value with the specified threshold. """ - GREATER_THAN = "GREATER_THAN" - JobsHealthOperatorParam = Literal["GREATER_THAN"] | JobsHealthOperator diff --git a/python/databricks/bundles/jobs/_models/jobs_health_rule.py b/python/databricks/bundles/jobs/_models/jobs_health_rule.py index 59447874eda..ed9f08952c2 100644 --- a/python/databricks/bundles/jobs/_models/jobs_health_rule.py +++ b/python/databricks/bundles/jobs/_models/jobs_health_rule.py @@ -1,24 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr -from databricks.bundles.jobs._models.jobs_health_metric import ( - JobsHealthMetric, - JobsHealthMetricParam, -) -from databricks.bundles.jobs._models.jobs_health_operator import ( - JobsHealthOperator, - JobsHealthOperatorParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobsHealthRule: """""" @@ -45,11 +198,12 @@ class JobsHealthRule: """ @classmethod - def from_dict(cls, value: "JobsHealthRuleDict") -> "Self": + def from_dict(cls, value: 'JobsHealthRuleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobsHealthRuleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobsHealthRuleDict': + return _transform_to_json_value(self) # type:ignore + class JobsHealthRuleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/jobs_health_rules.py b/python/databricks/bundles/jobs/_models/jobs_health_rules.py index daf613406b5..0754e1d108c 100644 --- a/python/databricks/bundles/jobs/_models/jobs_health_rules.py +++ b/python/databricks/bundles/jobs/_models/jobs_health_rules.py @@ -1,34 +1,191 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList -from databricks.bundles.jobs._models.jobs_health_rule import ( - JobsHealthRule, - JobsHealthRuleParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JobsHealthRules: """ An optional set of health rules that can be defined for this job. """ - rules: VariableOrList[JobsHealthRule] = field(default_factory=list) @classmethod - def from_dict(cls, value: "JobsHealthRulesDict") -> "Self": + def from_dict(cls, value: 'JobsHealthRulesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JobsHealthRulesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JobsHealthRulesDict': + return _transform_to_json_value(self) # type:ignore + class JobsHealthRulesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/kind.py b/python/databricks/bundles/jobs/_models/kind.py index 8614e25dd52..15e60a1c5d9 100644 --- a/python/databricks/bundles/jobs/_models/kind.py +++ b/python/databricks/bundles/jobs/_models/kind.py @@ -1,23 +1,185 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class Kind(Enum): """ The kind of compute described by this compute specification. - + Depending on `kind`, different validations and default values will be applied. - + Clusters with `kind = CLASSIC_PREVIEW` support the following fields, whereas clusters with no specified `kind` do not. * [is_single_node](/api/workspace/clusters/create#is_single_node) * [use_ml_runtime](/api/workspace/clusters/create#use_ml_runtime) - + By using the [simple form](https://docs.databricks.com/compute/simple-form.html), your clusters are automatically using `kind = CLASSIC_PREVIEW`. """ - CLASSIC_PREVIEW = "CLASSIC_PREVIEW" - KindParam = Literal["CLASSIC_PREVIEW"] | Kind diff --git a/python/databricks/bundles/jobs/_models/library.py b/python/databricks/bundles/jobs/_models/library.py index 29d4c5b1fdd..a7158011d72 100644 --- a/python/databricks/bundles/jobs/_models/library.py +++ b/python/databricks/bundles/jobs/_models/library.py @@ -1,28 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.maven_library import ( - MavenLibrary, - MavenLibraryParam, -) -from databricks.bundles.jobs._models.python_py_pi_library import ( - PythonPyPiLibrary, - PythonPyPiLibraryParam, -) -from databricks.bundles.jobs._models.r_cran_library import ( - RCranLibrary, - RCranLibraryParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Library: """""" @@ -74,11 +223,12 @@ class Library: """ @classmethod - def from_dict(cls, value: "LibraryDict") -> "Self": + def from_dict(cls, value: 'LibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LibraryDict': + return _transform_to_json_value(self) # type:ignore + class LibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/lifecycle.py b/python/databricks/bundles/jobs/_models/lifecycle.py index 697776a198e..f21f05311e9 100644 --- a/python/databricks/bundles/jobs/_models/lifecycle.py +++ b/python/databricks/bundles/jobs/_models/lifecycle.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +182,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/local_file_info.py b/python/databricks/bundles/jobs/_models/local_file_info.py index 875f2fe8353..1ed6df8e8b9 100644 --- a/python/databricks/bundles/jobs/_models/local_file_info.py +++ b/python/databricks/bundles/jobs/_models/local_file_info.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LocalFileInfo: """""" @@ -21,11 +182,12 @@ class LocalFileInfo: """ @classmethod - def from_dict(cls, value: "LocalFileInfoDict") -> "Self": + def from_dict(cls, value: 'LocalFileInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LocalFileInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LocalFileInfoDict': + return _transform_to_json_value(self) # type:ignore + class LocalFileInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/log_analytics_info.py b/python/databricks/bundles/jobs/_models/log_analytics_info.py index 67cdf85b2b9..1a4646a2154 100644 --- a/python/databricks/bundles/jobs/_models/log_analytics_info.py +++ b/python/databricks/bundles/jobs/_models/log_analytics_info.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LogAnalyticsInfo: """""" @@ -26,11 +187,12 @@ class LogAnalyticsInfo: """ @classmethod - def from_dict(cls, value: "LogAnalyticsInfoDict") -> "Self": + def from_dict(cls, value: 'LogAnalyticsInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LogAnalyticsInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LogAnalyticsInfoDict': + return _transform_to_json_value(self) # type:ignore + class LogAnalyticsInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/maintenance_window.py b/python/databricks/bundles/jobs/_models/maintenance_window.py index 180c65e75cc..0ac9f824faa 100644 --- a/python/databricks/bundles/jobs/_models/maintenance_window.py +++ b/python/databricks/bundles/jobs/_models/maintenance_window.py @@ -1,24 +1,183 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MaintenanceWindow: """ A recurring weekly time window during which platform-initiated maintenance is allowed to run for a continuous job. """ - day_of_week: VariableOr[DayOfWeek] """ The day of week on which maintenance is allowed to happen. This field is required. @@ -37,11 +196,12 @@ class MaintenanceWindow: """ @classmethod - def from_dict(cls, value: "MaintenanceWindowDict") -> "Self": + def from_dict(cls, value: 'MaintenanceWindowDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MaintenanceWindowDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MaintenanceWindowDict': + return _transform_to_json_value(self) # type:ignore + class MaintenanceWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/maven_library.py b/python/databricks/bundles/jobs/_models/maven_library.py index 25be1599c28..681257d6b53 100644 --- a/python/databricks/bundles/jobs/_models/maven_library.py +++ b/python/databricks/bundles/jobs/_models/maven_library.py @@ -1,20 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MavenLibrary: """""" @@ -39,11 +196,12 @@ class MavenLibrary: """ @classmethod - def from_dict(cls, value: "MavenLibraryDict") -> "Self": + def from_dict(cls, value: 'MavenLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MavenLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MavenLibraryDict': + return _transform_to_json_value(self) # type:ignore + class MavenLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/model_trigger_configuration.py b/python/databricks/bundles/jobs/_models/model_trigger_configuration.py index 29ba7ed1479..389371f378d 100644 --- a/python/databricks/bundles/jobs/_models/model_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/model_trigger_configuration.py @@ -1,30 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ( - ModelTriggerConfigurationCondition, - ModelTriggerConfigurationConditionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelTriggerConfiguration: """ :meta private: [EXPERIMENTAL] """ - condition: VariableOr[ModelTriggerConfigurationCondition] """ :meta private: [EXPERIMENTAL] @@ -65,11 +217,12 @@ class ModelTriggerConfiguration: """ @classmethod - def from_dict(cls, value: "ModelTriggerConfigurationDict") -> "Self": + def from_dict(cls, value: 'ModelTriggerConfigurationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelTriggerConfigurationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelTriggerConfigurationDict': + return _transform_to_json_value(self) # type:ignore + class ModelTriggerConfigurationDict(TypedDict, total=False): @@ -115,6 +268,4 @@ class ModelTriggerConfigurationDict(TypedDict, total=False): """ -ModelTriggerConfigurationParam = ( - ModelTriggerConfigurationDict | ModelTriggerConfiguration -) +ModelTriggerConfigurationParam = ModelTriggerConfigurationDict | ModelTriggerConfiguration diff --git a/python/databricks/bundles/jobs/_models/model_trigger_configuration_condition.py b/python/databricks/bundles/jobs/_models/model_trigger_configuration_condition.py index 013f413303b..99adeadbbed 100644 --- a/python/databricks/bundles/jobs/_models/model_trigger_configuration_condition.py +++ b/python/databricks/bundles/jobs/_models/model_trigger_configuration_condition.py @@ -1,20 +1,179 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class ModelTriggerConfigurationCondition(Enum): """ :meta private: [EXPERIMENTAL] """ - MODEL_CREATED = "MODEL_CREATED" MODEL_VERSION_READY = "MODEL_VERSION_READY" MODEL_ALIAS_SET = "MODEL_ALIAS_SET" - -ModelTriggerConfigurationConditionParam = ( - Literal["MODEL_CREATED", "MODEL_VERSION_READY", "MODEL_ALIAS_SET"] - | ModelTriggerConfigurationCondition -) +ModelTriggerConfigurationConditionParam = Literal["MODEL_CREATED", "MODEL_VERSION_READY", "MODEL_ALIAS_SET"] | ModelTriggerConfigurationCondition diff --git a/python/databricks/bundles/jobs/_models/node_type_flexibility.py b/python/databricks/bundles/jobs/_models/node_type_flexibility.py index 418ea9337bd..1eb9aef0a90 100644 --- a/python/databricks/bundles/jobs/_models/node_type_flexibility.py +++ b/python/databricks/bundles/jobs/_models/node_type_flexibility.py @@ -1,22 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class NodeTypeFlexibility: """ Configuration for flexible node types, allowing fallback to alternate node types during cluster launch and upscale. """ - alternate_node_type_ids: VariableOrList[str] = field(default_factory=list) """ A list of node type IDs to use as fallbacks when the primary node type is unavailable. @@ -29,11 +189,12 @@ class NodeTypeFlexibility: """ @classmethod - def from_dict(cls, value: "NodeTypeFlexibilityDict") -> "Self": + def from_dict(cls, value: 'NodeTypeFlexibilityDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "NodeTypeFlexibilityDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'NodeTypeFlexibilityDict': + return _transform_to_json_value(self) # type:ignore + class NodeTypeFlexibilityDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/notebook_task.py b/python/databricks/bundles/jobs/_models/notebook_task.py index 7b4329392db..f35946725a3 100644 --- a/python/databricks/bundles/jobs/_models/notebook_task.py +++ b/python/databricks/bundles/jobs/_models/notebook_task.py @@ -1,21 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + from databricks.bundles.jobs._models.source import Source, SourceParam +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class NotebookTask: """""" @@ -58,11 +214,12 @@ class NotebookTask: """ @classmethod - def from_dict(cls, value: "NotebookTaskDict") -> "Self": + def from_dict(cls, value: 'NotebookTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "NotebookTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'NotebookTaskDict': + return _transform_to_json_value(self) # type:ignore + class NotebookTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/pause_status.py b/python/databricks/bundles/jobs/_models/pause_status.py index 9e5044fb880..19bb45bf6ac 100644 --- a/python/databricks/bundles/jobs/_models/pause_status.py +++ b/python/databricks/bundles/jobs/_models/pause_status.py @@ -1,12 +1,175 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class PauseStatus(Enum): UNPAUSED = "UNPAUSED" PAUSED = "PAUSED" - PauseStatusParam = Literal["UNPAUSED", "PAUSED"] | PauseStatus diff --git a/python/databricks/bundles/jobs/_models/performance_target.py b/python/databricks/bundles/jobs/_models/performance_target.py index 8dbe7e4a435..b26e2ed6d46 100644 --- a/python/databricks/bundles/jobs/_models/performance_target.py +++ b/python/databricks/bundles/jobs/_models/performance_target.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class PerformanceTarget(Enum): @@ -10,11 +174,7 @@ class PerformanceTarget(Enum): The performance mode on the job or pipeline should map to a performance setting that is passed to Cluster Manager (see cluster-common PerformanceTarget). """ - PERFORMANCE_OPTIMIZED = "PERFORMANCE_OPTIMIZED" STANDARD = "STANDARD" - -PerformanceTargetParam = ( - Literal["PERFORMANCE_OPTIMIZED", "STANDARD"] | PerformanceTarget -) +PerformanceTargetParam = Literal["PERFORMANCE_OPTIMIZED", "STANDARD"] | PerformanceTarget diff --git a/python/databricks/bundles/jobs/_models/periodic_trigger_configuration.py b/python/databricks/bundles/jobs/_models/periodic_trigger_configuration.py index 2e7215aad06..288f44df438 100644 --- a/python/databricks/bundles/jobs/_models/periodic_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/periodic_trigger_configuration.py @@ -1,20 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import ( - PeriodicTriggerConfigurationTimeUnit, - PeriodicTriggerConfigurationTimeUnitParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PeriodicTriggerConfiguration: """""" @@ -30,11 +187,12 @@ class PeriodicTriggerConfiguration: """ @classmethod - def from_dict(cls, value: "PeriodicTriggerConfigurationDict") -> "Self": + def from_dict(cls, value: 'PeriodicTriggerConfigurationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PeriodicTriggerConfigurationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PeriodicTriggerConfigurationDict': + return _transform_to_json_value(self) # type:ignore + class PeriodicTriggerConfigurationDict(TypedDict, total=False): @@ -51,6 +209,4 @@ class PeriodicTriggerConfigurationDict(TypedDict, total=False): """ -PeriodicTriggerConfigurationParam = ( - PeriodicTriggerConfigurationDict | PeriodicTriggerConfiguration -) +PeriodicTriggerConfigurationParam = PeriodicTriggerConfigurationDict | PeriodicTriggerConfiguration diff --git a/python/databricks/bundles/jobs/_models/periodic_trigger_configuration_time_unit.py b/python/databricks/bundles/jobs/_models/periodic_trigger_configuration_time_unit.py index e035958b12c..58680def8f0 100644 --- a/python/databricks/bundles/jobs/_models/periodic_trigger_configuration_time_unit.py +++ b/python/databricks/bundles/jobs/_models/periodic_trigger_configuration_time_unit.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class PeriodicTriggerConfigurationTimeUnit(Enum): @@ -10,7 +174,4 @@ class PeriodicTriggerConfigurationTimeUnit(Enum): WEEKS = "WEEKS" MINUTES = "MINUTES" - -PeriodicTriggerConfigurationTimeUnitParam = ( - Literal["HOURS", "DAYS", "WEEKS", "MINUTES"] | PeriodicTriggerConfigurationTimeUnit -) +PeriodicTriggerConfigurationTimeUnitParam = Literal["HOURS", "DAYS", "WEEKS", "MINUTES"] | PeriodicTriggerConfigurationTimeUnit diff --git a/python/databricks/bundles/jobs/_models/pipeline_params.py b/python/databricks/bundles/jobs/_models/pipeline_params.py index 98550f271fb..c554bdfa76b 100644 --- a/python/databricks/bundles/jobs/_models/pipeline_params.py +++ b/python/databricks/bundles/jobs/_models/pipeline_params.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PipelineParams: """""" @@ -42,11 +203,12 @@ class PipelineParams: """ @classmethod - def from_dict(cls, value: "PipelineParamsDict") -> "Self": + def from_dict(cls, value: 'PipelineParamsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PipelineParamsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PipelineParamsDict': + return _transform_to_json_value(self) # type:ignore + class PipelineParamsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/pipeline_task.py b/python/databricks/bundles/jobs/_models/pipeline_task.py index 87d720fcd69..4af5e8a39d6 100644 --- a/python/databricks/bundles/jobs/_models/pipeline_task.py +++ b/python/databricks/bundles/jobs/_models/pipeline_task.py @@ -1,21 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PipelineTask: """""" @@ -60,11 +216,12 @@ class PipelineTask: """ @classmethod - def from_dict(cls, value: "PipelineTaskDict") -> "Self": + def from_dict(cls, value: 'PipelineTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PipelineTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PipelineTaskDict': + return _transform_to_json_value(self) # type:ignore + class PipelineTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/power_bi_model.py b/python/databricks/bundles/jobs/_models/power_bi_model.py index f791b1901a8..d30147df0be 100644 --- a/python/databricks/bundles/jobs/_models/power_bi_model.py +++ b/python/databricks/bundles/jobs/_models/power_bi_model.py @@ -1,21 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.authentication_method import ( - AuthenticationMethod, - AuthenticationMethodParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PowerBiModel: """""" @@ -46,11 +202,12 @@ class PowerBiModel: """ @classmethod - def from_dict(cls, value: "PowerBiModelDict") -> "Self": + def from_dict(cls, value: 'PowerBiModelDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PowerBiModelDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PowerBiModelDict': + return _transform_to_json_value(self) # type:ignore + class PowerBiModelDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/power_bi_table.py b/python/databricks/bundles/jobs/_models/power_bi_table.py index 368f432c45d..e7df4faff75 100644 --- a/python/databricks/bundles/jobs/_models/power_bi_table.py +++ b/python/databricks/bundles/jobs/_models/power_bi_table.py @@ -1,17 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PowerBiTable: """""" @@ -37,11 +197,12 @@ class PowerBiTable: """ @classmethod - def from_dict(cls, value: "PowerBiTableDict") -> "Self": + def from_dict(cls, value: 'PowerBiTableDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PowerBiTableDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PowerBiTableDict': + return _transform_to_json_value(self) # type:ignore + class PowerBiTableDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/power_bi_task.py b/python/databricks/bundles/jobs/_models/power_bi_task.py index cd6246cbca8..d2fa7428816 100644 --- a/python/databricks/bundles/jobs/_models/power_bi_task.py +++ b/python/databricks/bundles/jobs/_models/power_bi_task.py @@ -1,24 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.jobs._models.power_bi_model import ( - PowerBiModel, - PowerBiModelParam, -) -from databricks.bundles.jobs._models.power_bi_table import ( - PowerBiTable, - PowerBiTableParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PowerBiTask: """""" @@ -49,11 +202,12 @@ class PowerBiTask: """ @classmethod - def from_dict(cls, value: "PowerBiTaskDict") -> "Self": + def from_dict(cls, value: 'PowerBiTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PowerBiTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PowerBiTaskDict': + return _transform_to_json_value(self) # type:ignore + class PowerBiTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/python_operator_task.py b/python/databricks/bundles/jobs/_models/python_operator_task.py index a2ca2be4c3b..b49a402a573 100644 --- a/python/databricks/bundles/jobs/_models/python_operator_task.py +++ b/python/databricks/bundles/jobs/_models/python_operator_task.py @@ -1,26 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.jobs._models.python_operator_task_parameter import ( - PythonOperatorTaskParameter, - PythonOperatorTaskParameterParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PythonOperatorTask: """ :meta private: [EXPERIMENTAL] """ - main: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -29,9 +185,7 @@ class PythonOperatorTask: For example, `my_project.my_function` or `my_project.MyOperator`. """ - parameters: VariableOrList[PythonOperatorTaskParameter] = field( - default_factory=list - ) + parameters: VariableOrList[PythonOperatorTaskParameter] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -40,11 +194,12 @@ class PythonOperatorTask: """ @classmethod - def from_dict(cls, value: "PythonOperatorTaskDict") -> "Self": + def from_dict(cls, value: 'PythonOperatorTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PythonOperatorTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PythonOperatorTaskDict': + return _transform_to_json_value(self) # type:ignore + class PythonOperatorTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/python_operator_task_parameter.py b/python/databricks/bundles/jobs/_models/python_operator_task_parameter.py index 867dff0db64..865c01a0867 100644 --- a/python/databricks/bundles/jobs/_models/python_operator_task_parameter.py +++ b/python/databricks/bundles/jobs/_models/python_operator_task_parameter.py @@ -1,22 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PythonOperatorTaskParameter: """ :meta private: [EXPERIMENTAL] """ - name: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -32,11 +192,12 @@ class PythonOperatorTaskParameter: """ @classmethod - def from_dict(cls, value: "PythonOperatorTaskParameterDict") -> "Self": + def from_dict(cls, value: 'PythonOperatorTaskParameterDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PythonOperatorTaskParameterDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PythonOperatorTaskParameterDict': + return _transform_to_json_value(self) # type:ignore + class PythonOperatorTaskParameterDict(TypedDict, total=False): @@ -57,6 +218,4 @@ class PythonOperatorTaskParameterDict(TypedDict, total=False): """ -PythonOperatorTaskParameterParam = ( - PythonOperatorTaskParameterDict | PythonOperatorTaskParameter -) +PythonOperatorTaskParameterParam = PythonOperatorTaskParameterDict | PythonOperatorTaskParameter diff --git a/python/databricks/bundles/jobs/_models/python_py_pi_library.py b/python/databricks/bundles/jobs/_models/python_py_pi_library.py index 037c2029f50..782396b06b4 100644 --- a/python/databricks/bundles/jobs/_models/python_py_pi_library.py +++ b/python/databricks/bundles/jobs/_models/python_py_pi_library.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PythonPyPiLibrary: """""" @@ -28,11 +189,12 @@ class PythonPyPiLibrary: """ @classmethod - def from_dict(cls, value: "PythonPyPiLibraryDict") -> "Self": + def from_dict(cls, value: 'PythonPyPiLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PythonPyPiLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PythonPyPiLibraryDict': + return _transform_to_json_value(self) # type:ignore + class PythonPyPiLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/python_wheel_task.py b/python/databricks/bundles/jobs/_models/python_wheel_task.py index ce2dea3ea15..3417ad3e222 100644 --- a/python/databricks/bundles/jobs/_models/python_wheel_task.py +++ b/python/databricks/bundles/jobs/_models/python_wheel_task.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrDict, VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PythonWheelTask: """""" @@ -36,11 +197,12 @@ class PythonWheelTask: """ @classmethod - def from_dict(cls, value: "PythonWheelTaskDict") -> "Self": + def from_dict(cls, value: 'PythonWheelTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PythonWheelTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PythonWheelTaskDict': + return _transform_to_json_value(self) # type:ignore + class PythonWheelTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/queue_settings.py b/python/databricks/bundles/jobs/_models/queue_settings.py index a72921aed96..2f201600a6e 100644 --- a/python/databricks/bundles/jobs/_models/queue_settings.py +++ b/python/databricks/bundles/jobs/_models/queue_settings.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class QueueSettings: """""" @@ -21,11 +182,12 @@ class QueueSettings: """ @classmethod - def from_dict(cls, value: "QueueSettingsDict") -> "Self": + def from_dict(cls, value: 'QueueSettingsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "QueueSettingsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'QueueSettingsDict': + return _transform_to_json_value(self) # type:ignore + class QueueSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/r_cran_library.py b/python/databricks/bundles/jobs/_models/r_cran_library.py index 0bd2c4d5296..e13423f6fbe 100644 --- a/python/databricks/bundles/jobs/_models/r_cran_library.py +++ b/python/databricks/bundles/jobs/_models/r_cran_library.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RCranLibrary: """""" @@ -26,11 +187,12 @@ class RCranLibrary: """ @classmethod - def from_dict(cls, value: "RCranLibraryDict") -> "Self": + def from_dict(cls, value: 'RCranLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RCranLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RCranLibraryDict': + return _transform_to_json_value(self) # type:ignore + class RCranLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/run_if.py b/python/databricks/bundles/jobs/_models/run_if.py index fdbf95b1bc2..4e34c30dd61 100644 --- a/python/databricks/bundles/jobs/_models/run_if.py +++ b/python/databricks/bundles/jobs/_models/run_if.py @@ -1,13 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class RunIf(Enum): """ An optional value indicating the condition that determines whether the task should be run once its dependencies have been completed. When omitted, defaults to `ALL_SUCCESS`. - + Possible values are: * `ALL_SUCCESS`: All dependencies have executed and succeeded * `AT_LEAST_ONE_SUCCESS`: At least one dependency has succeeded @@ -16,7 +180,6 @@ class RunIf(Enum): * `AT_LEAST_ONE_FAILED`: At least one dependency failed * `ALL_FAILED`: ALl dependencies have failed """ - ALL_SUCCESS = "ALL_SUCCESS" ALL_DONE = "ALL_DONE" NONE_FAILED = "NONE_FAILED" @@ -24,15 +187,4 @@ class RunIf(Enum): ALL_FAILED = "ALL_FAILED" AT_LEAST_ONE_FAILED = "AT_LEAST_ONE_FAILED" - -RunIfParam = ( - Literal[ - "ALL_SUCCESS", - "ALL_DONE", - "NONE_FAILED", - "AT_LEAST_ONE_SUCCESS", - "ALL_FAILED", - "AT_LEAST_ONE_FAILED", - ] - | RunIf -) +RunIfParam = Literal["ALL_SUCCESS", "ALL_DONE", "NONE_FAILED", "AT_LEAST_ONE_SUCCESS", "ALL_FAILED", "AT_LEAST_ONE_FAILED"] | RunIf diff --git a/python/databricks/bundles/jobs/_models/run_job_task.py b/python/databricks/bundles/jobs/_models/run_job_task.py index 24d0246a8a7..78877f5fcce 100644 --- a/python/databricks/bundles/jobs/_models/run_job_task.py +++ b/python/databricks/bundles/jobs/_models/run_job_task.py @@ -1,24 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrOptional, -) -from databricks.bundles.jobs._models.pipeline_params import ( - PipelineParams, - PipelineParamsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RunJobTask: """""" @@ -39,11 +192,12 @@ class RunJobTask: """ @classmethod - def from_dict(cls, value: "RunJobTaskDict") -> "Self": + def from_dict(cls, value: 'RunJobTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RunJobTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RunJobTaskDict': + return _transform_to_json_value(self) # type:ignore + class RunJobTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/runtime_engine.py b/python/databricks/bundles/jobs/_models/runtime_engine.py index 2e1559fa801..5138fc821cc 100644 --- a/python/databricks/bundles/jobs/_models/runtime_engine.py +++ b/python/databricks/bundles/jobs/_models/runtime_engine.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class RuntimeEngine(Enum): @@ -9,5 +173,4 @@ class RuntimeEngine(Enum): STANDARD = "STANDARD" PHOTON = "PHOTON" - RuntimeEngineParam = Literal["NULL", "STANDARD", "PHOTON"] | RuntimeEngine diff --git a/python/databricks/bundles/jobs/_models/s3_storage_info.py b/python/databricks/bundles/jobs/_models/s3_storage_info.py index 071abd5b6f3..4818f09c213 100644 --- a/python/databricks/bundles/jobs/_models/s3_storage_info.py +++ b/python/databricks/bundles/jobs/_models/s3_storage_info.py @@ -1,22 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class S3StorageInfo: """ A storage location in Amazon S3 """ - destination: VariableOr[str] """ S3 destination, e.g. `s3://my-bucket/some-prefix` Note that logs will be delivered using @@ -64,11 +224,12 @@ class S3StorageInfo: """ @classmethod - def from_dict(cls, value: "S3StorageInfoDict") -> "Self": + def from_dict(cls, value: 'S3StorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "S3StorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'S3StorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class S3StorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/source.py b/python/databricks/bundles/jobs/_models/source.py index 380ab711022..4393bf79e24 100644 --- a/python/databricks/bundles/jobs/_models/source.py +++ b/python/databricks/bundles/jobs/_models/source.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class Source(Enum): @@ -13,9 +177,7 @@ class Source(Enum): * `WORKSPACE`: SQL file is located in Databricks workspace. * `GIT`: SQL file is located in cloud Git provider. """ - WORKSPACE = "WORKSPACE" GIT = "GIT" - SourceParam = Literal["WORKSPACE", "GIT"] | Source diff --git a/python/databricks/bundles/jobs/_models/spark_jar_task.py b/python/databricks/bundles/jobs/_models/spark_jar_task.py index d782da867f3..8522ffed962 100644 --- a/python/databricks/bundles/jobs/_models/spark_jar_task.py +++ b/python/databricks/bundles/jobs/_models/spark_jar_task.py @@ -1,20 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SparkJarTask: """""" @@ -46,11 +203,12 @@ class SparkJarTask: """ @classmethod - def from_dict(cls, value: "SparkJarTaskDict") -> "Self": + def from_dict(cls, value: 'SparkJarTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SparkJarTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SparkJarTaskDict': + return _transform_to_json_value(self) # type:ignore + class SparkJarTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/spark_python_task.py b/python/databricks/bundles/jobs/_models/spark_python_task.py index 001d9287b1c..efa1c04d656 100644 --- a/python/databricks/bundles/jobs/_models/spark_python_task.py +++ b/python/databricks/bundles/jobs/_models/spark_python_task.py @@ -1,21 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + from databricks.bundles.jobs._models.source import Source, SourceParam +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SparkPythonTask: """""" @@ -43,11 +199,12 @@ class SparkPythonTask: """ @classmethod - def from_dict(cls, value: "SparkPythonTaskDict") -> "Self": + def from_dict(cls, value: 'SparkPythonTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SparkPythonTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SparkPythonTaskDict': + return _transform_to_json_value(self) # type:ignore + class SparkPythonTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/spark_submit_task.py b/python/databricks/bundles/jobs/_models/spark_submit_task.py index 2acc389b91b..358a7248f7d 100644 --- a/python/databricks/bundles/jobs/_models/spark_submit_task.py +++ b/python/databricks/bundles/jobs/_models/spark_submit_task.py @@ -1,22 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SparkSubmitTask: """ - [DEPRECATED] + [DEPRECATED] """ - parameters: VariableOrList[str] = field(default_factory=list) """ Command-line parameters passed to spark submit. @@ -25,11 +185,12 @@ class SparkSubmitTask: """ @classmethod - def from_dict(cls, value: "SparkSubmitTaskDict") -> "Self": + def from_dict(cls, value: 'SparkSubmitTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SparkSubmitTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SparkSubmitTaskDict': + return _transform_to_json_value(self) # type:ignore + class SparkSubmitTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sparse_checkout.py b/python/databricks/bundles/jobs/_models/sparse_checkout.py index a42a78e49f1..232518a268c 100644 --- a/python/databricks/bundles/jobs/_models/sparse_checkout.py +++ b/python/databricks/bundles/jobs/_models/sparse_checkout.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SparseCheckout: """""" @@ -21,11 +182,12 @@ class SparseCheckout: """ @classmethod - def from_dict(cls, value: "SparseCheckoutDict") -> "Self": + def from_dict(cls, value: 'SparseCheckoutDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SparseCheckoutDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SparseCheckoutDict': + return _transform_to_json_value(self) # type:ignore + class SparseCheckoutDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_condition_configuration.py b/python/databricks/bundles/jobs/_models/sql_condition_configuration.py index f6b2da80685..028bb176dbe 100644 --- a/python/databricks/bundles/jobs/_models/sql_condition_configuration.py +++ b/python/databricks/bundles/jobs/_models/sql_condition_configuration.py @@ -1,26 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.jobs._models.sql_condition_trigger_mode import ( - SqlConditionTriggerMode, - SqlConditionTriggerModeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SqlConditionConfiguration: """ :meta private: [EXPERIMENTAL] """ - sql_query_id: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -47,11 +203,12 @@ class SqlConditionConfiguration: """ @classmethod - def from_dict(cls, value: "SqlConditionConfigurationDict") -> "Self": + def from_dict(cls, value: 'SqlConditionConfigurationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SqlConditionConfigurationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SqlConditionConfigurationDict': + return _transform_to_json_value(self) # type:ignore + class SqlConditionConfigurationDict(TypedDict, total=False): @@ -83,6 +240,4 @@ class SqlConditionConfigurationDict(TypedDict, total=False): """ -SqlConditionConfigurationParam = ( - SqlConditionConfigurationDict | SqlConditionConfiguration -) +SqlConditionConfigurationParam = SqlConditionConfigurationDict | SqlConditionConfiguration diff --git a/python/databricks/bundles/jobs/_models/sql_condition_trigger_mode.py b/python/databricks/bundles/jobs/_models/sql_condition_trigger_mode.py index c2b8472faea..5548ee01618 100644 --- a/python/databricks/bundles/jobs/_models/sql_condition_trigger_mode.py +++ b/python/databricks/bundles/jobs/_models/sql_condition_trigger_mode.py @@ -1,15 +1,179 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class SqlConditionTriggerMode(Enum): """ :meta private: [EXPERIMENTAL] - + The strategy used to evaluate a SQL condition trigger against a query result set. - + * `SQL_CONDITION_TRIGGER_MODE_UNSPECIFIED`: Sentinel zero-value. Not a valid input — the validator rejects this when sent explicitly. Internally treated as `QUERY_RETURNS_ROWS` when reading legacy data that predates this field. @@ -19,11 +183,7 @@ class SqlConditionTriggerMode(Enum): previous evaluation. The first evaluation always fires. Queries must return exactly one cell (one row, one column). """ - QUERY_RETURNS_ROWS = "QUERY_RETURNS_ROWS" RESULT_VALUE_CHANGES = "RESULT_VALUE_CHANGES" - -SqlConditionTriggerModeParam = ( - Literal["QUERY_RETURNS_ROWS", "RESULT_VALUE_CHANGES"] | SqlConditionTriggerMode -) +SqlConditionTriggerModeParam = Literal["QUERY_RETURNS_ROWS", "RESULT_VALUE_CHANGES"] | SqlConditionTriggerMode diff --git a/python/databricks/bundles/jobs/_models/sql_task.py b/python/databricks/bundles/jobs/_models/sql_task.py index dd731928af0..ea97314e1d7 100644 --- a/python/databricks/bundles/jobs/_models/sql_task.py +++ b/python/databricks/bundles/jobs/_models/sql_task.py @@ -1,33 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrOptional, -) -from databricks.bundles.jobs._models.sql_task_alert import ( - SqlTaskAlert, - SqlTaskAlertParam, -) -from databricks.bundles.jobs._models.sql_task_dashboard import ( - SqlTaskDashboard, - SqlTaskDashboardParam, -) -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import ( - SqlTaskQuery, - SqlTaskQueryParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SqlTask: """""" @@ -63,11 +207,12 @@ class SqlTask: """ @classmethod - def from_dict(cls, value: "SqlTaskDict") -> "Self": + def from_dict(cls, value: 'SqlTaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SqlTaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SqlTaskDict': + return _transform_to_json_value(self) # type:ignore + class SqlTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_alert.py b/python/databricks/bundles/jobs/_models/sql_task_alert.py index 6db18d1c876..0fa6e6141e7 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_alert.py +++ b/python/databricks/bundles/jobs/_models/sql_task_alert.py @@ -1,24 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.jobs._models.sql_task_subscription import ( - SqlTaskSubscription, - SqlTaskSubscriptionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SqlTaskAlert: """""" @@ -39,11 +192,12 @@ class SqlTaskAlert: """ @classmethod - def from_dict(cls, value: "SqlTaskAlertDict") -> "Self": + def from_dict(cls, value: 'SqlTaskAlertDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SqlTaskAlertDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SqlTaskAlertDict': + return _transform_to_json_value(self) # type:ignore + class SqlTaskAlertDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_dashboard.py b/python/databricks/bundles/jobs/_models/sql_task_dashboard.py index 98e544e5aa0..5f4970e2a37 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_dashboard.py +++ b/python/databricks/bundles/jobs/_models/sql_task_dashboard.py @@ -1,24 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.jobs._models.sql_task_subscription import ( - SqlTaskSubscription, - SqlTaskSubscriptionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SqlTaskDashboard: """""" @@ -44,11 +197,12 @@ class SqlTaskDashboard: """ @classmethod - def from_dict(cls, value: "SqlTaskDashboardDict") -> "Self": + def from_dict(cls, value: 'SqlTaskDashboardDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SqlTaskDashboardDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SqlTaskDashboardDict': + return _transform_to_json_value(self) # type:ignore + class SqlTaskDashboardDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_file.py b/python/databricks/bundles/jobs/_models/sql_task_file.py index 8a49f54c7b7..b9be171537f 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_file.py +++ b/python/databricks/bundles/jobs/_models/sql_task_file.py @@ -1,17 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + from databricks.bundles.jobs._models.source import Source, SourceParam +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SqlTaskFile: """""" @@ -32,11 +192,12 @@ class SqlTaskFile: """ @classmethod - def from_dict(cls, value: "SqlTaskFileDict") -> "Self": + def from_dict(cls, value: 'SqlTaskFileDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SqlTaskFileDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SqlTaskFileDict': + return _transform_to_json_value(self) # type:ignore + class SqlTaskFileDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_query.py b/python/databricks/bundles/jobs/_models/sql_task_query.py index 2682c425112..4ef7aed0ea0 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_query.py +++ b/python/databricks/bundles/jobs/_models/sql_task_query.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SqlTaskQuery: """""" @@ -21,11 +182,12 @@ class SqlTaskQuery: """ @classmethod - def from_dict(cls, value: "SqlTaskQueryDict") -> "Self": + def from_dict(cls, value: 'SqlTaskQueryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SqlTaskQueryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SqlTaskQueryDict': + return _transform_to_json_value(self) # type:ignore + class SqlTaskQueryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_subscription.py b/python/databricks/bundles/jobs/_models/sql_task_subscription.py index d5f2a65ddf5..340bbceff70 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_subscription.py +++ b/python/databricks/bundles/jobs/_models/sql_task_subscription.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SqlTaskSubscription: """""" @@ -26,11 +187,12 @@ class SqlTaskSubscription: """ @classmethod - def from_dict(cls, value: "SqlTaskSubscriptionDict") -> "Self": + def from_dict(cls, value: 'SqlTaskSubscriptionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SqlTaskSubscriptionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SqlTaskSubscriptionDict': + return _transform_to_json_value(self) # type:ignore + class SqlTaskSubscriptionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/storage_mode.py b/python/databricks/bundles/jobs/_models/storage_mode.py index f15a94689d6..c084b16175b 100644 --- a/python/databricks/bundles/jobs/_models/storage_mode.py +++ b/python/databricks/bundles/jobs/_models/storage_mode.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class StorageMode(Enum): @@ -9,5 +173,4 @@ class StorageMode(Enum): IMPORT = "IMPORT" DUAL = "DUAL" - StorageModeParam = Literal["DIRECT_QUERY", "IMPORT", "DUAL"] | StorageMode diff --git a/python/databricks/bundles/jobs/_models/subscription.py b/python/databricks/bundles/jobs/_models/subscription.py index 3c8d872f2f2..eceeafecebd 100644 --- a/python/databricks/bundles/jobs/_models/subscription.py +++ b/python/databricks/bundles/jobs/_models/subscription.py @@ -1,24 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.jobs._models.subscription_attachment import ( - SubscriptionAttachment, - SubscriptionAttachmentParam, -) -from databricks.bundles.jobs._models.subscription_subscriber import ( - SubscriptionSubscriber, - SubscriptionSubscriberParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Subscription: """""" @@ -27,7 +180,7 @@ class Subscription: """ :meta private: [EXPERIMENTAL] - [Private Preview] Optional: The configuration for the files attached to the dashboard snapshot notification. It + [Beta] Optional: The configuration for the files attached to the dashboard snapshot notification. It controls the underlying data of dashboard widgets that is attached as data files and which pages are included in the attached PDF. If not specified, the notification is sent with only the default snapshot. @@ -50,11 +203,12 @@ class Subscription: """ @classmethod - def from_dict(cls, value: "SubscriptionDict") -> "Self": + def from_dict(cls, value: 'SubscriptionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SubscriptionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SubscriptionDict': + return _transform_to_json_value(self) # type:ignore + class SubscriptionDict(TypedDict, total=False): @@ -64,7 +218,7 @@ class SubscriptionDict(TypedDict, total=False): """ :meta private: [EXPERIMENTAL] - [Private Preview] Optional: The configuration for the files attached to the dashboard snapshot notification. It + [Beta] Optional: The configuration for the files attached to the dashboard snapshot notification. It controls the underlying data of dashboard widgets that is attached as data files and which pages are included in the attached PDF. If not specified, the notification is sent with only the default snapshot. diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment.py b/python/databricks/bundles/jobs/_models/subscription_attachment.py index 4b51ea04602..f5e2c3b5252 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment.py @@ -1,40 +1,190 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import ( - SubscriptionAttachmentDataAttachment, - SubscriptionAttachmentDataAttachmentParam, -) -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import ( - SubscriptionAttachmentPageAttachment, - SubscriptionAttachmentPageAttachmentParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SubscriptionAttachment: """ - :meta private: [EXPERIMENTAL] - The configuration for the files attached to the dashboard snapshot notification sent to subscribers. `data_attachment` describes the underlying data of dashboard widgets attached as data files, and `page_attachment` describes which dashboard pages are included in the attached PDF. """ - data_attachment: VariableOrOptional[SubscriptionAttachmentDataAttachment] = None """ :meta private: [EXPERIMENTAL] - [Private Preview] Optional: The configuration for the underlying data of dashboard widgets attached as tabular + [Beta] Optional: The configuration for the underlying data of dashboard widgets attached as tabular data files. If not specified, no tabular data is attached. """ @@ -42,16 +192,17 @@ class SubscriptionAttachment: """ :meta private: [EXPERIMENTAL] - [Private Preview] Optional: The selection of dashboard pages included in the attached PDF. If not specified, + [Beta] Optional: The selection of dashboard pages included in the attached PDF. If not specified, only the first page is included. """ @classmethod - def from_dict(cls, value: "SubscriptionAttachmentDict") -> "Self": + def from_dict(cls, value: 'SubscriptionAttachmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SubscriptionAttachmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SubscriptionAttachmentDict': + return _transform_to_json_value(self) # type:ignore + class SubscriptionAttachmentDict(TypedDict, total=False): @@ -61,7 +212,7 @@ class SubscriptionAttachmentDict(TypedDict, total=False): """ :meta private: [EXPERIMENTAL] - [Private Preview] Optional: The configuration for the underlying data of dashboard widgets attached as tabular + [Beta] Optional: The configuration for the underlying data of dashboard widgets attached as tabular data files. If not specified, no tabular data is attached. """ @@ -69,7 +220,7 @@ class SubscriptionAttachmentDict(TypedDict, total=False): """ :meta private: [EXPERIMENTAL] - [Private Preview] Optional: The selection of dashboard pages included in the attached PDF. If not specified, + [Beta] Optional: The selection of dashboard pages included in the attached PDF. If not specified, only the first page is included. """ diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment.py b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment.py index b1cb52362b1..6e7f899176c 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment.py @@ -1,68 +1,213 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import ( - SubscriptionAttachmentDataAttachmentFileFormat, - SubscriptionAttachmentDataAttachmentFileFormatParam, -) -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import ( - SubscriptionAttachmentDataAttachmentWidget, - SubscriptionAttachmentDataAttachmentWidgetParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SubscriptionAttachmentDataAttachment: """ - :meta private: [EXPERIMENTAL] - The configuration for the tabular data files attached to the dashboard snapshot. `widgets` lists the widgets whose data is exported and `file_format` is the output format; each referenced widget's data is exported as a separate file. """ - file_format: VariableOr[SubscriptionAttachmentDataAttachmentFileFormat] """ :meta private: [EXPERIMENTAL] - [Private Preview] The file format of the exported tabular data files. + [Beta] The file format of the exported tabular data files. """ include_applied_filters: VariableOrOptional[bool] = None """ :meta private: [EXPERIMENTAL] - [Private Preview] When true, the dashboard's applied filter values are included alongside the tabular data as + [Beta] When true, the dashboard's applied filter values are included alongside the tabular data as an additional attached file. """ - widgets: VariableOrList[SubscriptionAttachmentDataAttachmentWidget] = field( - default_factory=list - ) + widgets: VariableOrList[SubscriptionAttachmentDataAttachmentWidget] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] - [Private Preview] The widgets whose underlying data is exported. At least one widget must be referenced. + [Beta] The widgets whose underlying data is exported. At least one widget must be referenced. """ @classmethod - def from_dict(cls, value: "SubscriptionAttachmentDataAttachmentDict") -> "Self": + def from_dict(cls, value: 'SubscriptionAttachmentDataAttachmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SubscriptionAttachmentDataAttachmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SubscriptionAttachmentDataAttachmentDict': + return _transform_to_json_value(self) # type:ignore + class SubscriptionAttachmentDataAttachmentDict(TypedDict, total=False): @@ -72,14 +217,14 @@ class SubscriptionAttachmentDataAttachmentDict(TypedDict, total=False): """ :meta private: [EXPERIMENTAL] - [Private Preview] The file format of the exported tabular data files. + [Beta] The file format of the exported tabular data files. """ include_applied_filters: VariableOrOptional[bool] """ :meta private: [EXPERIMENTAL] - [Private Preview] When true, the dashboard's applied filter values are included alongside the tabular data as + [Beta] When true, the dashboard's applied filter values are included alongside the tabular data as an additional attached file. """ @@ -87,10 +232,8 @@ class SubscriptionAttachmentDataAttachmentDict(TypedDict, total=False): """ :meta private: [EXPERIMENTAL] - [Private Preview] The widgets whose underlying data is exported. At least one widget must be referenced. + [Beta] The widgets whose underlying data is exported. At least one widget must be referenced. """ -SubscriptionAttachmentDataAttachmentParam = ( - SubscriptionAttachmentDataAttachmentDict | SubscriptionAttachmentDataAttachment -) +SubscriptionAttachmentDataAttachmentParam = SubscriptionAttachmentDataAttachmentDict | SubscriptionAttachmentDataAttachment diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_file_format.py b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_file_format.py index 659ff905aab..7d618540f87 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_file_format.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_file_format.py @@ -1,23 +1,180 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class SubscriptionAttachmentDataAttachmentFileFormat(Enum): """ - :meta private: [EXPERIMENTAL] - The output format of an exported tabular data file. Must be set to a value other than `FILE_FORMAT_UNSPECIFIED`. """ - FILE_FORMAT_CSV = "FILE_FORMAT_CSV" FILE_FORMAT_TSV = "FILE_FORMAT_TSV" FILE_FORMAT_EXCEL = "FILE_FORMAT_EXCEL" - -SubscriptionAttachmentDataAttachmentFileFormatParam = ( - Literal["FILE_FORMAT_CSV", "FILE_FORMAT_TSV", "FILE_FORMAT_EXCEL"] - | SubscriptionAttachmentDataAttachmentFileFormat -) +SubscriptionAttachmentDataAttachmentFileFormatParam = Literal["FILE_FORMAT_CSV", "FILE_FORMAT_TSV", "FILE_FORMAT_EXCEL"] | SubscriptionAttachmentDataAttachmentFileFormat diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_widget.py b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_widget.py index eeb0c29213c..ce56890dd9d 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_widget.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_widget.py @@ -1,31 +1,189 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SubscriptionAttachmentDataAttachmentWidget: """ - :meta private: [EXPERIMENTAL] - Identifies a dashboard widget whose data to attach. `page_id` and `widget_id` are the URL identifiers from the dashboard's serialized definition returned by the Get dashboard API (`serialized_dashboard`), and together they uniquely identify the widget. """ - page_id: VariableOr[str] """ :meta private: [EXPERIMENTAL] - [Private Preview] The URL identifier of the page containing the widget. This is the value of the `name` + [Beta] The URL identifier of the page containing the widget. This is the value of the `name` field of the corresponding entry under `pages` in the serialized dashboard definition. Format: alphanumeric, hyphen, and underscore characters allowed, max length 63. """ @@ -34,19 +192,18 @@ class SubscriptionAttachmentDataAttachmentWidget: """ :meta private: [EXPERIMENTAL] - [Private Preview] The URL identifier of the widget. This is the value of the `name` field of the + [Beta] The URL identifier of the widget. This is the value of the `name` field of the corresponding widget under that page's `layout` in the serialized dashboard definition. Format: alphanumeric, hyphen, and underscore characters allowed, max length 63. """ @classmethod - def from_dict( - cls, value: "SubscriptionAttachmentDataAttachmentWidgetDict" - ) -> "Self": + def from_dict(cls, value: 'SubscriptionAttachmentDataAttachmentWidgetDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SubscriptionAttachmentDataAttachmentWidgetDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SubscriptionAttachmentDataAttachmentWidgetDict': + return _transform_to_json_value(self) # type:ignore + class SubscriptionAttachmentDataAttachmentWidgetDict(TypedDict, total=False): @@ -56,7 +213,7 @@ class SubscriptionAttachmentDataAttachmentWidgetDict(TypedDict, total=False): """ :meta private: [EXPERIMENTAL] - [Private Preview] The URL identifier of the page containing the widget. This is the value of the `name` + [Beta] The URL identifier of the page containing the widget. This is the value of the `name` field of the corresponding entry under `pages` in the serialized dashboard definition. Format: alphanumeric, hyphen, and underscore characters allowed, max length 63. """ @@ -65,13 +222,10 @@ class SubscriptionAttachmentDataAttachmentWidgetDict(TypedDict, total=False): """ :meta private: [EXPERIMENTAL] - [Private Preview] The URL identifier of the widget. This is the value of the `name` field of the + [Beta] The URL identifier of the widget. This is the value of the `name` field of the corresponding widget under that page's `layout` in the serialized dashboard definition. Format: alphanumeric, hyphen, and underscore characters allowed, max length 63. """ -SubscriptionAttachmentDataAttachmentWidgetParam = ( - SubscriptionAttachmentDataAttachmentWidgetDict - | SubscriptionAttachmentDataAttachmentWidget -) +SubscriptionAttachmentDataAttachmentWidgetParam = SubscriptionAttachmentDataAttachmentWidgetDict | SubscriptionAttachmentDataAttachmentWidget diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment.py b/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment.py index 1d80e46e067..7860b18867a 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment.py @@ -1,56 +1,209 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import ( - SubscriptionAttachmentPageAttachmentPageSelection, - SubscriptionAttachmentPageAttachmentPageSelectionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SubscriptionAttachmentPageAttachment: """ - :meta private: [EXPERIMENTAL] - The selection of dashboard pages included in the attached PDF. `page_selection` chooses between all pages, the first page only, or an explicit subset listed in `page_ids`. """ - page_ids: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] - [Private Preview] The identifiers of the pages to include in the PDF. Only specified when `page_selection` is + [Beta] The identifiers of the pages to include in the PDF. Only specified when `page_selection` is `PAGE_SELECTION_SUBSET`. Each value is the identifier of a page in the dashboard's serialized definition returned by the Get dashboard API (`serialized_dashboard`): the value of the `name` field of an entry under `pages`. Format: alphanumeric, hyphen, and underscore characters allowed, max length 63. """ - page_selection: VariableOrOptional[ - SubscriptionAttachmentPageAttachmentPageSelection - ] = None + page_selection: VariableOrOptional[SubscriptionAttachmentPageAttachmentPageSelection] = None """ :meta private: [EXPERIMENTAL] - [Private Preview] How the attached pages are selected. When set to `PAGE_SELECTION_SUBSET`, the pages to + [Beta] How the attached pages are selected. When set to `PAGE_SELECTION_SUBSET`, the pages to include are listed in `page_ids`. """ @classmethod - def from_dict(cls, value: "SubscriptionAttachmentPageAttachmentDict") -> "Self": + def from_dict(cls, value: 'SubscriptionAttachmentPageAttachmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SubscriptionAttachmentPageAttachmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SubscriptionAttachmentPageAttachmentDict': + return _transform_to_json_value(self) # type:ignore + class SubscriptionAttachmentPageAttachmentDict(TypedDict, total=False): @@ -60,24 +213,20 @@ class SubscriptionAttachmentPageAttachmentDict(TypedDict, total=False): """ :meta private: [EXPERIMENTAL] - [Private Preview] The identifiers of the pages to include in the PDF. Only specified when `page_selection` is + [Beta] The identifiers of the pages to include in the PDF. Only specified when `page_selection` is `PAGE_SELECTION_SUBSET`. Each value is the identifier of a page in the dashboard's serialized definition returned by the Get dashboard API (`serialized_dashboard`): the value of the `name` field of an entry under `pages`. Format: alphanumeric, hyphen, and underscore characters allowed, max length 63. """ - page_selection: VariableOrOptional[ - SubscriptionAttachmentPageAttachmentPageSelectionParam - ] + page_selection: VariableOrOptional[SubscriptionAttachmentPageAttachmentPageSelectionParam] """ :meta private: [EXPERIMENTAL] - [Private Preview] How the attached pages are selected. When set to `PAGE_SELECTION_SUBSET`, the pages to + [Beta] How the attached pages are selected. When set to `PAGE_SELECTION_SUBSET`, the pages to include are listed in `page_ids`. """ -SubscriptionAttachmentPageAttachmentParam = ( - SubscriptionAttachmentPageAttachmentDict | SubscriptionAttachmentPageAttachment -) +SubscriptionAttachmentPageAttachmentParam = SubscriptionAttachmentPageAttachmentDict | SubscriptionAttachmentPageAttachment diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment_page_selection.py b/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment_page_selection.py index 0d21a769bb4..bbbf72ae20b 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment_page_selection.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment_page_selection.py @@ -1,21 +1,178 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam class SubscriptionAttachmentPageAttachmentPageSelection(Enum): """ - :meta private: [EXPERIMENTAL] - Determines which dashboard pages are attached to the notification. """ - PAGE_SELECTION_ALL = "PAGE_SELECTION_ALL" PAGE_SELECTION_SUBSET = "PAGE_SELECTION_SUBSET" - -SubscriptionAttachmentPageAttachmentPageSelectionParam = ( - Literal["PAGE_SELECTION_ALL", "PAGE_SELECTION_SUBSET"] - | SubscriptionAttachmentPageAttachmentPageSelection -) +SubscriptionAttachmentPageAttachmentPageSelectionParam = Literal["PAGE_SELECTION_ALL", "PAGE_SELECTION_SUBSET"] | SubscriptionAttachmentPageAttachmentPageSelection diff --git a/python/databricks/bundles/jobs/_models/subscription_subscriber.py b/python/databricks/bundles/jobs/_models/subscription_subscriber.py index d03875f1ab2..0e62012edd4 100644 --- a/python/databricks/bundles/jobs/_models/subscription_subscriber.py +++ b/python/databricks/bundles/jobs/_models/subscription_subscriber.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SubscriptionSubscriber: """""" @@ -26,11 +187,12 @@ class SubscriptionSubscriber: """ @classmethod - def from_dict(cls, value: "SubscriptionSubscriberDict") -> "Self": + def from_dict(cls, value: 'SubscriptionSubscriberDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SubscriptionSubscriberDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SubscriptionSubscriberDict': + return _transform_to_json_value(self) # type:ignore + class SubscriptionSubscriberDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/table_update_trigger_configuration.py b/python/databricks/bundles/jobs/_models/table_update_trigger_configuration.py index 24cf1301154..ba99e1b5c9a 100644 --- a/python/databricks/bundles/jobs/_models/table_update_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/table_update_trigger_configuration.py @@ -1,17 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + from databricks.bundles.jobs._models.condition import Condition, ConditionParam +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TableUpdateTriggerConfiguration: """""" @@ -40,11 +200,12 @@ class TableUpdateTriggerConfiguration: """ @classmethod - def from_dict(cls, value: "TableUpdateTriggerConfigurationDict") -> "Self": + def from_dict(cls, value: 'TableUpdateTriggerConfigurationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TableUpdateTriggerConfigurationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TableUpdateTriggerConfigurationDict': + return _transform_to_json_value(self) # type:ignore + class TableUpdateTriggerConfigurationDict(TypedDict, total=False): @@ -74,6 +235,4 @@ class TableUpdateTriggerConfigurationDict(TypedDict, total=False): """ -TableUpdateTriggerConfigurationParam = ( - TableUpdateTriggerConfigurationDict | TableUpdateTriggerConfiguration -) +TableUpdateTriggerConfigurationParam = TableUpdateTriggerConfigurationDict | TableUpdateTriggerConfiguration diff --git a/python/databricks/bundles/jobs/_models/task.py b/python/databricks/bundles/jobs/_models/task.py index c459192ba2b..c1529538a1e 100644 --- a/python/databricks/bundles/jobs/_models/task.py +++ b/python/databricks/bundles/jobs/_models/task.py @@ -1,105 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.jobs._models.ai_runtime_task import ( - AiRuntimeTask, - AiRuntimeTaskParam, -) -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import ( - CleanRoomsNotebookTask, - CleanRoomsNotebookTaskParam, -) -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecParam -from databricks.bundles.jobs._models.compute import Compute, ComputeParam -from databricks.bundles.jobs._models.condition_task import ( - ConditionTask, - ConditionTaskParam, -) -from databricks.bundles.jobs._models.dashboard_task import ( - DashboardTask, - DashboardTaskParam, -) -from databricks.bundles.jobs._models.dbt_platform_task import ( - DbtPlatformTask, - DbtPlatformTaskParam, -) -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import ( - GenAiComputeTask, - GenAiComputeTaskParam, -) -from databricks.bundles.jobs._models.jobs_health_rules import ( - JobsHealthRules, - JobsHealthRulesParam, -) -from databricks.bundles.jobs._models.library import Library, LibraryParam -from databricks.bundles.jobs._models.notebook_task import ( - NotebookTask, - NotebookTaskParam, -) -from databricks.bundles.jobs._models.pipeline_task import ( - PipelineTask, - PipelineTaskParam, -) -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import ( - PythonOperatorTask, - PythonOperatorTaskParam, -) -from databricks.bundles.jobs._models.python_wheel_task import ( - PythonWheelTask, - PythonWheelTaskParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam -from databricks.bundles.jobs._models.run_job_task import ( - RunJobTask, - RunJobTaskParam, -) -from databricks.bundles.jobs._models.spark_jar_task import ( - SparkJarTask, - SparkJarTaskParam, -) -from databricks.bundles.jobs._models.spark_python_task import ( - SparkPythonTask, - SparkPythonTaskParam, -) -from databricks.bundles.jobs._models.spark_submit_task import ( - SparkSubmitTask, - SparkSubmitTaskParam, -) -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskParam -from databricks.bundles.jobs._models.task_dependency import ( - TaskDependency, - TaskDependencyParam, -) -from databricks.bundles.jobs._models.task_email_notifications import ( - TaskEmailNotifications, - TaskEmailNotificationsParam, -) -from databricks.bundles.jobs._models.task_notification_settings import ( - TaskNotificationSettings, - TaskNotificationSettingsParam, -) -from databricks.bundles.jobs._models.webhook_notifications import ( - WebhookNotifications, - WebhookNotificationsParam, -) + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Task: """""" @@ -341,11 +413,12 @@ class Task: """ @classmethod - def from_dict(cls, value: "TaskDict") -> "Self": + def from_dict(cls, value: 'TaskDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TaskDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TaskDict': + return _transform_to_json_value(self) # type:ignore + class TaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/task_dependency.py b/python/databricks/bundles/jobs/_models/task_dependency.py index a1e18f0bb59..dcd597778e1 100644 --- a/python/databricks/bundles/jobs/_models/task_dependency.py +++ b/python/databricks/bundles/jobs/_models/task_dependency.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TaskDependency: """""" @@ -26,11 +187,12 @@ class TaskDependency: """ @classmethod - def from_dict(cls, value: "TaskDependencyDict") -> "Self": + def from_dict(cls, value: 'TaskDependencyDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TaskDependencyDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TaskDependencyDict': + return _transform_to_json_value(self) # type:ignore + class TaskDependencyDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/task_email_notifications.py b/python/databricks/bundles/jobs/_models/task_email_notifications.py index cdf0dee5fdf..4ab3ccc97d0 100644 --- a/python/databricks/bundles/jobs/_models/task_email_notifications.py +++ b/python/databricks/bundles/jobs/_models/task_email_notifications.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TaskEmailNotifications: """""" @@ -21,9 +182,7 @@ class TaskEmailNotifications: This field is `deprecated`. Please use the `notification_settings.no_alert_for_skipped_runs` field. """ - on_duration_warning_threshold_exceeded: VariableOrList[str] = field( - default_factory=list - ) + on_duration_warning_threshold_exceeded: VariableOrList[str] = field(default_factory=list) """ A list of email addresses to be notified when the duration of a run exceeds the threshold specified for the `RUN_DURATION_SECONDS` metric in the `health` field. If no rule for the `RUN_DURATION_SECONDS` metric is specified in the `health` field for the job, notifications are not sent. """ @@ -61,11 +220,12 @@ class TaskEmailNotifications: """ @classmethod - def from_dict(cls, value: "TaskEmailNotificationsDict") -> "Self": + def from_dict(cls, value: 'TaskEmailNotificationsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TaskEmailNotificationsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TaskEmailNotificationsDict': + return _transform_to_json_value(self) # type:ignore + class TaskEmailNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/task_notification_settings.py b/python/databricks/bundles/jobs/_models/task_notification_settings.py index 346f4463b9a..274dd11acff 100644 --- a/python/databricks/bundles/jobs/_models/task_notification_settings.py +++ b/python/databricks/bundles/jobs/_models/task_notification_settings.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TaskNotificationSettings: """""" @@ -31,11 +192,12 @@ class TaskNotificationSettings: """ @classmethod - def from_dict(cls, value: "TaskNotificationSettingsDict") -> "Self": + def from_dict(cls, value: 'TaskNotificationSettingsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TaskNotificationSettingsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TaskNotificationSettingsDict': + return _transform_to_json_value(self) # type:ignore + class TaskNotificationSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/task_retry_mode.py b/python/databricks/bundles/jobs/_models/task_retry_mode.py index f0642b54f2c..20a8d9fdd81 100644 --- a/python/databricks/bundles/jobs/_models/task_retry_mode.py +++ b/python/databricks/bundles/jobs/_models/task_retry_mode.py @@ -1,7 +1,171 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam class TaskRetryMode(Enum): @@ -11,9 +175,7 @@ class TaskRetryMode(Enum): * ON_FAILURE: Retry a failed task if at least one other task in the job is still running its first attempt. When this condition is no longer met or the retry limit is reached, the job run is cancelled and a new run is started. """ - NEVER = "NEVER" ON_FAILURE = "ON_FAILURE" - TaskRetryModeParam = Literal["NEVER", "ON_FAILURE"] | TaskRetryMode diff --git a/python/databricks/bundles/jobs/_models/trigger_configuration.py b/python/databricks/bundles/jobs/_models/trigger_configuration.py index b9663effddc..bbf33d9cdf3 100644 --- a/python/databricks/bundles/jobs/_models/trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/trigger_configuration.py @@ -1,45 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.continuous_trigger_configuration import ( - ContinuousTriggerConfiguration, - ContinuousTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.cron_trigger_configuration import ( - CronTriggerConfiguration, - CronTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import ( - FileArrivalTriggerConfiguration, - FileArrivalTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.model_trigger_configuration import ( - ModelTriggerConfiguration, - ModelTriggerConfigurationParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import ( - PeriodicTriggerConfiguration, - PeriodicTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.sql_condition_configuration import ( - SqlConditionConfiguration, - SqlConditionConfigurationParam, -) -from databricks.bundles.jobs._models.table_update_trigger_configuration import ( - TableUpdateTriggerConfiguration, - TableUpdateTriggerConfigurationParam, -) + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TriggerConfiguration: """ @@ -48,7 +180,6 @@ class TriggerConfiguration: mutual exclusivity is enforced in the API handler rather than via `oneof` so that codegen, validation, and JSON serialization across SDKs and Terraform behave consistently. """ - continuous: VariableOrOptional[ContinuousTriggerConfiguration] = None """ :meta private: [EXPERIMENTAL] @@ -107,11 +238,12 @@ class TriggerConfiguration: """ @classmethod - def from_dict(cls, value: "TriggerConfigurationDict") -> "Self": + def from_dict(cls, value: 'TriggerConfigurationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TriggerConfigurationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TriggerConfigurationDict': + return _transform_to_json_value(self) # type:ignore + class TriggerConfigurationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/trigger_settings.py b/python/databricks/bundles/jobs/_models/trigger_settings.py index 878f45732fa..c2c968fca6c 100644 --- a/python/databricks/bundles/jobs/_models/trigger_settings.py +++ b/python/databricks/bundles/jobs/_models/trigger_settings.py @@ -1,37 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import ( - FileArrivalTriggerConfiguration, - FileArrivalTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.model_trigger_configuration import ( - ModelTriggerConfiguration, - ModelTriggerConfigurationParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import ( - PeriodicTriggerConfiguration, - PeriodicTriggerConfigurationParam, -) -from databricks.bundles.jobs._models.sql_condition_configuration import ( - SqlConditionConfiguration, - SqlConditionConfigurationParam, -) -from databricks.bundles.jobs._models.table_update_trigger_configuration import ( - TableUpdateTriggerConfiguration, - TableUpdateTriggerConfigurationParam, -) + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TriggerSettings: """""" @@ -69,11 +209,12 @@ class TriggerSettings: table_update: VariableOrOptional[TableUpdateTriggerConfiguration] = None @classmethod - def from_dict(cls, value: "TriggerSettingsDict") -> "Self": + def from_dict(cls, value: 'TriggerSettingsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TriggerSettingsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TriggerSettingsDict': + return _transform_to_json_value(self) # type:ignore + class TriggerSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/volumes_storage_info.py b/python/databricks/bundles/jobs/_models/volumes_storage_info.py index 1eb4fff7bc2..da6897b5c29 100644 --- a/python/databricks/bundles/jobs/_models/volumes_storage_info.py +++ b/python/databricks/bundles/jobs/_models/volumes_storage_info.py @@ -1,22 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class VolumesStorageInfo: """ A storage location back by UC Volumes. """ - destination: VariableOr[str] """ UC Volumes destination, e.g. `/Volumes/catalog/schema/vol1/init-scripts/setup-datadog.sh` @@ -24,11 +184,12 @@ class VolumesStorageInfo: """ @classmethod - def from_dict(cls, value: "VolumesStorageInfoDict") -> "Self": + def from_dict(cls, value: 'VolumesStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "VolumesStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'VolumesStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class VolumesStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/webhook.py b/python/databricks/bundles/jobs/_models/webhook.py index 17a391ac2c2..5fdaf88e2b4 100644 --- a/python/databricks/bundles/jobs/_models/webhook.py +++ b/python/databricks/bundles/jobs/_models/webhook.py @@ -1,16 +1,177 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Webhook: """""" @@ -18,11 +179,12 @@ class Webhook: id: VariableOr[str] @classmethod - def from_dict(cls, value: "WebhookDict") -> "Self": + def from_dict(cls, value: 'WebhookDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "WebhookDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'WebhookDict': + return _transform_to_json_value(self) # type:ignore + class WebhookDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/webhook_notifications.py b/python/databricks/bundles/jobs/_models/webhook_notifications.py index 0eed809f68b..edf13270be2 100644 --- a/python/databricks/bundles/jobs/_models/webhook_notifications.py +++ b/python/databricks/bundles/jobs/_models/webhook_notifications.py @@ -1,24 +1,182 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList -from databricks.bundles.jobs._models.webhook import Webhook, WebhookParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class WebhookNotifications: """""" - on_duration_warning_threshold_exceeded: VariableOrList[Webhook] = field( - default_factory=list - ) + on_duration_warning_threshold_exceeded: VariableOrList[Webhook] = field(default_factory=list) """ An optional list of system notification IDs to call when the duration of a run exceeds the threshold specified for the `RUN_DURATION_SECONDS` metric in the `health` field. A maximum of 3 destinations can be specified for the `on_duration_warning_threshold_exceeded` property. """ @@ -59,11 +217,12 @@ class WebhookNotifications: """ @classmethod - def from_dict(cls, value: "WebhookNotificationsDict") -> "Self": + def from_dict(cls, value: 'WebhookNotificationsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "WebhookNotificationsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'WebhookNotificationsDict': + return _transform_to_json_value(self) # type:ignore + class WebhookNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/workload_type.py b/python/databricks/bundles/jobs/_models/workload_type.py index 033590c3949..9d21b749537 100644 --- a/python/databricks/bundles/jobs/_models/workload_type.py +++ b/python/databricks/bundles/jobs/_models/workload_type.py @@ -1,37 +1,194 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr -from databricks.bundles.jobs._models.clients_types import ( - ClientsTypes, - ClientsTypesParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class WorkloadType: """ Cluster Attributes showing for clusters workload types. """ - clients: VariableOr[ClientsTypes] """ defined what type of clients can use the cluster. E.g. Notebooks, Jobs """ @classmethod - def from_dict(cls, value: "WorkloadTypeDict") -> "Self": + def from_dict(cls, value: 'WorkloadTypeDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "WorkloadTypeDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'WorkloadTypeDict': + return _transform_to_json_value(self) # type:ignore + class WorkloadTypeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/workspace_storage_info.py b/python/databricks/bundles/jobs/_models/workspace_storage_info.py index 49aef9256c8..37f44aadfbe 100644 --- a/python/databricks/bundles/jobs/_models/workspace_storage_info.py +++ b/python/databricks/bundles/jobs/_models/workspace_storage_info.py @@ -1,33 +1,194 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam +from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam +from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam +from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam +from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam +from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam +from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam +from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam +from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam +from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam +from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam +from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam +from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam +from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam +from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam +from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam +from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam +from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam +from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam +from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam +from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam +from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam +from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam +from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam +from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam +from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam +from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam +from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam +from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam +from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam +from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam +from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam +from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam +from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam +from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam +from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam +from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam +from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam +from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam +from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam +from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam +from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam +from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam +from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam +from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam +from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam +from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam +from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam +from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam +from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam +from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam + +from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam + +from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam + +from databricks.bundles.jobs._models.kind import Kind, KindParam + +from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam + +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam + +from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam + +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam + +from databricks.bundles.jobs._models.condition import Condition, ConditionParam + +from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam + +from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam + +from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam + +from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam + +from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam + +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam + +from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam + +from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam + +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam + +from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam + +from databricks.bundles.jobs._models.source import Source, SourceParam + +from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam + +from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam + +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam + +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam + +from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class WorkspaceStorageInfo: """ A storage location in Workspace Filesystem (WSFS) """ - destination: VariableOr[str] """ wsfs destination, e.g. `workspace:/cluster-init-scripts/setup-datadog.sh` """ @classmethod - def from_dict(cls, value: "WorkspaceStorageInfoDict") -> "Self": + def from_dict(cls, value: 'WorkspaceStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "WorkspaceStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'WorkspaceStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class WorkspaceStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/__init__.py b/python/databricks/bundles/mcp_services/__init__.py index a4fe01b5199..00f577ad846 100644 --- a/python/databricks/bundles/mcp_services/__init__.py +++ b/python/databricks/bundles/mcp_services/__init__.py @@ -1,69 +1,42 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "McpService", - "McpServiceConfig", - "McpServiceConfigDict", - "McpServiceConfigParam", - "McpServiceConfigSourceConnection", - "McpServiceConfigSourceConnectionDict", - "McpServiceConfigSourceConnectionParam", - "McpServiceDict", - "McpServiceParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", - "RateLimit", - "RateLimitDict", - "RateLimitParam", - "RateLimitRateLimitKey", - "RateLimitRateLimitKeyParam", - "RateLimitRateLimitRenewalPeriod", - "RateLimitRateLimitRenewalPeriodParam", + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'McpService', + 'McpServiceConfig', + 'McpServiceConfigDict', + 'McpServiceConfigParam', + 'McpServiceConfigSourceConnection', + 'McpServiceConfigSourceConnectionDict', + 'McpServiceConfigSourceConnectionParam', + 'McpServiceDict', + 'McpServiceParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'RateLimit', + 'RateLimitDict', + 'RateLimitParam', + 'RateLimitRateLimitKey', + 'RateLimitRateLimitKeyParam', + 'RateLimitRateLimitRenewalPeriod', + 'RateLimitRateLimitRenewalPeriodParam', ] -from databricks.bundles.mcp_services._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.mcp_services._models.mcp_service import ( - McpService, - McpServiceDict, - McpServiceParam, -) -from databricks.bundles.mcp_services._models.mcp_service_config import ( - McpServiceConfig, - McpServiceConfigDict, - McpServiceConfigParam, -) -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import ( - McpServiceConfigSourceConnection, - McpServiceConfigSourceConnectionDict, - McpServiceConfigSourceConnectionParam, -) +from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam -from databricks.bundles.mcp_services._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) -from databricks.bundles.mcp_services._models.rate_limit import ( - RateLimit, - RateLimitDict, - RateLimitParam, -) -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import ( - RateLimitRateLimitKey, - RateLimitRateLimitKeyParam, -) -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import ( - RateLimitRateLimitRenewalPeriod, - RateLimitRateLimitRenewalPeriodParam, -) + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + diff --git a/python/databricks/bundles/mcp_services/_models/lifecycle.py b/python/databricks/bundles/mcp_services/_models/lifecycle.py index 697776a198e..02ab7b03257 100644 --- a/python/databricks/bundles/mcp_services/_models/lifecycle.py +++ b/python/databricks/bundles/mcp_services/_models/lifecycle.py @@ -1,16 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam +from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +34,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/mcp_service.py b/python/databricks/bundles/mcp_services/_models/mcp_service.py index 1050e5153f4..63dd67b86de 100644 --- a/python/databricks/bundles/mcp_services/_models/mcp_service.py +++ b/python/databricks/bundles/mcp_services/_models/mcp_service.py @@ -1,33 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.mcp_services._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.mcp_services._models.mcp_service_config import ( - McpServiceConfig, - McpServiceConfigParam, -) -from databricks.bundles.mcp_services._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class McpService(Resource): """""" @@ -45,11 +41,12 @@ class McpService(Resource): lifecycle: VariableOrOptional[Lifecycle] = None @classmethod - def from_dict(cls, value: "McpServiceDict") -> "Self": + def from_dict(cls, value: 'McpServiceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "McpServiceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'McpServiceDict': + return _transform_to_json_value(self) # type:ignore + class McpServiceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/mcp_service_config.py b/python/databricks/bundles/mcp_services/_models/mcp_service_config.py index 5a9a984bf01..36b70fd4bce 100644 --- a/python/databricks/bundles/mcp_services/_models/mcp_service_config.py +++ b/python/databricks/bundles/mcp_services/_models/mcp_service_config.py @@ -1,24 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import ( - McpServiceConfigSourceConnection, - McpServiceConfigSourceConnectionParam, -) -from databricks.bundles.mcp_services._models.rate_limit import ( - RateLimit, - RateLimitParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam +from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class McpServiceConfig: """ @@ -26,7 +31,6 @@ class McpServiceConfig: tool selectors, and rate limits -- the fields that configure how the MCP service behaves at invocation time. """ - include_tool_selectors: VariableOrList[str] = field(default_factory=list) """ Tool names or prefix patterns to expose from the MCP server. Use exact @@ -48,11 +52,12 @@ class McpServiceConfig: """ @classmethod - def from_dict(cls, value: "McpServiceConfigDict") -> "Self": + def from_dict(cls, value: 'McpServiceConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "McpServiceConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'McpServiceConfigDict': + return _transform_to_json_value(self) # type:ignore + class McpServiceConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/mcp_service_config_source_connection.py b/python/databricks/bundles/mcp_services/_models/mcp_service_config_source_connection.py index 4da924fcd38..837f6f2a2f9 100644 --- a/python/databricks/bundles/mcp_services/_models/mcp_service_config_source_connection.py +++ b/python/databricks/bundles/mcp_services/_models/mcp_service_config_source_connection.py @@ -1,16 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam +from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class McpServiceConfigSourceConnection: """ @@ -20,7 +33,6 @@ class McpServiceConfigSourceConnection: populates the resolved connection metadata. If the connection is deleted, its reference remains visible so you can identify the broken dependency. """ - name: VariableOr[str] """ Resource name of the Unity Catalog connection used to access the MCP @@ -28,11 +40,12 @@ class McpServiceConfigSourceConnection: """ @classmethod - def from_dict(cls, value: "McpServiceConfigSourceConnectionDict") -> "Self": + def from_dict(cls, value: 'McpServiceConfigSourceConnectionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "McpServiceConfigSourceConnectionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'McpServiceConfigSourceConnectionDict': + return _transform_to_json_value(self) # type:ignore + class McpServiceConfigSourceConnectionDict(TypedDict, total=False): @@ -45,6 +58,4 @@ class McpServiceConfigSourceConnectionDict(TypedDict, total=False): """ -McpServiceConfigSourceConnectionParam = ( - McpServiceConfigSourceConnectionDict | McpServiceConfigSourceConnection -) +McpServiceConfigSourceConnectionParam = McpServiceConfigSourceConnectionDict | McpServiceConfigSourceConnection diff --git a/python/databricks/bundles/mcp_services/_models/privilege.py b/python/databricks/bundles/mcp_services/_models/privilege.py index df45211a76a..f8ad08d6ec1 100644 --- a/python/databricks/bundles/mcp_services/_models/privilege.py +++ b/python/databricks/bundles/mcp_services/_models/privilege.py @@ -1,7 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam class Privilege(Enum): @@ -58,61 +74,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/mcp_services/_models/privilege_assignment.py b/python/databricks/bundles/mcp_services/_models/privilege_assignment.py index cdaba80f76a..c95110a7933 100644 --- a/python/databricks/bundles/mcp_services/_models/privilege_assignment.py +++ b/python/databricks/bundles/mcp_services/_models/privilege_assignment.py @@ -1,17 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam +from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -28,11 +40,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/rate_limit.py b/python/databricks/bundles/mcp_services/_models/rate_limit.py index cba7fed7056..5de5d60246a 100644 --- a/python/databricks/bundles/mcp_services/_models/rate_limit.py +++ b/python/databricks/bundles/mcp_services/_models/rate_limit.py @@ -1,24 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import ( - RateLimitRateLimitKey, - RateLimitRateLimitKeyParam, -) -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import ( - RateLimitRateLimitRenewalPeriod, - RateLimitRateLimitRenewalPeriodParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam +from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RateLimit: """ @@ -26,7 +31,6 @@ class RateLimit: unset to impose no limit on that dimension; set a value to cap that dimension within the renewal period. """ - key: VariableOr[RateLimitRateLimitKey] """ Scope of the rate limit. Depending on this value, the limit applies to a @@ -58,11 +62,12 @@ class RateLimit: """ @classmethod - def from_dict(cls, value: "RateLimitDict") -> "Self": + def from_dict(cls, value: 'RateLimitDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RateLimitDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RateLimitDict': + return _transform_to_json_value(self) # type:ignore + class RateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_key.py b/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_key.py index fb907da810f..8f5958333b3 100644 --- a/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_key.py +++ b/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_key.py @@ -1,28 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam +from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam class RateLimitRateLimitKey(Enum): """ Scope key for a rate limit. """ - RATE_LIMIT_KEY_USER = "RATE_LIMIT_KEY_USER" RATE_LIMIT_KEY_USER_GROUP = "RATE_LIMIT_KEY_USER_GROUP" RATE_LIMIT_KEY_SERVICE_PRINCIPAL = "RATE_LIMIT_KEY_SERVICE_PRINCIPAL" RATE_LIMIT_KEY_SERVICE = "RATE_LIMIT_KEY_SERVICE" RATE_LIMIT_KEY_USER_DEFAULT = "RATE_LIMIT_KEY_USER_DEFAULT" - -RateLimitRateLimitKeyParam = ( - Literal[ - "RATE_LIMIT_KEY_USER", - "RATE_LIMIT_KEY_USER_GROUP", - "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", - "RATE_LIMIT_KEY_SERVICE", - "RATE_LIMIT_KEY_USER_DEFAULT", - ] - | RateLimitRateLimitKey -) +RateLimitRateLimitKeyParam = Literal["RATE_LIMIT_KEY_USER", "RATE_LIMIT_KEY_USER_GROUP", "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", "RATE_LIMIT_KEY_SERVICE", "RATE_LIMIT_KEY_USER_DEFAULT"] | RateLimitRateLimitKey diff --git a/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_renewal_period.py b/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_renewal_period.py index 58636c784c0..c37c8d583a4 100644 --- a/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_renewal_period.py +++ b/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_renewal_period.py @@ -1,19 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam +from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam class RateLimitRateLimitRenewalPeriod(Enum): """ Renewal period for a rate limit. """ - RATE_LIMIT_RENEWAL_PERIOD_MINUTE = "RATE_LIMIT_RENEWAL_PERIOD_MINUTE" RATE_LIMIT_RENEWAL_PERIOD_HOUR = "RATE_LIMIT_RENEWAL_PERIOD_HOUR" - -RateLimitRateLimitRenewalPeriodParam = ( - Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] - | RateLimitRateLimitRenewalPeriod -) +RateLimitRateLimitRenewalPeriodParam = Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] | RateLimitRateLimitRenewalPeriod diff --git a/python/databricks/bundles/model_provider_services/__init__.py b/python/databricks/bundles/model_provider_services/__init__.py index e7b2a5b13f7..14f8af90796 100644 --- a/python/databricks/bundles/model_provider_services/__init__.py +++ b/python/databricks/bundles/model_provider_services/__init__.py @@ -1,262 +1,138 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "InferenceTableConfig", - "InferenceTableConfigDict", - "InferenceTableConfigParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "ModelProviderService", - "ModelProviderServiceConfig", - "ModelProviderServiceConfigAmazonBedrockProviderConfig", - "ModelProviderServiceConfigAmazonBedrockProviderConfigDict", - "ModelProviderServiceConfigAmazonBedrockProviderConfigParam", - "ModelProviderServiceConfigAmazonBedrockProviderDirectConfig", - "ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict", - "ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam", - "ModelProviderServiceConfigAnthropicProviderConfig", - "ModelProviderServiceConfigAnthropicProviderConfigDict", - "ModelProviderServiceConfigAnthropicProviderConfigParam", - "ModelProviderServiceConfigAnthropicProviderDirectConfig", - "ModelProviderServiceConfigAnthropicProviderDirectConfigDict", - "ModelProviderServiceConfigAnthropicProviderDirectConfigParam", - "ModelProviderServiceConfigAnthropicProviderRelayedConfig", - "ModelProviderServiceConfigAnthropicProviderRelayedConfigDict", - "ModelProviderServiceConfigAnthropicProviderRelayedConfigParam", - "ModelProviderServiceConfigAwsAccessKey", - "ModelProviderServiceConfigAwsAccessKeyDict", - "ModelProviderServiceConfigAwsAccessKeyParam", - "ModelProviderServiceConfigAzureOpenAiProviderConfig", - "ModelProviderServiceConfigAzureOpenAiProviderConfigDict", - "ModelProviderServiceConfigAzureOpenAiProviderConfigParam", - "ModelProviderServiceConfigAzureOpenAiProviderDirectConfig", - "ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict", - "ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam", - "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth", - "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict", - "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam", - "ModelProviderServiceConfigCustomProviderConfig", - "ModelProviderServiceConfigCustomProviderConfigDict", - "ModelProviderServiceConfigCustomProviderConfigParam", - "ModelProviderServiceConfigCustomProviderDirectConfig", - "ModelProviderServiceConfigCustomProviderDirectConfigDict", - "ModelProviderServiceConfigCustomProviderDirectConfigParam", - "ModelProviderServiceConfigDict", - "ModelProviderServiceConfigEntraServicePrincipal", - "ModelProviderServiceConfigEntraServicePrincipalDict", - "ModelProviderServiceConfigEntraServicePrincipalParam", - "ModelProviderServiceConfigExternalModelProviderType", - "ModelProviderServiceConfigExternalModelProviderTypeParam", - "ModelProviderServiceConfigGeminiEnterpriseProviderConfig", - "ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict", - "ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam", - "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig", - "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict", - "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam", - "ModelProviderServiceConfigMicrosoftFoundryProviderConfig", - "ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict", - "ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam", - "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig", - "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict", - "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam", - "ModelProviderServiceConfigModelTargetConfig", - "ModelProviderServiceConfigModelTargetConfigDict", - "ModelProviderServiceConfigModelTargetConfigParam", - "ModelProviderServiceConfigOpenAiProviderConfig", - "ModelProviderServiceConfigOpenAiProviderConfigDict", - "ModelProviderServiceConfigOpenAiProviderConfigParam", - "ModelProviderServiceConfigOpenAiProviderDirectConfig", - "ModelProviderServiceConfigOpenAiProviderDirectConfigDict", - "ModelProviderServiceConfigOpenAiProviderDirectConfigParam", - "ModelProviderServiceConfigParam", - "ModelProviderServiceConfigProviderPricingConfig", - "ModelProviderServiceConfigProviderPricingConfigDict", - "ModelProviderServiceConfigProviderPricingConfigParam", - "ModelProviderServiceConfigProviderSecret", - "ModelProviderServiceConfigProviderSecretDict", - "ModelProviderServiceConfigProviderSecretParam", - "ModelProviderServiceConfigSecretReference", - "ModelProviderServiceConfigSecretReferenceDict", - "ModelProviderServiceConfigSecretReferenceParam", - "ModelProviderServiceConfigServiceCredential", - "ModelProviderServiceConfigServiceCredentialDict", - "ModelProviderServiceConfigServiceCredentialParam", - "ModelProviderServiceDict", - "ModelProviderServiceParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", - "RateLimit", - "RateLimitDict", - "RateLimitParam", - "RateLimitRateLimitKey", - "RateLimitRateLimitKeyParam", - "RateLimitRateLimitRenewalPeriod", - "RateLimitRateLimitRenewalPeriodParam", + 'InferenceTableConfig', + 'InferenceTableConfigDict', + 'InferenceTableConfigParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'ModelProviderService', + 'ModelProviderServiceConfig', + 'ModelProviderServiceConfigAmazonBedrockProviderConfig', + 'ModelProviderServiceConfigAmazonBedrockProviderConfigDict', + 'ModelProviderServiceConfigAmazonBedrockProviderConfigParam', + 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfig', + 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict', + 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam', + 'ModelProviderServiceConfigAnthropicProviderConfig', + 'ModelProviderServiceConfigAnthropicProviderConfigDict', + 'ModelProviderServiceConfigAnthropicProviderConfigParam', + 'ModelProviderServiceConfigAnthropicProviderDirectConfig', + 'ModelProviderServiceConfigAnthropicProviderDirectConfigDict', + 'ModelProviderServiceConfigAnthropicProviderDirectConfigParam', + 'ModelProviderServiceConfigAnthropicProviderRelayedConfig', + 'ModelProviderServiceConfigAnthropicProviderRelayedConfigDict', + 'ModelProviderServiceConfigAnthropicProviderRelayedConfigParam', + 'ModelProviderServiceConfigAwsAccessKey', + 'ModelProviderServiceConfigAwsAccessKeyDict', + 'ModelProviderServiceConfigAwsAccessKeyParam', + 'ModelProviderServiceConfigAzureOpenAiProviderConfig', + 'ModelProviderServiceConfigAzureOpenAiProviderConfigDict', + 'ModelProviderServiceConfigAzureOpenAiProviderConfigParam', + 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfig', + 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict', + 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam', + 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth', + 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict', + 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam', + 'ModelProviderServiceConfigCustomProviderConfig', + 'ModelProviderServiceConfigCustomProviderConfigDict', + 'ModelProviderServiceConfigCustomProviderConfigParam', + 'ModelProviderServiceConfigCustomProviderDirectConfig', + 'ModelProviderServiceConfigCustomProviderDirectConfigDict', + 'ModelProviderServiceConfigCustomProviderDirectConfigParam', + 'ModelProviderServiceConfigDict', + 'ModelProviderServiceConfigEntraServicePrincipal', + 'ModelProviderServiceConfigEntraServicePrincipalDict', + 'ModelProviderServiceConfigEntraServicePrincipalParam', + 'ModelProviderServiceConfigExternalModelProviderType', + 'ModelProviderServiceConfigExternalModelProviderTypeParam', + 'ModelProviderServiceConfigGeminiEnterpriseProviderConfig', + 'ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict', + 'ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam', + 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig', + 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict', + 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam', + 'ModelProviderServiceConfigMicrosoftFoundryProviderConfig', + 'ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict', + 'ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam', + 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig', + 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict', + 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam', + 'ModelProviderServiceConfigModelTargetConfig', + 'ModelProviderServiceConfigModelTargetConfigDict', + 'ModelProviderServiceConfigModelTargetConfigParam', + 'ModelProviderServiceConfigOpenAiProviderConfig', + 'ModelProviderServiceConfigOpenAiProviderConfigDict', + 'ModelProviderServiceConfigOpenAiProviderConfigParam', + 'ModelProviderServiceConfigOpenAiProviderDirectConfig', + 'ModelProviderServiceConfigOpenAiProviderDirectConfigDict', + 'ModelProviderServiceConfigOpenAiProviderDirectConfigParam', + 'ModelProviderServiceConfigParam', + 'ModelProviderServiceConfigProviderPricingConfig', + 'ModelProviderServiceConfigProviderPricingConfigDict', + 'ModelProviderServiceConfigProviderPricingConfigParam', + 'ModelProviderServiceConfigProviderSecret', + 'ModelProviderServiceConfigProviderSecretDict', + 'ModelProviderServiceConfigProviderSecretParam', + 'ModelProviderServiceConfigSecretReference', + 'ModelProviderServiceConfigSecretReferenceDict', + 'ModelProviderServiceConfigSecretReferenceParam', + 'ModelProviderServiceConfigServiceCredential', + 'ModelProviderServiceConfigServiceCredentialDict', + 'ModelProviderServiceConfigServiceCredentialParam', + 'ModelProviderServiceDict', + 'ModelProviderServiceParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'RateLimit', + 'RateLimitDict', + 'RateLimitParam', + 'RateLimitRateLimitKey', + 'RateLimitRateLimitKeyParam', + 'RateLimitRateLimitRenewalPeriod', + 'RateLimitRateLimitRenewalPeriodParam', ] -from databricks.bundles.model_provider_services._models.inference_table_config import ( - InferenceTableConfig, - InferenceTableConfigDict, - InferenceTableConfigParam, -) -from databricks.bundles.model_provider_services._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service import ( - ModelProviderService, - ModelProviderServiceDict, - ModelProviderServiceParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config import ( - ModelProviderServiceConfig, - ModelProviderServiceConfigDict, - ModelProviderServiceConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ( - ModelProviderServiceConfigAmazonBedrockProviderConfig, - ModelProviderServiceConfigAmazonBedrockProviderConfigDict, - ModelProviderServiceConfigAmazonBedrockProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ( - ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, - ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, - ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ( - ModelProviderServiceConfigAnthropicProviderConfig, - ModelProviderServiceConfigAnthropicProviderConfigDict, - ModelProviderServiceConfigAnthropicProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ( - ModelProviderServiceConfigAnthropicProviderDirectConfig, - ModelProviderServiceConfigAnthropicProviderDirectConfigDict, - ModelProviderServiceConfigAnthropicProviderDirectConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ( - ModelProviderServiceConfigAnthropicProviderRelayedConfig, - ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, - ModelProviderServiceConfigAnthropicProviderRelayedConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ( - ModelProviderServiceConfigAwsAccessKey, - ModelProviderServiceConfigAwsAccessKeyDict, - ModelProviderServiceConfigAwsAccessKeyParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ( - ModelProviderServiceConfigAzureOpenAiProviderConfig, - ModelProviderServiceConfigAzureOpenAiProviderConfigDict, - ModelProviderServiceConfigAzureOpenAiProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ( - ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, - ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, - ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ( - ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, - ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, - ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ( - ModelProviderServiceConfigCustomProviderConfig, - ModelProviderServiceConfigCustomProviderConfigDict, - ModelProviderServiceConfigCustomProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ( - ModelProviderServiceConfigCustomProviderDirectConfig, - ModelProviderServiceConfigCustomProviderDirectConfigDict, - ModelProviderServiceConfigCustomProviderDirectConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ( - ModelProviderServiceConfigEntraServicePrincipal, - ModelProviderServiceConfigEntraServicePrincipalDict, - ModelProviderServiceConfigEntraServicePrincipalParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ( - ModelProviderServiceConfigExternalModelProviderType, - ModelProviderServiceConfigExternalModelProviderTypeParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ( - ModelProviderServiceConfigGeminiEnterpriseProviderConfig, - ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, - ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ( - ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, - ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, - ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ( - ModelProviderServiceConfigMicrosoftFoundryProviderConfig, - ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, - ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ( - ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, - ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, - ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ( - ModelProviderServiceConfigModelTargetConfig, - ModelProviderServiceConfigModelTargetConfigDict, - ModelProviderServiceConfigModelTargetConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ( - ModelProviderServiceConfigOpenAiProviderConfig, - ModelProviderServiceConfigOpenAiProviderConfigDict, - ModelProviderServiceConfigOpenAiProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ( - ModelProviderServiceConfigOpenAiProviderDirectConfig, - ModelProviderServiceConfigOpenAiProviderDirectConfigDict, - ModelProviderServiceConfigOpenAiProviderDirectConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ( - ModelProviderServiceConfigProviderPricingConfig, - ModelProviderServiceConfigProviderPricingConfigDict, - ModelProviderServiceConfigProviderPricingConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretDict, - ModelProviderServiceConfigProviderSecretParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ( - ModelProviderServiceConfigSecretReference, - ModelProviderServiceConfigSecretReferenceDict, - ModelProviderServiceConfigSecretReferenceParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( - ModelProviderServiceConfigServiceCredential, - ModelProviderServiceConfigServiceCredentialDict, - ModelProviderServiceConfigServiceCredentialParam, -) -from databricks.bundles.model_provider_services._models.privilege import ( - Privilege, - PrivilegeParam, -) -from databricks.bundles.model_provider_services._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) -from databricks.bundles.model_provider_services._models.rate_limit import ( - RateLimit, - RateLimitDict, - RateLimitParam, -) -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import ( - RateLimitRateLimitKey, - RateLimitRateLimitKeyParam, -) -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import ( - RateLimitRateLimitRenewalPeriod, - RateLimitRateLimitRenewalPeriodParam, -) +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + diff --git a/python/databricks/bundles/model_provider_services/_models/inference_table_config.py b/python/databricks/bundles/model_provider_services/_models/inference_table_config.py index 45165416e17..c246dd20353 100644 --- a/python/databricks/bundles/model_provider_services/_models/inference_table_config.py +++ b/python/databricks/bundles/model_provider_services/_models/inference_table_config.py @@ -1,16 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InferenceTableConfig: """ @@ -18,7 +56,6 @@ class InferenceTableConfig: inference table. When this configuration is present, payload logging is enabled by default. """ - parent: VariableOr[str] """ Parent Unity Catalog schema where the inference table is created, in the @@ -36,11 +73,12 @@ class InferenceTableConfig: """ @classmethod - def from_dict(cls, value: "InferenceTableConfigDict") -> "Self": + def from_dict(cls, value: 'InferenceTableConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InferenceTableConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InferenceTableConfigDict': + return _transform_to_json_value(self) # type:ignore + class InferenceTableConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/lifecycle.py b/python/databricks/bundles/model_provider_services/_models/lifecycle.py index 697776a198e..fd87749f79e 100644 --- a/python/databricks/bundles/model_provider_services/_models/lifecycle.py +++ b/python/databricks/bundles/model_provider_services/_models/lifecycle.py @@ -1,16 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +59,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service.py index 41a21652ac8..520c98c8609 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service.py @@ -1,33 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.model_provider_services._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config import ( - ModelProviderServiceConfig, - ModelProviderServiceConfigParam, -) -from databricks.bundles.model_provider_services._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderService(Resource): """""" @@ -45,11 +66,12 @@ class ModelProviderService(Resource): lifecycle: VariableOrOptional[Lifecycle] = None @classmethod - def from_dict(cls, value: "ModelProviderServiceDict") -> "Self": + def from_dict(cls, value: 'ModelProviderServiceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config.py index 57ff32a4bc9..7b945a3e222 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config.py @@ -1,64 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.model_provider_services._models.inference_table_config import ( - InferenceTableConfig, - InferenceTableConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ( - ModelProviderServiceConfigAmazonBedrockProviderConfig, - ModelProviderServiceConfigAmazonBedrockProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ( - ModelProviderServiceConfigAnthropicProviderConfig, - ModelProviderServiceConfigAnthropicProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ( - ModelProviderServiceConfigAzureOpenAiProviderConfig, - ModelProviderServiceConfigAzureOpenAiProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ( - ModelProviderServiceConfigCustomProviderConfig, - ModelProviderServiceConfigCustomProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ( - ModelProviderServiceConfigExternalModelProviderType, - ModelProviderServiceConfigExternalModelProviderTypeParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ( - ModelProviderServiceConfigGeminiEnterpriseProviderConfig, - ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ( - ModelProviderServiceConfigMicrosoftFoundryProviderConfig, - ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ( - ModelProviderServiceConfigModelTargetConfig, - ModelProviderServiceConfigModelTargetConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ( - ModelProviderServiceConfigOpenAiProviderConfig, - ModelProviderServiceConfigOpenAiProviderConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ( - ModelProviderServiceConfigProviderPricingConfig, - ModelProviderServiceConfigProviderPricingConfigParam, -) -from databricks.bundles.model_provider_services._models.rate_limit import ( - RateLimit, - RateLimitParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfig: """ @@ -68,7 +58,6 @@ class ModelProviderServiceConfig: headers, query parameters, and unmanaged subpaths cross the trust boundary to the upstream provider. """ - allow_all_targets: VariableOrOptional[bool] = None """ When true, accepts any model exposed by the upstream provider; `targets` @@ -76,39 +65,30 @@ class ModelProviderServiceConfig: models listed in `targets` are routable. Defaults to false. """ - amazon_bedrock: VariableOrOptional[ - ModelProviderServiceConfigAmazonBedrockProviderConfig - ] = None + amazon_bedrock: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderConfig] = None """ Amazon Bedrock provider configuration. """ - anthropic: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderConfig] = ( - None - ) + anthropic: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderConfig] = None """ Anthropic provider configuration. Exactly one of `direct` or `relayed` must be set on Create; the two are mutually exclusive. """ - azure_openai: VariableOrOptional[ - ModelProviderServiceConfigAzureOpenAiProviderConfig - ] = None + azure_openai: VariableOrOptional[ModelProviderServiceConfigAzureOpenAiProviderConfig] = None """ Azure OpenAI provider configuration. """ - bedrock_mantle: VariableOrOptional[ - ModelProviderServiceConfigAmazonBedrockProviderConfig - ] = None + bedrock_mantle: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderConfig] = None """ Amazon Bedrock provider configuration. """ custom: VariableOrOptional[ModelProviderServiceConfigCustomProviderConfig] = None """ - Custom OpenAI-compatible provider configuration with bearer-token - authentication. + Custom OpenAI-compatible provider configuration. """ forward_headers: VariableOrOptional[bool] = None @@ -137,9 +117,7 @@ class ModelProviderServiceConfig: rejected whenever Service Policies are attached to this securable. """ - gemini_enterprise: VariableOrOptional[ - ModelProviderServiceConfigGeminiEnterpriseProviderConfig - ] = None + gemini_enterprise: VariableOrOptional[ModelProviderServiceConfigGeminiEnterpriseProviderConfig] = None """ Gemini Enterprise provider configuration. """ @@ -151,9 +129,7 @@ class ModelProviderServiceConfig: service's inference table instead. """ - microsoft_foundry: VariableOrOptional[ - ModelProviderServiceConfigMicrosoftFoundryProviderConfig - ] = None + microsoft_foundry: VariableOrOptional[ModelProviderServiceConfigMicrosoftFoundryProviderConfig] = None """ Microsoft Foundry provider configuration. """ @@ -169,9 +145,7 @@ class ModelProviderServiceConfig: configuration is retained on Create and Update and applies no adjustment. """ - provider_type: VariableOrOptional[ - ModelProviderServiceConfigExternalModelProviderType - ] = None + provider_type: VariableOrOptional[ModelProviderServiceConfigExternalModelProviderType] = None """ External model provider. Required on Create and immutable thereafter. Set the matching provider-specific configuration, such as `openai`, @@ -184,9 +158,7 @@ class ModelProviderServiceConfig: routed through a model service use that model service's rate limits instead. """ - targets: VariableOrList[ModelProviderServiceConfigModelTargetConfig] = field( - default_factory=list - ) + targets: VariableOrList[ModelProviderServiceConfigModelTargetConfig] = field(default_factory=list) """ Models and provider-native API types exposed by this provider service. Each entry must include at least one `native_api_types` value. When @@ -197,11 +169,12 @@ class ModelProviderServiceConfig: """ @classmethod - def from_dict(cls, value: "ModelProviderServiceConfigDict") -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigDict(TypedDict, total=False): @@ -214,39 +187,30 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): models listed in `targets` are routable. Defaults to false. """ - amazon_bedrock: VariableOrOptional[ - ModelProviderServiceConfigAmazonBedrockProviderConfigParam - ] + amazon_bedrock: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderConfigParam] """ Amazon Bedrock provider configuration. """ - anthropic: VariableOrOptional[ - ModelProviderServiceConfigAnthropicProviderConfigParam - ] + anthropic: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderConfigParam] """ Anthropic provider configuration. Exactly one of `direct` or `relayed` must be set on Create; the two are mutually exclusive. """ - azure_openai: VariableOrOptional[ - ModelProviderServiceConfigAzureOpenAiProviderConfigParam - ] + azure_openai: VariableOrOptional[ModelProviderServiceConfigAzureOpenAiProviderConfigParam] """ Azure OpenAI provider configuration. """ - bedrock_mantle: VariableOrOptional[ - ModelProviderServiceConfigAmazonBedrockProviderConfigParam - ] + bedrock_mantle: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderConfigParam] """ Amazon Bedrock provider configuration. """ custom: VariableOrOptional[ModelProviderServiceConfigCustomProviderConfigParam] """ - Custom OpenAI-compatible provider configuration with bearer-token - authentication. + Custom OpenAI-compatible provider configuration. """ forward_headers: VariableOrOptional[bool] @@ -275,9 +239,7 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): rejected whenever Service Policies are attached to this securable. """ - gemini_enterprise: VariableOrOptional[ - ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam - ] + gemini_enterprise: VariableOrOptional[ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam] """ Gemini Enterprise provider configuration. """ @@ -289,9 +251,7 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): service's inference table instead. """ - microsoft_foundry: VariableOrOptional[ - ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam - ] + microsoft_foundry: VariableOrOptional[ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam] """ Microsoft Foundry provider configuration. """ @@ -307,9 +267,7 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): configuration is retained on Create and Update and applies no adjustment. """ - provider_type: VariableOrOptional[ - ModelProviderServiceConfigExternalModelProviderTypeParam - ] + provider_type: VariableOrOptional[ModelProviderServiceConfigExternalModelProviderTypeParam] """ External model provider. Required on Create and immutable thereafter. Set the matching provider-specific configuration, such as `openai`, @@ -333,6 +291,4 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): """ -ModelProviderServiceConfigParam = ( - ModelProviderServiceConfigDict | ModelProviderServiceConfig -) +ModelProviderServiceConfigParam = ModelProviderServiceConfigDict | ModelProviderServiceConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_config.py index cf6015f13c5..4a4823b1188 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_config.py @@ -1,55 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ( - ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, - ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigAmazonBedrockProviderConfig: """ Amazon Bedrock provider configuration. """ - - direct: VariableOrOptional[ - ModelProviderServiceConfigAmazonBedrockProviderDirectConfig - ] = None + direct: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderDirectConfig] = None """ Amazon Bedrock region and authentication configuration. """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigAmazonBedrockProviderConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigAmazonBedrockProviderConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigAmazonBedrockProviderConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigAmazonBedrockProviderConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigAmazonBedrockProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ - ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam - ] + direct: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam] """ Amazon Bedrock region and authentication configuration. """ -ModelProviderServiceConfigAmazonBedrockProviderConfigParam = ( - ModelProviderServiceConfigAmazonBedrockProviderConfigDict - | ModelProviderServiceConfigAmazonBedrockProviderConfig -) +ModelProviderServiceConfigAmazonBedrockProviderConfigParam = ModelProviderServiceConfigAmazonBedrockProviderConfigDict | ModelProviderServiceConfigAmazonBedrockProviderConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_direct_config.py index 911338ad79e..f075c1e30e4 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_direct_config.py @@ -1,29 +1,59 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ( - ModelProviderServiceConfigAwsAccessKey, - ModelProviderServiceConfigAwsAccessKeyParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( - ModelProviderServiceConfigServiceCredential, - ModelProviderServiceConfigServiceCredentialParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigAmazonBedrockProviderDirectConfig: """ Direct form of Amazon Bedrock provider config. - + Authentication is one of two mutually exclusive modes, exactly one of which must be supplied on Create: - Access keys: set `aws_access_key`, leave `service_credential` unset. @@ -33,7 +63,6 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfig: held on this message. Setting more than one mode is rejected. """ - aws_access_key: VariableOrOptional[ModelProviderServiceConfigAwsAccessKey] = None """ AWS access-key-pair authentication. Set `access_key_id` and @@ -47,9 +76,7 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfig: Required on Create. """ - service_credential: VariableOrOptional[ - ModelProviderServiceConfigServiceCredential - ] = None + service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredential] = None """ Reference to a Unity Catalog service credential authorizing Bedrock requests. On Create, supply `service_credential.name` in the form @@ -60,20 +87,15 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict') -> 'Self': return _transform(cls, value) - def as_dict( - self, - ) -> "ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict': + return _transform_to_json_value(self) # type:ignore + -class ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict( - TypedDict, total=False -): +class ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict(TypedDict, total=False): """""" aws_access_key: VariableOrOptional[ModelProviderServiceConfigAwsAccessKeyParam] @@ -89,9 +111,7 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict( Required on Create. """ - service_credential: VariableOrOptional[ - ModelProviderServiceConfigServiceCredentialParam - ] + service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredentialParam] """ Reference to a Unity Catalog service credential authorizing Bedrock requests. On Create, supply `service_credential.name` in the form @@ -102,7 +122,4 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict( """ -ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam = ( - ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict - | ModelProviderServiceConfigAmazonBedrockProviderDirectConfig -) +ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam = ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict | ModelProviderServiceConfigAmazonBedrockProviderDirectConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_config.py index c7b3311ca93..ea8c077414a 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_config.py @@ -1,42 +1,67 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ( - ModelProviderServiceConfigAnthropicProviderDirectConfig, - ModelProviderServiceConfigAnthropicProviderDirectConfigParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ( - ModelProviderServiceConfigAnthropicProviderRelayedConfig, - ModelProviderServiceConfigAnthropicProviderRelayedConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigAnthropicProviderConfig: """ Anthropic provider configuration. Exactly one of `direct` or `relayed` must be set on Create; the two are mutually exclusive. """ - - direct: VariableOrOptional[ - ModelProviderServiceConfigAnthropicProviderDirectConfig - ] = None + direct: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderDirectConfig] = None """ Direct authentication with an API key supplied in `direct.api_key.plaintext`. Required unless `relayed` is set. """ - relayed: VariableOrOptional[ - ModelProviderServiceConfigAnthropicProviderRelayedConfig - ] = None + relayed: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderRelayedConfig] = None """ Relayed authentication. Each inference request supplies the caller's OAuth token, which is forwarded to Anthropic. No Anthropic credential is @@ -44,29 +69,24 @@ class ModelProviderServiceConfigAnthropicProviderConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigAnthropicProviderConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigAnthropicProviderConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigAnthropicProviderConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigAnthropicProviderConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigAnthropicProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ - ModelProviderServiceConfigAnthropicProviderDirectConfigParam - ] + direct: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderDirectConfigParam] """ Direct authentication with an API key supplied in `direct.api_key.plaintext`. Required unless `relayed` is set. """ - relayed: VariableOrOptional[ - ModelProviderServiceConfigAnthropicProviderRelayedConfigParam - ] + relayed: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderRelayedConfigParam] """ Relayed authentication. Each inference request supplies the caller's OAuth token, which is forwarded to Anthropic. No Anthropic credential is @@ -74,7 +94,4 @@ class ModelProviderServiceConfigAnthropicProviderConfigDict(TypedDict, total=Fal """ -ModelProviderServiceConfigAnthropicProviderConfigParam = ( - ModelProviderServiceConfigAnthropicProviderConfigDict - | ModelProviderServiceConfigAnthropicProviderConfig -) +ModelProviderServiceConfigAnthropicProviderConfigParam = ModelProviderServiceConfigAnthropicProviderConfigDict | ModelProviderServiceConfigAnthropicProviderConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_direct_config.py index d03a79d596d..b6a0705bb01 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_direct_config.py @@ -1,26 +1,59 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigAnthropicProviderDirectConfig: """ Direct form of Anthropic provider config. """ - api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Anthropic API key. Required when creating the service. Supply the value @@ -28,18 +61,15 @@ class ModelProviderServiceConfigAnthropicProviderDirectConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigAnthropicProviderDirectConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigAnthropicProviderDirectConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigAnthropicProviderDirectConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigAnthropicProviderDirectConfigDict': + return _transform_to_json_value(self) # type:ignore + -class ModelProviderServiceConfigAnthropicProviderDirectConfigDict( - TypedDict, total=False -): +class ModelProviderServiceConfigAnthropicProviderDirectConfigDict(TypedDict, total=False): """""" api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecretParam] @@ -49,7 +79,4 @@ class ModelProviderServiceConfigAnthropicProviderDirectConfigDict( """ -ModelProviderServiceConfigAnthropicProviderDirectConfigParam = ( - ModelProviderServiceConfigAnthropicProviderDirectConfigDict - | ModelProviderServiceConfigAnthropicProviderDirectConfig -) +ModelProviderServiceConfigAnthropicProviderDirectConfigParam = ModelProviderServiceConfigAnthropicProviderDirectConfigDict | ModelProviderServiceConfigAnthropicProviderDirectConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_relayed_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_relayed_config.py index 7754ff58862..1f3c3a7c38e 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_relayed_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_relayed_config.py @@ -1,15 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigAnthropicProviderRelayedConfig: """ @@ -17,24 +56,17 @@ class ModelProviderServiceConfigAnthropicProviderRelayedConfig: the caller's OAuth token, which is forwarded to Anthropic. No Anthropic credential is stored. """ - @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigAnthropicProviderRelayedConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigAnthropicProviderRelayedConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigAnthropicProviderRelayedConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigAnthropicProviderRelayedConfigDict': + return _transform_to_json_value(self) # type:ignore + -class ModelProviderServiceConfigAnthropicProviderRelayedConfigDict( - TypedDict, total=False -): +class ModelProviderServiceConfigAnthropicProviderRelayedConfigDict(TypedDict, total=False): """""" -ModelProviderServiceConfigAnthropicProviderRelayedConfigParam = ( - ModelProviderServiceConfigAnthropicProviderRelayedConfigDict - | ModelProviderServiceConfigAnthropicProviderRelayedConfig -) +ModelProviderServiceConfigAnthropicProviderRelayedConfigParam = ModelProviderServiceConfigAnthropicProviderRelayedConfigDict | ModelProviderServiceConfigAnthropicProviderRelayedConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_aws_access_key.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_aws_access_key.py index 6c57b6912d8..2672d0e6fa8 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_aws_access_key.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_aws_access_key.py @@ -1,26 +1,59 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigAwsAccessKey: """ AWS access-key-pair auth for Amazon Bedrock: a SigV4-signing key pair. """ - access_key_id: VariableOrOptional[str] = None """ AWS access key ID. Required on Create when using access-key auth. Treated as @@ -28,9 +61,7 @@ class ModelProviderServiceConfigAwsAccessKey: scrubbed from audit logs. """ - secret_access_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = ( - None - ) + secret_access_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ AWS secret access key paired with `access_key_id`. Required when creating a service with access-key authentication. Supply the value in @@ -38,11 +69,12 @@ class ModelProviderServiceConfigAwsAccessKey: """ @classmethod - def from_dict(cls, value: "ModelProviderServiceConfigAwsAccessKeyDict") -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigAwsAccessKeyDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigAwsAccessKeyDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigAwsAccessKeyDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigAwsAccessKeyDict(TypedDict, total=False): @@ -63,6 +95,4 @@ class ModelProviderServiceConfigAwsAccessKeyDict(TypedDict, total=False): """ -ModelProviderServiceConfigAwsAccessKeyParam = ( - ModelProviderServiceConfigAwsAccessKeyDict | ModelProviderServiceConfigAwsAccessKey -) +ModelProviderServiceConfigAwsAccessKeyParam = ModelProviderServiceConfigAwsAccessKeyDict | ModelProviderServiceConfigAwsAccessKey diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_config.py index 60726eeffed..803e33f89c4 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_config.py @@ -1,55 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ( - ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, - ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigAzureOpenAiProviderConfig: """ Azure OpenAI provider configuration. """ - - direct: VariableOrOptional[ - ModelProviderServiceConfigAzureOpenAiProviderDirectConfig - ] = None + direct: VariableOrOptional[ModelProviderServiceConfigAzureOpenAiProviderDirectConfig] = None """ Azure OpenAI endpoint and authentication configuration. """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigAzureOpenAiProviderConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigAzureOpenAiProviderConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigAzureOpenAiProviderConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigAzureOpenAiProviderConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigAzureOpenAiProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ - ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam - ] + direct: VariableOrOptional[ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam] """ Azure OpenAI endpoint and authentication configuration. """ -ModelProviderServiceConfigAzureOpenAiProviderConfigParam = ( - ModelProviderServiceConfigAzureOpenAiProviderConfigDict - | ModelProviderServiceConfigAzureOpenAiProviderConfig -) +ModelProviderServiceConfigAzureOpenAiProviderConfigParam = ModelProviderServiceConfigAzureOpenAiProviderConfigDict | ModelProviderServiceConfigAzureOpenAiProviderConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_direct_config.py index 9ba84e99974..fd5a80176c9 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_direct_config.py @@ -1,28 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ( - ModelProviderServiceConfigEntraServicePrincipal, - ModelProviderServiceConfigEntraServicePrincipalParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( - ModelProviderServiceConfigServiceCredential, - ModelProviderServiceConfigServiceCredentialParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigAzureOpenAiProviderDirectConfig: """ @@ -39,7 +65,6 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfig: supported on Azure-hosted workspaces. Setting more than one mode is rejected. """ - api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Azure OpenAI API key. Supply the value in `api_key.plaintext`. Mutually @@ -53,18 +78,14 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfig: `https://myresource.openai.azure.com`. Required on Create. """ - entra_service_principal: VariableOrOptional[ - ModelProviderServiceConfigEntraServicePrincipal - ] = None + entra_service_principal: VariableOrOptional[ModelProviderServiceConfigEntraServicePrincipal] = None """ Entra ID service-principal authentication. Set `tenant_id`, `client_id`, and `client_secret.plaintext`. Mutually exclusive with `api_key` and `service_credential`. """ - service_credential: VariableOrOptional[ - ModelProviderServiceConfigServiceCredential - ] = None + service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredential] = None """ Reference to a Unity Catalog service credential authorizing Azure OpenAI requests. On Create, supply `service_credential.name` in the form @@ -75,20 +96,15 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict') -> 'Self': return _transform(cls, value) - def as_dict( - self, - ) -> "ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict': + return _transform_to_json_value(self) # type:ignore + -class ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict( - TypedDict, total=False -): +class ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict(TypedDict, total=False): """""" api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecretParam] @@ -104,18 +120,14 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict( `https://myresource.openai.azure.com`. Required on Create. """ - entra_service_principal: VariableOrOptional[ - ModelProviderServiceConfigEntraServicePrincipalParam - ] + entra_service_principal: VariableOrOptional[ModelProviderServiceConfigEntraServicePrincipalParam] """ Entra ID service-principal authentication. Set `tenant_id`, `client_id`, and `client_secret.plaintext`. Mutually exclusive with `api_key` and `service_credential`. """ - service_credential: VariableOrOptional[ - ModelProviderServiceConfigServiceCredentialParam - ] + service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredentialParam] """ Reference to a Unity Catalog service credential authorizing Azure OpenAI requests. On Create, supply `service_credential.name` in the form @@ -126,7 +138,4 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict( """ -ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam = ( - ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict - | ModelProviderServiceConfigAzureOpenAiProviderDirectConfig -) +ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam = ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict | ModelProviderServiceConfigAzureOpenAiProviderDirectConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_api_key_header_auth.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_api_key_header_auth.py index 78865d1c8f2..12a4f31be79 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_api_key_header_auth.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_api_key_header_auth.py @@ -1,20 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth: """ @@ -22,7 +56,6 @@ class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth: forwarded on outbound requests under a caller-chosen HTTP header, as `: `. """ - api_key_name: VariableOrOptional[str] = None """ HTTP header name that carries the API key on outbound requests (e.g., @@ -37,18 +70,15 @@ class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict': + return _transform_to_json_value(self) # type:ignore + -class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict( - TypedDict, total=False -): +class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict(TypedDict, total=False): """""" api_key_name: VariableOrOptional[str] @@ -65,7 +95,4 @@ class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict( """ -ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam = ( - ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict - | ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth -) +ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam = ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict | ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_config.py index e10c0335481..92e5f98d4fd 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_config.py @@ -1,56 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ( - ModelProviderServiceConfigCustomProviderDirectConfig, - ModelProviderServiceConfigCustomProviderDirectConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigCustomProviderConfig: """ - Custom OpenAI-compatible provider configuration with bearer-token - authentication. + Custom OpenAI-compatible provider configuration. """ - - direct: VariableOrOptional[ModelProviderServiceConfigCustomProviderDirectConfig] = ( - None - ) + direct: VariableOrOptional[ModelProviderServiceConfigCustomProviderDirectConfig] = None """ Endpoint and authentication configuration for the custom provider. """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigCustomProviderConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigCustomProviderConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigCustomProviderConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigCustomProviderConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigCustomProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ - ModelProviderServiceConfigCustomProviderDirectConfigParam - ] + direct: VariableOrOptional[ModelProviderServiceConfigCustomProviderDirectConfigParam] """ Endpoint and authentication configuration for the custom provider. """ -ModelProviderServiceConfigCustomProviderConfigParam = ( - ModelProviderServiceConfigCustomProviderConfigDict - | ModelProviderServiceConfigCustomProviderConfig -) +ModelProviderServiceConfigCustomProviderConfigParam = ModelProviderServiceConfigCustomProviderConfigDict | ModelProviderServiceConfigCustomProviderConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_direct_config.py index 42e4d350a24..d16fe13463a 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_direct_config.py @@ -1,24 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ( - ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, - ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigCustomProviderDirectConfig: """ @@ -26,7 +56,6 @@ class ModelProviderServiceConfigCustomProviderDirectConfig: secret as an `Authorization` bearer token, or `header_auth` to forward it under a caller-chosen HTTP header. """ - api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Bearer token forwarded in the `Authorization` header. Supply the value @@ -39,9 +68,7 @@ class ModelProviderServiceConfigCustomProviderDirectConfig: `https://api.example.com/v1`). Required on Create. """ - header_auth: VariableOrOptional[ - ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth - ] = None + header_auth: VariableOrOptional[ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth] = None """ Header-based API-key auth: the secret is forwarded on outbound requests under a caller-chosen HTTP header rather than as an `Authorization` @@ -49,13 +76,12 @@ class ModelProviderServiceConfigCustomProviderDirectConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigCustomProviderDirectConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigCustomProviderDirectConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigCustomProviderDirectConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigCustomProviderDirectConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigCustomProviderDirectConfigDict(TypedDict, total=False): @@ -73,9 +99,7 @@ class ModelProviderServiceConfigCustomProviderDirectConfigDict(TypedDict, total= `https://api.example.com/v1`). Required on Create. """ - header_auth: VariableOrOptional[ - ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam - ] + header_auth: VariableOrOptional[ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam] """ Header-based API-key auth: the secret is forwarded on outbound requests under a caller-chosen HTTP header rather than as an `Authorization` @@ -83,7 +107,4 @@ class ModelProviderServiceConfigCustomProviderDirectConfigDict(TypedDict, total= """ -ModelProviderServiceConfigCustomProviderDirectConfigParam = ( - ModelProviderServiceConfigCustomProviderDirectConfigDict - | ModelProviderServiceConfigCustomProviderDirectConfig -) +ModelProviderServiceConfigCustomProviderDirectConfigParam = ModelProviderServiceConfigCustomProviderDirectConfigDict | ModelProviderServiceConfigCustomProviderDirectConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_entra_service_principal.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_entra_service_principal.py index cf5fcefd569..8924ee016ee 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_entra_service_principal.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_entra_service_principal.py @@ -1,20 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigEntraServicePrincipal: """ @@ -23,7 +57,6 @@ class ModelProviderServiceConfigEntraServicePrincipal: it. AI Gateway exchanges these credentials for an Entra bearer token for requests to Azure OpenAI or Microsoft Foundry. """ - client_id: VariableOrOptional[str] = None """ Entra ID client (application) ID. Required on Create. @@ -40,13 +73,12 @@ class ModelProviderServiceConfigEntraServicePrincipal: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigEntraServicePrincipalDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigEntraServicePrincipalDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigEntraServicePrincipalDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigEntraServicePrincipalDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigEntraServicePrincipalDict(TypedDict, total=False): @@ -68,7 +100,4 @@ class ModelProviderServiceConfigEntraServicePrincipalDict(TypedDict, total=False """ -ModelProviderServiceConfigEntraServicePrincipalParam = ( - ModelProviderServiceConfigEntraServicePrincipalDict - | ModelProviderServiceConfigEntraServicePrincipal -) +ModelProviderServiceConfigEntraServicePrincipalParam = ModelProviderServiceConfigEntraServicePrincipalDict | ModelProviderServiceConfigEntraServicePrincipal diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_external_model_provider_type.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_external_model_provider_type.py index f3a79bd6cdf..2416cffa698 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_external_model_provider_type.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_external_model_provider_type.py @@ -1,44 +1,61 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam class ModelProviderServiceConfigExternalModelProviderType(Enum): """ External LLM provider for an EXTERNAL_FOUNDATION_MODEL destination. """ - EXTERNAL_MODEL_PROVIDER_TYPE_OPENAI = "EXTERNAL_MODEL_PROVIDER_TYPE_OPENAI" - EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI = ( - "EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI" - ) + EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI = "EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI" EXTERNAL_MODEL_PROVIDER_TYPE_ANTHROPIC = "EXTERNAL_MODEL_PROVIDER_TYPE_ANTHROPIC" - EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK = ( - "EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK" - ) + EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK = "EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK" EXTERNAL_MODEL_PROVIDER_TYPE_CUSTOM = "EXTERNAL_MODEL_PROVIDER_TYPE_CUSTOM" - EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY = ( - "EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY" - ) - EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE = ( - "EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE" - ) - EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE = ( - "EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE" - ) - - -ModelProviderServiceConfigExternalModelProviderTypeParam = ( - Literal[ - "EXTERNAL_MODEL_PROVIDER_TYPE_OPENAI", - "EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI", - "EXTERNAL_MODEL_PROVIDER_TYPE_ANTHROPIC", - "EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK", - "EXTERNAL_MODEL_PROVIDER_TYPE_CUSTOM", - "EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY", - "EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE", - "EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE", - ] - | ModelProviderServiceConfigExternalModelProviderType -) + EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY = "EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY" + EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE = "EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE" + EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE = "EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE" + +ModelProviderServiceConfigExternalModelProviderTypeParam = Literal["EXTERNAL_MODEL_PROVIDER_TYPE_OPENAI", "EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI", "EXTERNAL_MODEL_PROVIDER_TYPE_ANTHROPIC", "EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK", "EXTERNAL_MODEL_PROVIDER_TYPE_CUSTOM", "EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY", "EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE", "EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE"] | ModelProviderServiceConfigExternalModelProviderType diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_config.py index 38f005ad002..9ea1598cc3b 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_config.py @@ -1,57 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ( - ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, - ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigGeminiEnterpriseProviderConfig: """ Gemini Enterprise provider configuration. """ - - direct: VariableOrOptional[ - ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig - ] = None + direct: VariableOrOptional[ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig] = None """ Gemini Enterprise project, region, and authentication configuration. """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict': + return _transform_to_json_value(self) # type:ignore + -class ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict( - TypedDict, total=False -): +class ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ - ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam - ] + direct: VariableOrOptional[ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam] """ Gemini Enterprise project, region, and authentication configuration. """ -ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam = ( - ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict - | ModelProviderServiceConfigGeminiEnterpriseProviderConfig -) +ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam = ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict | ModelProviderServiceConfigGeminiEnterpriseProviderConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_direct_config.py index 157ef00adfd..803dadf3019 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_direct_config.py @@ -1,31 +1,60 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( - ModelProviderServiceConfigServiceCredential, - ModelProviderServiceConfigServiceCredentialParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig: """ Direct Gemini Enterprise provider configuration. An API key is required when creating the service. """ - api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Google Gemini Enterprise API key. Required when creating the service. @@ -43,9 +72,7 @@ class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig: Required on Create. """ - service_credential: VariableOrOptional[ - ModelProviderServiceConfigServiceCredential - ] = None + service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredential] = None """ Reference to a Unity Catalog service credential authorizing Gemini Enterprise requests. On Create, supply `service_credential.name` as @@ -57,20 +84,15 @@ class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict') -> 'Self': return _transform(cls, value) - def as_dict( - self, - ) -> "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict': + return _transform_to_json_value(self) # type:ignore + -class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict( - TypedDict, total=False -): +class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict(TypedDict, total=False): """""" api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecretParam] @@ -90,9 +112,7 @@ class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict( Required on Create. """ - service_credential: VariableOrOptional[ - ModelProviderServiceConfigServiceCredentialParam - ] + service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredentialParam] """ Reference to a Unity Catalog service credential authorizing Gemini Enterprise requests. On Create, supply `service_credential.name` as @@ -104,7 +124,4 @@ class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict( """ -ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam = ( - ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict - | ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig -) +ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam = ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict | ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_config.py index c63fe203888..8d69e4636ee 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_config.py @@ -1,57 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ( - ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, - ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigMicrosoftFoundryProviderConfig: """ Microsoft Foundry provider configuration. """ - - direct: VariableOrOptional[ - ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig - ] = None + direct: VariableOrOptional[ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig] = None """ Microsoft Foundry endpoint and authentication configuration. """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict': + return _transform_to_json_value(self) # type:ignore + -class ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict( - TypedDict, total=False -): +class ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ - ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam - ] + direct: VariableOrOptional[ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam] """ Microsoft Foundry endpoint and authentication configuration. """ -ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam = ( - ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict - | ModelProviderServiceConfigMicrosoftFoundryProviderConfig -) +ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam = ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict | ModelProviderServiceConfigMicrosoftFoundryProviderConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_direct_config.py index 5caad765601..6f2d1945bb5 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_direct_config.py @@ -1,33 +1,59 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ( - ModelProviderServiceConfigEntraServicePrincipal, - ModelProviderServiceConfigEntraServicePrincipalParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretParam, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( - ModelProviderServiceConfigServiceCredential, - ModelProviderServiceConfigServiceCredentialParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig: """ Direct form of Microsoft Foundry provider config. - + Authentication is one of three mutually exclusive modes, exactly one of which must be supplied on Create: - API key: set `api_key`, leave `entra_service_principal` and @@ -43,7 +69,6 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig: supported on Azure-hosted workspaces. Setting more than one mode is rejected. """ - api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Microsoft Foundry API key. Supply the value in `api_key.plaintext`. @@ -56,18 +81,14 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig: Microsoft Foundry endpoint URL. Required on Create. """ - entra_service_principal: VariableOrOptional[ - ModelProviderServiceConfigEntraServicePrincipal - ] = None + entra_service_principal: VariableOrOptional[ModelProviderServiceConfigEntraServicePrincipal] = None """ Entra ID service-principal authentication. Set `tenant_id`, `client_id`, and `client_secret.plaintext`. Mutually exclusive with `api_key` and `service_credential`. """ - service_credential: VariableOrOptional[ - ModelProviderServiceConfigServiceCredential - ] = None + service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredential] = None """ Reference to a Unity Catalog service credential authorizing Microsoft Foundry requests. On Create, supply `service_credential.name` in the form @@ -78,20 +99,15 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict') -> 'Self': return _transform(cls, value) - def as_dict( - self, - ) -> "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict': + return _transform_to_json_value(self) # type:ignore + -class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict( - TypedDict, total=False -): +class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict(TypedDict, total=False): """""" api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecretParam] @@ -106,18 +122,14 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict( Microsoft Foundry endpoint URL. Required on Create. """ - entra_service_principal: VariableOrOptional[ - ModelProviderServiceConfigEntraServicePrincipalParam - ] + entra_service_principal: VariableOrOptional[ModelProviderServiceConfigEntraServicePrincipalParam] """ Entra ID service-principal authentication. Set `tenant_id`, `client_id`, and `client_secret.plaintext`. Mutually exclusive with `api_key` and `service_credential`. """ - service_credential: VariableOrOptional[ - ModelProviderServiceConfigServiceCredentialParam - ] + service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredentialParam] """ Reference to a Unity Catalog service credential authorizing Microsoft Foundry requests. On Create, supply `service_credential.name` in the form @@ -128,7 +140,4 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict( """ -ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam = ( - ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict - | ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig -) +ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam = ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict | ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_model_target_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_model_target_config.py index 671c9eb4490..6d91404cc1e 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_model_target_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_model_target_config.py @@ -1,27 +1,60 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigModelTargetConfig: """ Model target configuration shared by model provider service targets and external model destinations on model services. """ - model: VariableOr[str] """ Provider-side model identifier, such as `gpt-5` or `claude-opus-4-7`. @@ -49,13 +82,12 @@ class ModelProviderServiceConfigModelTargetConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigModelTargetConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigModelTargetConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigModelTargetConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigModelTargetConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigModelTargetConfigDict(TypedDict, total=False): @@ -88,7 +120,4 @@ class ModelProviderServiceConfigModelTargetConfigDict(TypedDict, total=False): """ -ModelProviderServiceConfigModelTargetConfigParam = ( - ModelProviderServiceConfigModelTargetConfigDict - | ModelProviderServiceConfigModelTargetConfig -) +ModelProviderServiceConfigModelTargetConfigParam = ModelProviderServiceConfigModelTargetConfigDict | ModelProviderServiceConfigModelTargetConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_config.py index 70200550381..598b38067f6 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_config.py @@ -1,55 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ( - ModelProviderServiceConfigOpenAiProviderDirectConfig, - ModelProviderServiceConfigOpenAiProviderDirectConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigOpenAiProviderConfig: """ OpenAI provider configuration. """ - - direct: VariableOrOptional[ModelProviderServiceConfigOpenAiProviderDirectConfig] = ( - None - ) + direct: VariableOrOptional[ModelProviderServiceConfigOpenAiProviderDirectConfig] = None """ OpenAI configuration with an API key supplied in the request. """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigOpenAiProviderConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigOpenAiProviderConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigOpenAiProviderConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigOpenAiProviderConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigOpenAiProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ - ModelProviderServiceConfigOpenAiProviderDirectConfigParam - ] + direct: VariableOrOptional[ModelProviderServiceConfigOpenAiProviderDirectConfigParam] """ OpenAI configuration with an API key supplied in the request. """ -ModelProviderServiceConfigOpenAiProviderConfigParam = ( - ModelProviderServiceConfigOpenAiProviderConfigDict - | ModelProviderServiceConfigOpenAiProviderConfig -) +ModelProviderServiceConfigOpenAiProviderConfigParam = ModelProviderServiceConfigOpenAiProviderConfigDict | ModelProviderServiceConfigOpenAiProviderConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_direct_config.py index 128ce50e14e..148137f7f90 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_direct_config.py @@ -1,26 +1,59 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( - ModelProviderServiceConfigProviderSecret, - ModelProviderServiceConfigProviderSecretParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigOpenAiProviderDirectConfig: """ Direct (inline-credentials) form of the OpenAI provider config. """ - api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ OpenAI API key. Required when creating the service. Supply the value in @@ -40,13 +73,12 @@ class ModelProviderServiceConfigOpenAiProviderDirectConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigOpenAiProviderDirectConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigOpenAiProviderDirectConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigOpenAiProviderDirectConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigOpenAiProviderDirectConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigOpenAiProviderDirectConfigDict(TypedDict, total=False): @@ -71,7 +103,4 @@ class ModelProviderServiceConfigOpenAiProviderDirectConfigDict(TypedDict, total= """ -ModelProviderServiceConfigOpenAiProviderDirectConfigParam = ( - ModelProviderServiceConfigOpenAiProviderDirectConfigDict - | ModelProviderServiceConfigOpenAiProviderDirectConfig -) +ModelProviderServiceConfigOpenAiProviderDirectConfigParam = ModelProviderServiceConfigOpenAiProviderDirectConfigDict | ModelProviderServiceConfigOpenAiProviderDirectConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_pricing_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_pricing_config.py index 88e0d02484c..bbbef776c6e 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_pricing_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_pricing_config.py @@ -1,23 +1,60 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigProviderPricingConfig: """ Pricing adjustments applied to this provider service's external-model spend estimates. """ - default_discount_basis_points: VariableOrOptional[int] = None """ Provider-wide adjustment to external-model spend estimates in basis points: @@ -27,13 +64,12 @@ class ModelProviderServiceConfigProviderPricingConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigProviderPricingConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigProviderPricingConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigProviderPricingConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigProviderPricingConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigProviderPricingConfigDict(TypedDict, total=False): @@ -48,7 +84,4 @@ class ModelProviderServiceConfigProviderPricingConfigDict(TypedDict, total=False """ -ModelProviderServiceConfigProviderPricingConfigParam = ( - ModelProviderServiceConfigProviderPricingConfigDict - | ModelProviderServiceConfigProviderPricingConfig -) +ModelProviderServiceConfigProviderPricingConfigParam = ModelProviderServiceConfigProviderPricingConfigDict | ModelProviderServiceConfigProviderPricingConfig diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_secret.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_secret.py index fd6cef72cf8..6839557023a 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_secret.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_secret.py @@ -1,20 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ( - ModelProviderServiceConfigSecretReference, - ModelProviderServiceConfigSecretReferenceParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigProviderSecret: """ @@ -23,7 +57,6 @@ class ModelProviderServiceConfigProviderSecret: encrypted. The `plaintext` field is `INPUT_ONLY` and never round-trips on reads. """ - plaintext: VariableOrOptional[str] = None """ Inline plaintext credential. INPUT_ONLY: the value never round-trips on @@ -31,9 +64,7 @@ class ModelProviderServiceConfigProviderSecret: object remains present to indicate that a secret is configured. """ - secret_reference: VariableOrOptional[ModelProviderServiceConfigSecretReference] = ( - None - ) + secret_reference: VariableOrOptional[ModelProviderServiceConfigSecretReference] = None """ Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service @@ -44,11 +75,12 @@ class ModelProviderServiceConfigProviderSecret: """ @classmethod - def from_dict(cls, value: "ModelProviderServiceConfigProviderSecretDict") -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigProviderSecretDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigProviderSecretDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigProviderSecretDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigProviderSecretDict(TypedDict, total=False): @@ -72,7 +104,4 @@ class ModelProviderServiceConfigProviderSecretDict(TypedDict, total=False): """ -ModelProviderServiceConfigProviderSecretParam = ( - ModelProviderServiceConfigProviderSecretDict - | ModelProviderServiceConfigProviderSecret -) +ModelProviderServiceConfigProviderSecretParam = ModelProviderServiceConfigProviderSecretDict | ModelProviderServiceConfigProviderSecret diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_secret_reference.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_secret_reference.py index 9c7e95309ff..f4de98a2c4d 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_secret_reference.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_secret_reference.py @@ -1,23 +1,60 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigSecretReference: """ Reference to a customer-owned UC Secret backing a secret-bearing provider field, in the `ProviderSecret.secret_reference` arm. """ - name: VariableOr[str] """ Resource name of the bound UC Secret, in the form @@ -26,13 +63,12 @@ class ModelProviderServiceConfigSecretReference: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigSecretReferenceDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigSecretReferenceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigSecretReferenceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigSecretReferenceDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigSecretReferenceDict(TypedDict, total=False): @@ -46,7 +82,4 @@ class ModelProviderServiceConfigSecretReferenceDict(TypedDict, total=False): """ -ModelProviderServiceConfigSecretReferenceParam = ( - ModelProviderServiceConfigSecretReferenceDict - | ModelProviderServiceConfigSecretReference -) +ModelProviderServiceConfigSecretReferenceParam = ModelProviderServiceConfigSecretReferenceDict | ModelProviderServiceConfigSecretReference diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_service_credential.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_service_credential.py index 12753ea2865..db571e166cd 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_service_credential.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_service_credential.py @@ -1,23 +1,60 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigServiceCredential: """ The customer-owned Unity Catalog service credential a model provider service uses to authenticate to its provider, referenced by name. """ - name: VariableOr[str] """ Resource name of the bound Unity Catalog service credential, in the form @@ -27,13 +64,12 @@ class ModelProviderServiceConfigServiceCredential: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigServiceCredentialDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigServiceCredentialDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigServiceCredentialDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigServiceCredentialDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigServiceCredentialDict(TypedDict, total=False): @@ -48,7 +84,4 @@ class ModelProviderServiceConfigServiceCredentialDict(TypedDict, total=False): """ -ModelProviderServiceConfigServiceCredentialParam = ( - ModelProviderServiceConfigServiceCredentialDict - | ModelProviderServiceConfigServiceCredential -) +ModelProviderServiceConfigServiceCredentialParam = ModelProviderServiceConfigServiceCredentialDict | ModelProviderServiceConfigServiceCredential diff --git a/python/databricks/bundles/model_provider_services/_models/privilege.py b/python/databricks/bundles/model_provider_services/_models/privilege.py index df45211a76a..da63f51e84a 100644 --- a/python/databricks/bundles/model_provider_services/_models/privilege.py +++ b/python/databricks/bundles/model_provider_services/_models/privilege.py @@ -1,7 +1,48 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam class Privilege(Enum): @@ -58,61 +99,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/model_provider_services/_models/privilege_assignment.py b/python/databricks/bundles/model_provider_services/_models/privilege_assignment.py index 810d4ebb627..4512b6d63e8 100644 --- a/python/databricks/bundles/model_provider_services/_models/privilege_assignment.py +++ b/python/databricks/bundles/model_provider_services/_models/privilege_assignment.py @@ -1,20 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.model_provider_services._models.privilege import ( - Privilege, - PrivilegeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -31,11 +65,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/rate_limit.py b/python/databricks/bundles/model_provider_services/_models/rate_limit.py index 5bbb533157a..8b75b82fe94 100644 --- a/python/databricks/bundles/model_provider_services/_models/rate_limit.py +++ b/python/databricks/bundles/model_provider_services/_models/rate_limit.py @@ -1,24 +1,54 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import ( - RateLimitRateLimitKey, - RateLimitRateLimitKeyParam, -) -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import ( - RateLimitRateLimitRenewalPeriod, - RateLimitRateLimitRenewalPeriodParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RateLimit: """ @@ -26,7 +56,6 @@ class RateLimit: unset to impose no limit on that dimension; set a value to cap that dimension within the renewal period. """ - key: VariableOr[RateLimitRateLimitKey] """ Scope of the rate limit. Depending on this value, the limit applies to a @@ -58,11 +87,12 @@ class RateLimit: """ @classmethod - def from_dict(cls, value: "RateLimitDict") -> "Self": + def from_dict(cls, value: 'RateLimitDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RateLimitDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RateLimitDict': + return _transform_to_json_value(self) # type:ignore + class RateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_key.py b/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_key.py index fb907da810f..840e8e25e56 100644 --- a/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_key.py +++ b/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_key.py @@ -1,28 +1,58 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam class RateLimitRateLimitKey(Enum): """ Scope key for a rate limit. """ - RATE_LIMIT_KEY_USER = "RATE_LIMIT_KEY_USER" RATE_LIMIT_KEY_USER_GROUP = "RATE_LIMIT_KEY_USER_GROUP" RATE_LIMIT_KEY_SERVICE_PRINCIPAL = "RATE_LIMIT_KEY_SERVICE_PRINCIPAL" RATE_LIMIT_KEY_SERVICE = "RATE_LIMIT_KEY_SERVICE" RATE_LIMIT_KEY_USER_DEFAULT = "RATE_LIMIT_KEY_USER_DEFAULT" - -RateLimitRateLimitKeyParam = ( - Literal[ - "RATE_LIMIT_KEY_USER", - "RATE_LIMIT_KEY_USER_GROUP", - "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", - "RATE_LIMIT_KEY_SERVICE", - "RATE_LIMIT_KEY_USER_DEFAULT", - ] - | RateLimitRateLimitKey -) +RateLimitRateLimitKeyParam = Literal["RATE_LIMIT_KEY_USER", "RATE_LIMIT_KEY_USER_GROUP", "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", "RATE_LIMIT_KEY_SERVICE", "RATE_LIMIT_KEY_USER_DEFAULT"] | RateLimitRateLimitKey diff --git a/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_renewal_period.py b/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_renewal_period.py index 58636c784c0..b0ff8ae0779 100644 --- a/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_renewal_period.py +++ b/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_renewal_period.py @@ -1,19 +1,55 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam + +from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam class RateLimitRateLimitRenewalPeriod(Enum): """ Renewal period for a rate limit. """ - RATE_LIMIT_RENEWAL_PERIOD_MINUTE = "RATE_LIMIT_RENEWAL_PERIOD_MINUTE" RATE_LIMIT_RENEWAL_PERIOD_HOUR = "RATE_LIMIT_RENEWAL_PERIOD_HOUR" - -RateLimitRateLimitRenewalPeriodParam = ( - Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] - | RateLimitRateLimitRenewalPeriod -) +RateLimitRateLimitRenewalPeriodParam = Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] | RateLimitRateLimitRenewalPeriod diff --git a/python/databricks/bundles/model_services/__init__.py b/python/databricks/bundles/model_services/__init__.py index 9e10e6df19c..86cdfa8031b 100644 --- a/python/databricks/bundles/model_services/__init__.py +++ b/python/databricks/bundles/model_services/__init__.py @@ -1,134 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "InferenceTableConfig", - "InferenceTableConfigDict", - "InferenceTableConfigParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "ModelProviderServiceConfigModelTargetConfig", - "ModelProviderServiceConfigModelTargetConfigDict", - "ModelProviderServiceConfigModelTargetConfigParam", - "ModelService", - "ModelServiceConfig", - "ModelServiceConfigDestinationConfig", - "ModelServiceConfigDestinationConfigDestinationType", - "ModelServiceConfigDestinationConfigDestinationTypeParam", - "ModelServiceConfigDestinationConfigDict", - "ModelServiceConfigDestinationConfigParam", - "ModelServiceConfigDict", - "ModelServiceConfigExternalModelConfig", - "ModelServiceConfigExternalModelConfigDict", - "ModelServiceConfigExternalModelConfigParam", - "ModelServiceConfigFallbackConfig", - "ModelServiceConfigFallbackConfigDict", - "ModelServiceConfigFallbackConfigParam", - "ModelServiceConfigParam", - "ModelServiceConfigPayPerTokenConfig", - "ModelServiceConfigPayPerTokenConfigDict", - "ModelServiceConfigPayPerTokenConfigParam", - "ModelServiceConfigProvisionedThroughputConfig", - "ModelServiceConfigProvisionedThroughputConfigDict", - "ModelServiceConfigProvisionedThroughputConfigParam", - "ModelServiceConfigRoutingConfig", - "ModelServiceConfigRoutingConfigDict", - "ModelServiceConfigRoutingConfigParam", - "ModelServiceDict", - "ModelServiceParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", - "RateLimit", - "RateLimitDict", - "RateLimitParam", - "RateLimitRateLimitKey", - "RateLimitRateLimitKeyParam", - "RateLimitRateLimitRenewalPeriod", - "RateLimitRateLimitRenewalPeriodParam", + 'InferenceTableConfig', + 'InferenceTableConfigDict', + 'InferenceTableConfigParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'ModelProviderServiceConfigModelTargetConfig', + 'ModelProviderServiceConfigModelTargetConfigDict', + 'ModelProviderServiceConfigModelTargetConfigParam', + 'ModelService', + 'ModelServiceConfig', + 'ModelServiceConfigDestinationConfig', + 'ModelServiceConfigDestinationConfigDestinationType', + 'ModelServiceConfigDestinationConfigDestinationTypeParam', + 'ModelServiceConfigDestinationConfigDict', + 'ModelServiceConfigDestinationConfigParam', + 'ModelServiceConfigDict', + 'ModelServiceConfigExternalModelConfig', + 'ModelServiceConfigExternalModelConfigDict', + 'ModelServiceConfigExternalModelConfigParam', + 'ModelServiceConfigFallbackConfig', + 'ModelServiceConfigFallbackConfigDict', + 'ModelServiceConfigFallbackConfigParam', + 'ModelServiceConfigParam', + 'ModelServiceConfigPayPerTokenConfig', + 'ModelServiceConfigPayPerTokenConfigDict', + 'ModelServiceConfigPayPerTokenConfigParam', + 'ModelServiceConfigProvisionedThroughputConfig', + 'ModelServiceConfigProvisionedThroughputConfigDict', + 'ModelServiceConfigProvisionedThroughputConfigParam', + 'ModelServiceConfigRoutingConfig', + 'ModelServiceConfigRoutingConfigDict', + 'ModelServiceConfigRoutingConfigParam', + 'ModelServiceDict', + 'ModelServiceParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'RateLimit', + 'RateLimitDict', + 'RateLimitParam', + 'RateLimitRateLimitKey', + 'RateLimitRateLimitKeyParam', + 'RateLimitRateLimitRenewalPeriod', + 'RateLimitRateLimitRenewalPeriodParam', ] -from databricks.bundles.model_services._models.inference_table_config import ( - InferenceTableConfig, - InferenceTableConfigDict, - InferenceTableConfigParam, -) -from databricks.bundles.model_services._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ( - ModelProviderServiceConfigModelTargetConfig, - ModelProviderServiceConfigModelTargetConfigDict, - ModelProviderServiceConfigModelTargetConfigParam, -) -from databricks.bundles.model_services._models.model_service import ( - ModelService, - ModelServiceDict, - ModelServiceParam, -) -from databricks.bundles.model_services._models.model_service_config import ( - ModelServiceConfig, - ModelServiceConfigDict, - ModelServiceConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_destination_config import ( - ModelServiceConfigDestinationConfig, - ModelServiceConfigDestinationConfigDict, - ModelServiceConfigDestinationConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ( - ModelServiceConfigDestinationConfigDestinationType, - ModelServiceConfigDestinationConfigDestinationTypeParam, -) -from databricks.bundles.model_services._models.model_service_config_external_model_config import ( - ModelServiceConfigExternalModelConfig, - ModelServiceConfigExternalModelConfigDict, - ModelServiceConfigExternalModelConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_fallback_config import ( - ModelServiceConfigFallbackConfig, - ModelServiceConfigFallbackConfigDict, - ModelServiceConfigFallbackConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ( - ModelServiceConfigPayPerTokenConfig, - ModelServiceConfigPayPerTokenConfigDict, - ModelServiceConfigPayPerTokenConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ( - ModelServiceConfigProvisionedThroughputConfig, - ModelServiceConfigProvisionedThroughputConfigDict, - ModelServiceConfigProvisionedThroughputConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_routing_config import ( - ModelServiceConfigRoutingConfig, - ModelServiceConfigRoutingConfigDict, - ModelServiceConfigRoutingConfigParam, -) -from databricks.bundles.model_services._models.privilege import ( - Privilege, - PrivilegeParam, -) -from databricks.bundles.model_services._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) -from databricks.bundles.model_services._models.rate_limit import ( - RateLimit, - RateLimitDict, - RateLimitParam, -) -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import ( - RateLimitRateLimitKey, - RateLimitRateLimitKeyParam, -) -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import ( - RateLimitRateLimitRenewalPeriod, - RateLimitRateLimitRenewalPeriodParam, -) +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + diff --git a/python/databricks/bundles/model_services/_models/inference_table_config.py b/python/databricks/bundles/model_services/_models/inference_table_config.py index 45165416e17..159ec1af624 100644 --- a/python/databricks/bundles/model_services/_models/inference_table_config.py +++ b/python/databricks/bundles/model_services/_models/inference_table_config.py @@ -1,16 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InferenceTableConfig: """ @@ -18,7 +40,6 @@ class InferenceTableConfig: inference table. When this configuration is present, payload logging is enabled by default. """ - parent: VariableOr[str] """ Parent Unity Catalog schema where the inference table is created, in the @@ -36,11 +57,12 @@ class InferenceTableConfig: """ @classmethod - def from_dict(cls, value: "InferenceTableConfigDict") -> "Self": + def from_dict(cls, value: 'InferenceTableConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InferenceTableConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InferenceTableConfigDict': + return _transform_to_json_value(self) # type:ignore + class InferenceTableConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/lifecycle.py b/python/databricks/bundles/model_services/_models/lifecycle.py index 697776a198e..f84b9e7e08b 100644 --- a/python/databricks/bundles/model_services/_models/lifecycle.py +++ b/python/databricks/bundles/model_services/_models/lifecycle.py @@ -1,16 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +43,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/model_provider_service_config_model_target_config.py b/python/databricks/bundles/model_services/_models/model_provider_service_config_model_target_config.py index 671c9eb4490..50a9d818ab6 100644 --- a/python/databricks/bundles/model_services/_models/model_provider_service_config_model_target_config.py +++ b/python/databricks/bundles/model_services/_models/model_provider_service_config_model_target_config.py @@ -1,27 +1,44 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelProviderServiceConfigModelTargetConfig: """ Model target configuration shared by model provider service targets and external model destinations on model services. """ - model: VariableOr[str] """ Provider-side model identifier, such as `gpt-5` or `claude-opus-4-7`. @@ -49,13 +66,12 @@ class ModelProviderServiceConfigModelTargetConfig: """ @classmethod - def from_dict( - cls, value: "ModelProviderServiceConfigModelTargetConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelProviderServiceConfigModelTargetConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelProviderServiceConfigModelTargetConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelProviderServiceConfigModelTargetConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelProviderServiceConfigModelTargetConfigDict(TypedDict, total=False): @@ -88,7 +104,4 @@ class ModelProviderServiceConfigModelTargetConfigDict(TypedDict, total=False): """ -ModelProviderServiceConfigModelTargetConfigParam = ( - ModelProviderServiceConfigModelTargetConfigDict - | ModelProviderServiceConfigModelTargetConfig -) +ModelProviderServiceConfigModelTargetConfigParam = ModelProviderServiceConfigModelTargetConfigDict | ModelProviderServiceConfigModelTargetConfig diff --git a/python/databricks/bundles/model_services/_models/model_service.py b/python/databricks/bundles/model_services/_models/model_service.py index 88f9d8cdeb7..f234c7746fd 100644 --- a/python/databricks/bundles/model_services/_models/model_service.py +++ b/python/databricks/bundles/model_services/_models/model_service.py @@ -1,33 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.model_services._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.model_services._models.model_service_config import ( - ModelServiceConfig, - ModelServiceConfigParam, -) -from databricks.bundles.model_services._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelService(Resource): """""" @@ -45,11 +50,12 @@ class ModelService(Resource): lifecycle: VariableOrOptional[Lifecycle] = None @classmethod - def from_dict(cls, value: "ModelServiceDict") -> "Self": + def from_dict(cls, value: 'ModelServiceDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServiceDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServiceDict': + return _transform_to_json_value(self) # type:ignore + class ModelServiceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/model_service_config.py b/python/databricks/bundles/model_services/_models/model_service_config.py index 88fb652d3ce..50c7c68458c 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config.py @@ -1,34 +1,43 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.model_services._models.inference_table_config import ( - InferenceTableConfig, - InferenceTableConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_routing_config import ( - ModelServiceConfigRoutingConfig, - ModelServiceConfigRoutingConfigParam, -) -from databricks.bundles.model_services._models.rate_limit import ( - RateLimit, - RateLimitParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelServiceConfig: """ Operational configuration wrapped around the ModelService resource. """ - inference_table: VariableOrOptional[InferenceTableConfig] = None """ Inference table configuration for payload logging. @@ -45,11 +54,12 @@ class ModelServiceConfig: """ @classmethod - def from_dict(cls, value: "ModelServiceConfigDict") -> "Self": + def from_dict(cls, value: 'ModelServiceConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServiceConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServiceConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelServiceConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/model_service_config_destination_config.py b/python/databricks/bundles/model_services/_models/model_service_config_destination_config.py index e4277476e5f..a18b409d074 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_destination_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_destination_config.py @@ -1,32 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ( - ModelServiceConfigDestinationConfigDestinationType, - ModelServiceConfigDestinationConfigDestinationTypeParam, -) -from databricks.bundles.model_services._models.model_service_config_external_model_config import ( - ModelServiceConfigExternalModelConfig, - ModelServiceConfigExternalModelConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ( - ModelServiceConfigPayPerTokenConfig, - ModelServiceConfigPayPerTokenConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ( - ModelServiceConfigProvisionedThroughputConfig, - ModelServiceConfigProvisionedThroughputConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelServiceConfigDestinationConfig: """ @@ -34,7 +40,6 @@ class ModelServiceConfigDestinationConfig: per-type configs inside `type_config` must be set, and it must match `destination_type`. """ - destination_type: VariableOr[ModelServiceConfigDestinationConfigDestinationType] """ Backing-model category. Provide the matching type-specific configuration @@ -46,9 +51,7 @@ class ModelServiceConfigDestinationConfig: User-facing label for this destination, used in routing references. """ - external_model_config: VariableOrOptional[ModelServiceConfigExternalModelConfig] = ( - None - ) + external_model_config: VariableOrOptional[ModelServiceConfigExternalModelConfig] = None """ Configuration for an external model reached through a model provider service. """ @@ -58,9 +61,7 @@ class ModelServiceConfigDestinationConfig: Configuration for a pay-per-token Databricks foundation model. """ - provisioned_throughput_config: VariableOrOptional[ - ModelServiceConfigProvisionedThroughputConfig - ] = None + provisioned_throughput_config: VariableOrOptional[ModelServiceConfigProvisionedThroughputConfig] = None """ Configuration for a provisioned-throughput Databricks foundation model. """ @@ -74,19 +75,18 @@ class ModelServiceConfigDestinationConfig: """ @classmethod - def from_dict(cls, value: "ModelServiceConfigDestinationConfigDict") -> "Self": + def from_dict(cls, value: 'ModelServiceConfigDestinationConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServiceConfigDestinationConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServiceConfigDestinationConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelServiceConfigDestinationConfigDict(TypedDict, total=False): """""" - destination_type: VariableOr[ - ModelServiceConfigDestinationConfigDestinationTypeParam - ] + destination_type: VariableOr[ModelServiceConfigDestinationConfigDestinationTypeParam] """ Backing-model category. Provide the matching type-specific configuration and leave the other type-specific configurations unset. @@ -97,9 +97,7 @@ class ModelServiceConfigDestinationConfigDict(TypedDict, total=False): User-facing label for this destination, used in routing references. """ - external_model_config: VariableOrOptional[ - ModelServiceConfigExternalModelConfigParam - ] + external_model_config: VariableOrOptional[ModelServiceConfigExternalModelConfigParam] """ Configuration for an external model reached through a model provider service. """ @@ -109,9 +107,7 @@ class ModelServiceConfigDestinationConfigDict(TypedDict, total=False): Configuration for a pay-per-token Databricks foundation model. """ - provisioned_throughput_config: VariableOrOptional[ - ModelServiceConfigProvisionedThroughputConfigParam - ] + provisioned_throughput_config: VariableOrOptional[ModelServiceConfigProvisionedThroughputConfigParam] """ Configuration for a provisioned-throughput Databricks foundation model. """ @@ -125,6 +121,4 @@ class ModelServiceConfigDestinationConfigDict(TypedDict, total=False): """ -ModelServiceConfigDestinationConfigParam = ( - ModelServiceConfigDestinationConfigDict | ModelServiceConfigDestinationConfig -) +ModelServiceConfigDestinationConfigParam = ModelServiceConfigDestinationConfigDict | ModelServiceConfigDestinationConfig diff --git a/python/databricks/bundles/model_services/_models/model_service_config_destination_config_destination_type.py b/python/databricks/bundles/model_services/_models/model_service_config_destination_config_destination_type.py index a98c2ca8161..96aacf91f80 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_destination_config_destination_type.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_destination_config_destination_type.py @@ -1,30 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam class ModelServiceConfigDestinationConfigDestinationType(Enum): """ Backing-model category for a model service destination. """ + DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL = "DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL" + DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL = "DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL" + DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL = "DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL" - DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL = ( - "DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL" - ) - DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL = ( - "DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL" - ) - DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL = ( - "DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL" - ) - - -ModelServiceConfigDestinationConfigDestinationTypeParam = ( - Literal[ - "DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL", - "DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL", - "DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL", - ] - | ModelServiceConfigDestinationConfigDestinationType -) +ModelServiceConfigDestinationConfigDestinationTypeParam = Literal["DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL", "DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL", "DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL"] | ModelServiceConfigDestinationConfigDestinationType diff --git a/python/databricks/bundles/model_services/_models/model_service_config_external_model_config.py b/python/databricks/bundles/model_services/_models/model_service_config_external_model_config.py index d82c7ad7d19..9d2ea2b3ed7 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_external_model_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_external_model_config.py @@ -1,20 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ( - ModelProviderServiceConfigModelTargetConfig, - ModelProviderServiceConfigModelTargetConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelServiceConfigExternalModelConfig: """ @@ -23,7 +41,6 @@ class ModelServiceConfigExternalModelConfig: ModelProviderService entity referenced via `model_provider_service`; the platform resolves the provider at invocation time. """ - model_provider_service: VariableOr[str] """ Resource name of the governed ModelProviderService that owns provider @@ -42,11 +59,12 @@ class ModelServiceConfigExternalModelConfig: """ @classmethod - def from_dict(cls, value: "ModelServiceConfigExternalModelConfigDict") -> "Self": + def from_dict(cls, value: 'ModelServiceConfigExternalModelConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServiceConfigExternalModelConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServiceConfigExternalModelConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelServiceConfigExternalModelConfigDict(TypedDict, total=False): @@ -70,6 +88,4 @@ class ModelServiceConfigExternalModelConfigDict(TypedDict, total=False): """ -ModelServiceConfigExternalModelConfigParam = ( - ModelServiceConfigExternalModelConfigDict | ModelServiceConfigExternalModelConfig -) +ModelServiceConfigExternalModelConfigParam = ModelServiceConfigExternalModelConfigDict | ModelServiceConfigExternalModelConfig diff --git a/python/databricks/bundles/model_services/_models/model_service_config_fallback_config.py b/python/databricks/bundles/model_services/_models/model_service_config_fallback_config.py index e450152f88d..70e65e6b069 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_fallback_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_fallback_config.py @@ -1,40 +1,56 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList -from databricks.bundles.model_services._models.model_service_config_destination_config import ( - ModelServiceConfigDestinationConfig, - ModelServiceConfigDestinationConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelServiceConfigFallbackConfig: """ Fallback routing applied after a primary destination fails. Fallback destinations are tried in the listed order. """ - - destinations: VariableOrList[ModelServiceConfigDestinationConfig] = field( - default_factory=list - ) + destinations: VariableOrList[ModelServiceConfigDestinationConfig] = field(default_factory=list) """ Fallback destinations, tried in the listed order. At most 5 are allowed. """ @classmethod - def from_dict(cls, value: "ModelServiceConfigFallbackConfigDict") -> "Self": + def from_dict(cls, value: 'ModelServiceConfigFallbackConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServiceConfigFallbackConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServiceConfigFallbackConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelServiceConfigFallbackConfigDict(TypedDict, total=False): @@ -46,6 +62,4 @@ class ModelServiceConfigFallbackConfigDict(TypedDict, total=False): """ -ModelServiceConfigFallbackConfigParam = ( - ModelServiceConfigFallbackConfigDict | ModelServiceConfigFallbackConfig -) +ModelServiceConfigFallbackConfigParam = ModelServiceConfigFallbackConfigDict | ModelServiceConfigFallbackConfig diff --git a/python/databricks/bundles/model_services/_models/model_service_config_pay_per_token_config.py b/python/databricks/bundles/model_services/_models/model_service_config_pay_per_token_config.py index a42bbed4465..4e4062cd410 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_pay_per_token_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_pay_per_token_config.py @@ -1,16 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelServiceConfigPayPerTokenConfig: """ @@ -18,7 +40,6 @@ class ModelServiceConfigPayPerTokenConfig: the foundation model by its UC resource name; the platform resolves it to a Model Serving endpoint at request time. """ - model: VariableOr[str] """ Resource name of the Unity Catalog model. @@ -26,11 +47,12 @@ class ModelServiceConfigPayPerTokenConfig: """ @classmethod - def from_dict(cls, value: "ModelServiceConfigPayPerTokenConfigDict") -> "Self": + def from_dict(cls, value: 'ModelServiceConfigPayPerTokenConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServiceConfigPayPerTokenConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServiceConfigPayPerTokenConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelServiceConfigPayPerTokenConfigDict(TypedDict, total=False): @@ -43,6 +65,4 @@ class ModelServiceConfigPayPerTokenConfigDict(TypedDict, total=False): """ -ModelServiceConfigPayPerTokenConfigParam = ( - ModelServiceConfigPayPerTokenConfigDict | ModelServiceConfigPayPerTokenConfig -) +ModelServiceConfigPayPerTokenConfigParam = ModelServiceConfigPayPerTokenConfigDict | ModelServiceConfigPayPerTokenConfig diff --git a/python/databricks/bundles/model_services/_models/model_service_config_provisioned_throughput_config.py b/python/databricks/bundles/model_services/_models/model_service_config_provisioned_throughput_config.py index d8a0c63cc3d..ad88b18c112 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_provisioned_throughput_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_provisioned_throughput_config.py @@ -1,16 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelServiceConfigProvisionedThroughputConfig: """ @@ -19,7 +41,6 @@ class ModelServiceConfigProvisionedThroughputConfig: sizing (provisioned throughput, burst scaling, model version) is owned by the Model Serving endpoint itself, not by this message. """ - model_serving_endpoint: VariableOr[str] """ Name of the backing Model Serving endpoint serving the provisioned- @@ -31,13 +52,12 @@ class ModelServiceConfigProvisionedThroughputConfig: """ @classmethod - def from_dict( - cls, value: "ModelServiceConfigProvisionedThroughputConfigDict" - ) -> "Self": + def from_dict(cls, value: 'ModelServiceConfigProvisionedThroughputConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServiceConfigProvisionedThroughputConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServiceConfigProvisionedThroughputConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelServiceConfigProvisionedThroughputConfigDict(TypedDict, total=False): @@ -54,7 +74,4 @@ class ModelServiceConfigProvisionedThroughputConfigDict(TypedDict, total=False): """ -ModelServiceConfigProvisionedThroughputConfigParam = ( - ModelServiceConfigProvisionedThroughputConfigDict - | ModelServiceConfigProvisionedThroughputConfig -) +ModelServiceConfigProvisionedThroughputConfigParam = ModelServiceConfigProvisionedThroughputConfigDict | ModelServiceConfigProvisionedThroughputConfig diff --git a/python/databricks/bundles/model_services/_models/model_service_config_routing_config.py b/python/databricks/bundles/model_services/_models/model_service_config_routing_config.py index 36b547f806f..6054d43e882 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_routing_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_routing_config.py @@ -1,34 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.model_services._models.model_service_config_destination_config import ( - ModelServiceConfigDestinationConfig, - ModelServiceConfigDestinationConfigParam, -) -from databricks.bundles.model_services._models.model_service_config_fallback_config import ( - ModelServiceConfigFallbackConfig, - ModelServiceConfigFallbackConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelServiceConfigRoutingConfig: """ Routing configuration for a model service, nesting destinations and fallback under a single sub-message. """ - - destinations: VariableOrList[ModelServiceConfigDestinationConfig] = field( - default_factory=list - ) + destinations: VariableOrList[ModelServiceConfigDestinationConfig] = field(default_factory=list) """ Primary routing destinations. At most 10 are allowed. At least one is required on Create. On Update, provide this list when replacing the full @@ -44,11 +55,12 @@ class ModelServiceConfigRoutingConfig: """ @classmethod - def from_dict(cls, value: "ModelServiceConfigRoutingConfigDict") -> "Self": + def from_dict(cls, value: 'ModelServiceConfigRoutingConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServiceConfigRoutingConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServiceConfigRoutingConfigDict': + return _transform_to_json_value(self) # type:ignore + class ModelServiceConfigRoutingConfigDict(TypedDict, total=False): @@ -70,6 +82,4 @@ class ModelServiceConfigRoutingConfigDict(TypedDict, total=False): """ -ModelServiceConfigRoutingConfigParam = ( - ModelServiceConfigRoutingConfigDict | ModelServiceConfigRoutingConfig -) +ModelServiceConfigRoutingConfigParam = ModelServiceConfigRoutingConfigDict | ModelServiceConfigRoutingConfig diff --git a/python/databricks/bundles/model_services/_models/privilege.py b/python/databricks/bundles/model_services/_models/privilege.py index df45211a76a..a51483854b2 100644 --- a/python/databricks/bundles/model_services/_models/privilege.py +++ b/python/databricks/bundles/model_services/_models/privilege.py @@ -1,7 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam class Privilege(Enum): @@ -58,61 +83,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/model_services/_models/privilege_assignment.py b/python/databricks/bundles/model_services/_models/privilege_assignment.py index 7bbd0ac80ee..5f91055ff3a 100644 --- a/python/databricks/bundles/model_services/_models/privilege_assignment.py +++ b/python/databricks/bundles/model_services/_models/privilege_assignment.py @@ -1,20 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.model_services._models.privilege import ( - Privilege, - PrivilegeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -31,11 +49,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/rate_limit.py b/python/databricks/bundles/model_services/_models/rate_limit.py index 355414bc28c..9f6b7feb8bb 100644 --- a/python/databricks/bundles/model_services/_models/rate_limit.py +++ b/python/databricks/bundles/model_services/_models/rate_limit.py @@ -1,24 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import ( - RateLimitRateLimitKey, - RateLimitRateLimitKeyParam, -) -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import ( - RateLimitRateLimitRenewalPeriod, - RateLimitRateLimitRenewalPeriodParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RateLimit: """ @@ -26,7 +40,6 @@ class RateLimit: unset to impose no limit on that dimension; set a value to cap that dimension within the renewal period. """ - key: VariableOr[RateLimitRateLimitKey] """ Scope of the rate limit. Depending on this value, the limit applies to a @@ -58,11 +71,12 @@ class RateLimit: """ @classmethod - def from_dict(cls, value: "RateLimitDict") -> "Self": + def from_dict(cls, value: 'RateLimitDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RateLimitDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RateLimitDict': + return _transform_to_json_value(self) # type:ignore + class RateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_key.py b/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_key.py index fb907da810f..fef5d8b91ac 100644 --- a/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_key.py +++ b/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_key.py @@ -1,28 +1,42 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam class RateLimitRateLimitKey(Enum): """ Scope key for a rate limit. """ - RATE_LIMIT_KEY_USER = "RATE_LIMIT_KEY_USER" RATE_LIMIT_KEY_USER_GROUP = "RATE_LIMIT_KEY_USER_GROUP" RATE_LIMIT_KEY_SERVICE_PRINCIPAL = "RATE_LIMIT_KEY_SERVICE_PRINCIPAL" RATE_LIMIT_KEY_SERVICE = "RATE_LIMIT_KEY_SERVICE" RATE_LIMIT_KEY_USER_DEFAULT = "RATE_LIMIT_KEY_USER_DEFAULT" - -RateLimitRateLimitKeyParam = ( - Literal[ - "RATE_LIMIT_KEY_USER", - "RATE_LIMIT_KEY_USER_GROUP", - "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", - "RATE_LIMIT_KEY_SERVICE", - "RATE_LIMIT_KEY_USER_DEFAULT", - ] - | RateLimitRateLimitKey -) +RateLimitRateLimitKeyParam = Literal["RATE_LIMIT_KEY_USER", "RATE_LIMIT_KEY_USER_GROUP", "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", "RATE_LIMIT_KEY_SERVICE", "RATE_LIMIT_KEY_USER_DEFAULT"] | RateLimitRateLimitKey diff --git a/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_renewal_period.py b/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_renewal_period.py index 58636c784c0..f4fbc895749 100644 --- a/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_renewal_period.py +++ b/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_renewal_period.py @@ -1,19 +1,39 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam +from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam +from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam +from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam +from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam + +from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam class RateLimitRateLimitRenewalPeriod(Enum): """ Renewal period for a rate limit. """ - RATE_LIMIT_RENEWAL_PERIOD_MINUTE = "RATE_LIMIT_RENEWAL_PERIOD_MINUTE" RATE_LIMIT_RENEWAL_PERIOD_HOUR = "RATE_LIMIT_RENEWAL_PERIOD_HOUR" - -RateLimitRateLimitRenewalPeriodParam = ( - Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] - | RateLimitRateLimitRenewalPeriod -) +RateLimitRateLimitRenewalPeriodParam = Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] | RateLimitRateLimitRenewalPeriod diff --git a/python/databricks/bundles/model_serving_endpoints/__init__.py b/python/databricks/bundles/model_serving_endpoints/__init__.py index 0137c5e76b9..c9be1165854 100644 --- a/python/databricks/bundles/model_serving_endpoints/__init__.py +++ b/python/databricks/bundles/model_serving_endpoints/__init__.py @@ -1,352 +1,190 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Ai21LabsConfig", - "Ai21LabsConfigDict", - "Ai21LabsConfigParam", - "AiGatewayConfig", - "AiGatewayConfigDict", - "AiGatewayConfigParam", - "AiGatewayGuardrailParameters", - "AiGatewayGuardrailParametersDict", - "AiGatewayGuardrailParametersParam", - "AiGatewayGuardrailPiiBehavior", - "AiGatewayGuardrailPiiBehaviorBehavior", - "AiGatewayGuardrailPiiBehaviorBehaviorParam", - "AiGatewayGuardrailPiiBehaviorDict", - "AiGatewayGuardrailPiiBehaviorParam", - "AiGatewayGuardrails", - "AiGatewayGuardrailsDict", - "AiGatewayGuardrailsParam", - "AiGatewayInferenceTableConfig", - "AiGatewayInferenceTableConfigDict", - "AiGatewayInferenceTableConfigParam", - "AiGatewayRateLimit", - "AiGatewayRateLimitDict", - "AiGatewayRateLimitKey", - "AiGatewayRateLimitKeyParam", - "AiGatewayRateLimitParam", - "AiGatewayRateLimitRenewalPeriod", - "AiGatewayRateLimitRenewalPeriodParam", - "AiGatewayUsageTrackingConfig", - "AiGatewayUsageTrackingConfigDict", - "AiGatewayUsageTrackingConfigParam", - "AmazonBedrockConfig", - "AmazonBedrockConfigBedrockProvider", - "AmazonBedrockConfigBedrockProviderParam", - "AmazonBedrockConfigDict", - "AmazonBedrockConfigParam", - "AnthropicConfig", - "AnthropicConfigDict", - "AnthropicConfigParam", - "ApiKeyAuth", - "ApiKeyAuthDict", - "ApiKeyAuthParam", - "AutoCaptureConfigInput", - "AutoCaptureConfigInputDict", - "AutoCaptureConfigInputParam", - "BearerTokenAuth", - "BearerTokenAuthDict", - "BearerTokenAuthParam", - "CohereConfig", - "CohereConfigDict", - "CohereConfigParam", - "CustomProviderConfig", - "CustomProviderConfigDict", - "CustomProviderConfigParam", - "DatabricksModelServingConfig", - "DatabricksModelServingConfigDict", - "DatabricksModelServingConfigParam", - "EmailNotifications", - "EmailNotificationsDict", - "EmailNotificationsParam", - "EndpointCoreConfigInput", - "EndpointCoreConfigInputDict", - "EndpointCoreConfigInputParam", - "EndpointTag", - "EndpointTagDict", - "EndpointTagParam", - "ExternalModel", - "ExternalModelDict", - "ExternalModelParam", - "ExternalModelProvider", - "ExternalModelProviderParam", - "FallbackConfig", - "FallbackConfigDict", - "FallbackConfigParam", - "GoogleCloudVertexAiConfig", - "GoogleCloudVertexAiConfigDict", - "GoogleCloudVertexAiConfigParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "ModelServingEndpoint", - "ModelServingEndpointDict", - "ModelServingEndpointParam", - "ModelServingEndpointPermission", - "ModelServingEndpointPermissionDict", - "ModelServingEndpointPermissionParam", - "OpenAiConfig", - "OpenAiConfigDict", - "OpenAiConfigParam", - "PaLmConfig", - "PaLmConfigDict", - "PaLmConfigParam", - "RateLimit", - "RateLimitDict", - "RateLimitKey", - "RateLimitKeyParam", - "RateLimitParam", - "RateLimitRenewalPeriod", - "RateLimitRenewalPeriodParam", - "Route", - "RouteDict", - "RouteParam", - "ServedEntityInput", - "ServedEntityInputDict", - "ServedEntityInputParam", - "ServedModelInput", - "ServedModelInputDict", - "ServedModelInputParam", - "ServedModelInputWorkloadType", - "ServedModelInputWorkloadTypeParam", - "ServingEndpointPermissionLevel", - "ServingEndpointPermissionLevelParam", - "ServingModelWorkloadType", - "ServingModelWorkloadTypeParam", - "TelemetryConfig", - "TelemetryConfigDict", - "TelemetryConfigParam", - "TelemetryFeature", - "TelemetryFeatureParam", - "TelemetryInferenceTableConfig", - "TelemetryInferenceTableConfigDict", - "TelemetryInferenceTableConfigParam", - "TrafficConfig", - "TrafficConfigDict", - "TrafficConfigParam", - "UnityCatalogTableNames", - "UnityCatalogTableNamesDict", - "UnityCatalogTableNamesParam", + 'Ai21LabsConfig', + 'Ai21LabsConfigDict', + 'Ai21LabsConfigParam', + 'AiGatewayConfig', + 'AiGatewayConfigDict', + 'AiGatewayConfigParam', + 'AiGatewayGuardrailParameters', + 'AiGatewayGuardrailParametersDict', + 'AiGatewayGuardrailParametersParam', + 'AiGatewayGuardrailPiiBehavior', + 'AiGatewayGuardrailPiiBehaviorBehavior', + 'AiGatewayGuardrailPiiBehaviorBehaviorParam', + 'AiGatewayGuardrailPiiBehaviorDict', + 'AiGatewayGuardrailPiiBehaviorParam', + 'AiGatewayGuardrails', + 'AiGatewayGuardrailsDict', + 'AiGatewayGuardrailsParam', + 'AiGatewayInferenceTableConfig', + 'AiGatewayInferenceTableConfigDict', + 'AiGatewayInferenceTableConfigParam', + 'AiGatewayRateLimit', + 'AiGatewayRateLimitDict', + 'AiGatewayRateLimitKey', + 'AiGatewayRateLimitKeyParam', + 'AiGatewayRateLimitParam', + 'AiGatewayRateLimitRenewalPeriod', + 'AiGatewayRateLimitRenewalPeriodParam', + 'AiGatewayUsageTrackingConfig', + 'AiGatewayUsageTrackingConfigDict', + 'AiGatewayUsageTrackingConfigParam', + 'AmazonBedrockConfig', + 'AmazonBedrockConfigBedrockProvider', + 'AmazonBedrockConfigBedrockProviderParam', + 'AmazonBedrockConfigDict', + 'AmazonBedrockConfigParam', + 'AnthropicConfig', + 'AnthropicConfigDict', + 'AnthropicConfigParam', + 'ApiKeyAuth', + 'ApiKeyAuthDict', + 'ApiKeyAuthParam', + 'AutoCaptureConfigInput', + 'AutoCaptureConfigInputDict', + 'AutoCaptureConfigInputParam', + 'BearerTokenAuth', + 'BearerTokenAuthDict', + 'BearerTokenAuthParam', + 'CohereConfig', + 'CohereConfigDict', + 'CohereConfigParam', + 'CustomProviderConfig', + 'CustomProviderConfigDict', + 'CustomProviderConfigParam', + 'DatabricksModelServingConfig', + 'DatabricksModelServingConfigDict', + 'DatabricksModelServingConfigParam', + 'EmailNotifications', + 'EmailNotificationsDict', + 'EmailNotificationsParam', + 'EndpointCoreConfigInput', + 'EndpointCoreConfigInputDict', + 'EndpointCoreConfigInputParam', + 'EndpointTag', + 'EndpointTagDict', + 'EndpointTagParam', + 'ExternalModel', + 'ExternalModelDict', + 'ExternalModelParam', + 'ExternalModelProvider', + 'ExternalModelProviderParam', + 'FallbackConfig', + 'FallbackConfigDict', + 'FallbackConfigParam', + 'GoogleCloudVertexAiConfig', + 'GoogleCloudVertexAiConfigDict', + 'GoogleCloudVertexAiConfigParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'ModelServingEndpoint', + 'ModelServingEndpointDict', + 'ModelServingEndpointParam', + 'ModelServingEndpointPermission', + 'ModelServingEndpointPermissionDict', + 'ModelServingEndpointPermissionParam', + 'OpenAiConfig', + 'OpenAiConfigDict', + 'OpenAiConfigParam', + 'PaLmConfig', + 'PaLmConfigDict', + 'PaLmConfigParam', + 'RateLimit', + 'RateLimitDict', + 'RateLimitKey', + 'RateLimitKeyParam', + 'RateLimitParam', + 'RateLimitRenewalPeriod', + 'RateLimitRenewalPeriodParam', + 'Route', + 'RouteDict', + 'RouteParam', + 'ServedEntityInput', + 'ServedEntityInputDict', + 'ServedEntityInputParam', + 'ServedModelInput', + 'ServedModelInputDict', + 'ServedModelInputParam', + 'ServedModelInputWorkloadType', + 'ServedModelInputWorkloadTypeParam', + 'ServingEndpointPermissionLevel', + 'ServingEndpointPermissionLevelParam', + 'ServingModelWorkloadType', + 'ServingModelWorkloadTypeParam', + 'TelemetryConfig', + 'TelemetryConfigDict', + 'TelemetryConfigParam', + 'TelemetryFeature', + 'TelemetryFeatureParam', + 'TelemetryInferenceTableConfig', + 'TelemetryInferenceTableConfigDict', + 'TelemetryInferenceTableConfigParam', + 'TrafficConfig', + 'TrafficConfigDict', + 'TrafficConfigParam', + 'UnityCatalogTableNames', + 'UnityCatalogTableNamesDict', + 'UnityCatalogTableNamesParam', ] -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import ( - Ai21LabsConfig, - Ai21LabsConfigDict, - Ai21LabsConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import ( - AiGatewayConfig, - AiGatewayConfigDict, - AiGatewayConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import ( - AiGatewayGuardrailParameters, - AiGatewayGuardrailParametersDict, - AiGatewayGuardrailParametersParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import ( - AiGatewayGuardrailPiiBehavior, - AiGatewayGuardrailPiiBehaviorDict, - AiGatewayGuardrailPiiBehaviorParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import ( - AiGatewayGuardrailPiiBehaviorBehavior, - AiGatewayGuardrailPiiBehaviorBehaviorParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import ( - AiGatewayGuardrails, - AiGatewayGuardrailsDict, - AiGatewayGuardrailsParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import ( - AiGatewayInferenceTableConfig, - AiGatewayInferenceTableConfigDict, - AiGatewayInferenceTableConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import ( - AiGatewayRateLimit, - AiGatewayRateLimitDict, - AiGatewayRateLimitParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import ( - AiGatewayRateLimitKey, - AiGatewayRateLimitKeyParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import ( - AiGatewayRateLimitRenewalPeriod, - AiGatewayRateLimitRenewalPeriodParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import ( - AiGatewayUsageTrackingConfig, - AiGatewayUsageTrackingConfigDict, - AiGatewayUsageTrackingConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import ( - AmazonBedrockConfig, - AmazonBedrockConfigDict, - AmazonBedrockConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import ( - AmazonBedrockConfigBedrockProvider, - AmazonBedrockConfigBedrockProviderParam, -) -from databricks.bundles.model_serving_endpoints._models.anthropic_config import ( - AnthropicConfig, - AnthropicConfigDict, - AnthropicConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ( - ApiKeyAuth, - ApiKeyAuthDict, - ApiKeyAuthParam, -) -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import ( - AutoCaptureConfigInput, - AutoCaptureConfigInputDict, - AutoCaptureConfigInputParam, -) -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import ( - BearerTokenAuth, - BearerTokenAuthDict, - BearerTokenAuthParam, -) -from databricks.bundles.model_serving_endpoints._models.cohere_config import ( - CohereConfig, - CohereConfigDict, - CohereConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import ( - CustomProviderConfig, - CustomProviderConfigDict, - CustomProviderConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import ( - DatabricksModelServingConfig, - DatabricksModelServingConfigDict, - DatabricksModelServingConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.email_notifications import ( - EmailNotifications, - EmailNotificationsDict, - EmailNotificationsParam, -) -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import ( - EndpointCoreConfigInput, - EndpointCoreConfigInputDict, - EndpointCoreConfigInputParam, -) -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import ( - EndpointTag, - EndpointTagDict, - EndpointTagParam, -) -from databricks.bundles.model_serving_endpoints._models.external_model import ( - ExternalModel, - ExternalModelDict, - ExternalModelParam, -) -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ( - ExternalModelProvider, - ExternalModelProviderParam, -) -from databricks.bundles.model_serving_endpoints._models.fallback_config import ( - FallbackConfig, - FallbackConfigDict, - FallbackConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import ( - GoogleCloudVertexAiConfig, - GoogleCloudVertexAiConfigDict, - GoogleCloudVertexAiConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( - ModelServingEndpoint, - ModelServingEndpointDict, - ModelServingEndpointParam, -) -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ( - ModelServingEndpointPermission, - ModelServingEndpointPermissionDict, - ModelServingEndpointPermissionParam, -) -from databricks.bundles.model_serving_endpoints._models.open_ai_config import ( - OpenAiConfig, - OpenAiConfigDict, - OpenAiConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import ( - PaLmConfig, - PaLmConfigDict, - PaLmConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.rate_limit import ( - RateLimit, - RateLimitDict, - RateLimitParam, -) -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import ( - RateLimitKey, - RateLimitKeyParam, -) -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import ( - RateLimitRenewalPeriod, - RateLimitRenewalPeriodParam, -) -from databricks.bundles.model_serving_endpoints._models.route import ( - Route, - RouteDict, - RouteParam, -) -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ( - ServedEntityInput, - ServedEntityInputDict, - ServedEntityInputParam, -) -from databricks.bundles.model_serving_endpoints._models.served_model_input import ( - ServedModelInput, - ServedModelInputDict, - ServedModelInputParam, -) -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ( - ServedModelInputWorkloadType, - ServedModelInputWorkloadTypeParam, -) -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ( - ServingEndpointPermissionLevel, - ServingEndpointPermissionLevelParam, -) -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ( - ServingModelWorkloadType, - ServingModelWorkloadTypeParam, -) -from databricks.bundles.model_serving_endpoints._models.telemetry_config import ( - TelemetryConfig, - TelemetryConfigDict, - TelemetryConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import ( - TelemetryFeature, - TelemetryFeatureParam, -) -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import ( - TelemetryInferenceTableConfig, - TelemetryInferenceTableConfigDict, - TelemetryInferenceTableConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.traffic_config import ( - TrafficConfig, - TrafficConfigDict, - TrafficConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import ( - UnityCatalogTableNames, - UnityCatalogTableNamesDict, - UnityCatalogTableNamesParam, -) +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai21_labs_config.py b/python/databricks/bundles/model_serving_endpoints/_models/ai21_labs_config.py index af06b6f6f1e..46afde31118 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai21_labs_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai21_labs_config.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Ai21LabsConfig: """""" @@ -32,11 +90,12 @@ class Ai21LabsConfig: """ @classmethod - def from_dict(cls, value: "Ai21LabsConfigDict") -> "Self": + def from_dict(cls, value: 'Ai21LabsConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "Ai21LabsConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'Ai21LabsConfigDict': + return _transform_to_json_value(self) # type:ignore + class Ai21LabsConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_config.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_config.py index 24a4a967599..9c1a3e23361 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_config.py @@ -1,36 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import ( - AiGatewayGuardrails, - AiGatewayGuardrailsParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import ( - AiGatewayInferenceTableConfig, - AiGatewayInferenceTableConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import ( - AiGatewayRateLimit, - AiGatewayRateLimitParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import ( - AiGatewayUsageTrackingConfig, - AiGatewayUsageTrackingConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.fallback_config import ( - FallbackConfig, - FallbackConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AiGatewayConfig: """""" @@ -64,11 +102,12 @@ class AiGatewayConfig: """ @classmethod - def from_dict(cls, value: "AiGatewayConfigDict") -> "Self": + def from_dict(cls, value: 'AiGatewayConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AiGatewayConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AiGatewayConfigDict': + return _transform_to_json_value(self) # type:ignore + class AiGatewayConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_parameters.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_parameters.py index 55b70a82669..82b90be32b2 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_parameters.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_parameters.py @@ -1,20 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import ( - AiGatewayGuardrailPiiBehavior, - AiGatewayGuardrailPiiBehaviorParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AiGatewayGuardrailParameters: """""" @@ -42,11 +96,12 @@ class AiGatewayGuardrailParameters: """ @classmethod - def from_dict(cls, value: "AiGatewayGuardrailParametersDict") -> "Self": + def from_dict(cls, value: 'AiGatewayGuardrailParametersDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AiGatewayGuardrailParametersDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AiGatewayGuardrailParametersDict': + return _transform_to_json_value(self) # type:ignore + class AiGatewayGuardrailParametersDict(TypedDict, total=False): @@ -75,6 +130,4 @@ class AiGatewayGuardrailParametersDict(TypedDict, total=False): """ -AiGatewayGuardrailParametersParam = ( - AiGatewayGuardrailParametersDict | AiGatewayGuardrailParameters -) +AiGatewayGuardrailParametersParam = AiGatewayGuardrailParametersDict | AiGatewayGuardrailParameters diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior.py index 577af194a55..4d34c2ac3ac 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior.py @@ -1,20 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import ( - AiGatewayGuardrailPiiBehaviorBehavior, - AiGatewayGuardrailPiiBehaviorBehaviorParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AiGatewayGuardrailPiiBehavior: """""" @@ -25,11 +79,12 @@ class AiGatewayGuardrailPiiBehavior: """ @classmethod - def from_dict(cls, value: "AiGatewayGuardrailPiiBehaviorDict") -> "Self": + def from_dict(cls, value: 'AiGatewayGuardrailPiiBehaviorDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AiGatewayGuardrailPiiBehaviorDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AiGatewayGuardrailPiiBehaviorDict': + return _transform_to_json_value(self) # type:ignore + class AiGatewayGuardrailPiiBehaviorDict(TypedDict, total=False): @@ -41,6 +96,4 @@ class AiGatewayGuardrailPiiBehaviorDict(TypedDict, total=False): """ -AiGatewayGuardrailPiiBehaviorParam = ( - AiGatewayGuardrailPiiBehaviorDict | AiGatewayGuardrailPiiBehavior -) +AiGatewayGuardrailPiiBehaviorParam = AiGatewayGuardrailPiiBehaviorDict | AiGatewayGuardrailPiiBehavior diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior_behavior.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior_behavior.py index bf9bf86fc68..f93de2dc112 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior_behavior.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior_behavior.py @@ -1,7 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class AiGatewayGuardrailPiiBehaviorBehavior(Enum): @@ -9,7 +70,4 @@ class AiGatewayGuardrailPiiBehaviorBehavior(Enum): BLOCK = "BLOCK" MASK = "MASK" - -AiGatewayGuardrailPiiBehaviorBehaviorParam = ( - Literal["NONE", "BLOCK", "MASK"] | AiGatewayGuardrailPiiBehaviorBehavior -) +AiGatewayGuardrailPiiBehaviorBehaviorParam = Literal["NONE", "BLOCK", "MASK"] | AiGatewayGuardrailPiiBehaviorBehavior diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrails.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrails.py index 299859aa015..c601b74f99b 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrails.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrails.py @@ -1,20 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import ( - AiGatewayGuardrailParameters, - AiGatewayGuardrailParametersParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AiGatewayGuardrails: """""" @@ -30,11 +84,12 @@ class AiGatewayGuardrails: """ @classmethod - def from_dict(cls, value: "AiGatewayGuardrailsDict") -> "Self": + def from_dict(cls, value: 'AiGatewayGuardrailsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AiGatewayGuardrailsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AiGatewayGuardrailsDict': + return _transform_to_json_value(self) # type:ignore + class AiGatewayGuardrailsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_inference_table_config.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_inference_table_config.py index a4d748dee3a..fc51a362abc 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_inference_table_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_inference_table_config.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AiGatewayInferenceTableConfig: """""" @@ -39,11 +97,12 @@ class AiGatewayInferenceTableConfig: """ @classmethod - def from_dict(cls, value: "AiGatewayInferenceTableConfigDict") -> "Self": + def from_dict(cls, value: 'AiGatewayInferenceTableConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AiGatewayInferenceTableConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AiGatewayInferenceTableConfigDict': + return _transform_to_json_value(self) # type:ignore + class AiGatewayInferenceTableConfigDict(TypedDict, total=False): @@ -73,6 +132,4 @@ class AiGatewayInferenceTableConfigDict(TypedDict, total=False): """ -AiGatewayInferenceTableConfigParam = ( - AiGatewayInferenceTableConfigDict | AiGatewayInferenceTableConfig -) +AiGatewayInferenceTableConfigParam = AiGatewayInferenceTableConfigDict | AiGatewayInferenceTableConfig diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit.py index 5848e04f895..089e3ad35c6 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit.py @@ -1,24 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import ( - AiGatewayRateLimitKey, - AiGatewayRateLimitKeyParam, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import ( - AiGatewayRateLimitRenewalPeriod, - AiGatewayRateLimitRenewalPeriodParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AiGatewayRateLimit: """""" @@ -50,11 +100,12 @@ class AiGatewayRateLimit: """ @classmethod - def from_dict(cls, value: "AiGatewayRateLimitDict") -> "Self": + def from_dict(cls, value: 'AiGatewayRateLimitDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AiGatewayRateLimitDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AiGatewayRateLimitDict': + return _transform_to_json_value(self) # type:ignore + class AiGatewayRateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_key.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_key.py index f166207b7e7..2b81fda6005 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_key.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_key.py @@ -1,7 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class AiGatewayRateLimitKey(Enum): @@ -10,8 +71,4 @@ class AiGatewayRateLimitKey(Enum): USER_GROUP = "user_group" SERVICE_PRINCIPAL = "service_principal" - -AiGatewayRateLimitKeyParam = ( - Literal["user", "endpoint", "user_group", "service_principal"] - | AiGatewayRateLimitKey -) +AiGatewayRateLimitKeyParam = Literal["user", "endpoint", "user_group", "service_principal"] | AiGatewayRateLimitKey diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_renewal_period.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_renewal_period.py index e315d878c83..d64c74e995d 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_renewal_period.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_renewal_period.py @@ -1,13 +1,71 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class AiGatewayRateLimitRenewalPeriod(Enum): MINUTE = "minute" - -AiGatewayRateLimitRenewalPeriodParam = ( - Literal["minute"] | AiGatewayRateLimitRenewalPeriod -) +AiGatewayRateLimitRenewalPeriodParam = Literal["minute"] | AiGatewayRateLimitRenewalPeriod diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_usage_tracking_config.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_usage_tracking_config.py index bd5b5dd2806..8648140e77d 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_usage_tracking_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_usage_tracking_config.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AiGatewayUsageTrackingConfig: """""" @@ -21,11 +79,12 @@ class AiGatewayUsageTrackingConfig: """ @classmethod - def from_dict(cls, value: "AiGatewayUsageTrackingConfigDict") -> "Self": + def from_dict(cls, value: 'AiGatewayUsageTrackingConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AiGatewayUsageTrackingConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AiGatewayUsageTrackingConfigDict': + return _transform_to_json_value(self) # type:ignore + class AiGatewayUsageTrackingConfigDict(TypedDict, total=False): @@ -37,6 +96,4 @@ class AiGatewayUsageTrackingConfigDict(TypedDict, total=False): """ -AiGatewayUsageTrackingConfigParam = ( - AiGatewayUsageTrackingConfigDict | AiGatewayUsageTrackingConfig -) +AiGatewayUsageTrackingConfigParam = AiGatewayUsageTrackingConfigDict | AiGatewayUsageTrackingConfig diff --git a/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config.py b/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config.py index 0a51df09ba8..9eb73c0a2b0 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config.py @@ -1,20 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import ( - AmazonBedrockConfigBedrockProvider, - AmazonBedrockConfigBedrockProviderParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AmazonBedrockConfig: """""" @@ -83,11 +137,12 @@ class AmazonBedrockConfig: """ @classmethod - def from_dict(cls, value: "AmazonBedrockConfigDict") -> "Self": + def from_dict(cls, value: 'AmazonBedrockConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AmazonBedrockConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AmazonBedrockConfigDict': + return _transform_to_json_value(self) # type:ignore + class AmazonBedrockConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config_bedrock_provider.py b/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config_bedrock_provider.py index bc3de02a2e0..0242f334506 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config_bedrock_provider.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config_bedrock_provider.py @@ -1,7 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class AmazonBedrockConfigBedrockProvider(Enum): @@ -10,8 +71,4 @@ class AmazonBedrockConfigBedrockProvider(Enum): AI21LABS = "ai21labs" AMAZON = "amazon" - -AmazonBedrockConfigBedrockProviderParam = ( - Literal["anthropic", "cohere", "ai21labs", "amazon"] - | AmazonBedrockConfigBedrockProvider -) +AmazonBedrockConfigBedrockProviderParam = Literal["anthropic", "cohere", "ai21labs", "amazon"] | AmazonBedrockConfigBedrockProvider diff --git a/python/databricks/bundles/model_serving_endpoints/_models/anthropic_config.py b/python/databricks/bundles/model_serving_endpoints/_models/anthropic_config.py index 97ab29711b0..a1a191a8d5e 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/anthropic_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/anthropic_config.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AnthropicConfig: """""" @@ -32,11 +90,12 @@ class AnthropicConfig: """ @classmethod - def from_dict(cls, value: "AnthropicConfigDict") -> "Self": + def from_dict(cls, value: 'AnthropicConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AnthropicConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AnthropicConfigDict': + return _transform_to_json_value(self) # type:ignore + class AnthropicConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/api_key_auth.py b/python/databricks/bundles/model_serving_endpoints/_models/api_key_auth.py index a5a33f0aad5..1fd37db68c9 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/api_key_auth.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/api_key_auth.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ApiKeyAuth: """""" @@ -33,11 +91,12 @@ class ApiKeyAuth: """ @classmethod - def from_dict(cls, value: "ApiKeyAuthDict") -> "Self": + def from_dict(cls, value: 'ApiKeyAuthDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ApiKeyAuthDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ApiKeyAuthDict': + return _transform_to_json_value(self) # type:ignore + class ApiKeyAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/auto_capture_config_input.py b/python/databricks/bundles/model_serving_endpoints/_models/auto_capture_config_input.py index b7e3a5820bb..53c2e3b608a 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/auto_capture_config_input.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/auto_capture_config_input.py @@ -1,23 +1,80 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AutoCaptureConfigInput: """ [DEPRECATED] Deprecated: legacy inference table configuration. Please use AI Gateway inference tables instead. See https://docs.databricks.com/aws/en/ai-gateway/inference-tables. """ - catalog_name: VariableOrOptional[str] = None """ The name of the catalog in Unity Catalog. NOTE: On update, you cannot change the catalog name if the inference table is already enabled. @@ -39,11 +96,12 @@ class AutoCaptureConfigInput: """ @classmethod - def from_dict(cls, value: "AutoCaptureConfigInputDict") -> "Self": + def from_dict(cls, value: 'AutoCaptureConfigInputDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AutoCaptureConfigInputDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AutoCaptureConfigInputDict': + return _transform_to_json_value(self) # type:ignore + class AutoCaptureConfigInputDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/bearer_token_auth.py b/python/databricks/bundles/model_serving_endpoints/_models/bearer_token_auth.py index dc05f6c53c2..b48cba675ab 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/bearer_token_auth.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/bearer_token_auth.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class BearerTokenAuth: """""" @@ -28,11 +86,12 @@ class BearerTokenAuth: """ @classmethod - def from_dict(cls, value: "BearerTokenAuthDict") -> "Self": + def from_dict(cls, value: 'BearerTokenAuthDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "BearerTokenAuthDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'BearerTokenAuthDict': + return _transform_to_json_value(self) # type:ignore + class BearerTokenAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/cohere_config.py b/python/databricks/bundles/model_serving_endpoints/_models/cohere_config.py index e70c767e9c2..ca8209db65b 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/cohere_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/cohere_config.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class CohereConfig: """""" @@ -38,11 +96,12 @@ class CohereConfig: """ @classmethod - def from_dict(cls, value: "CohereConfigDict") -> "Self": + def from_dict(cls, value: 'CohereConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CohereConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CohereConfigDict': + return _transform_to_json_value(self) # type:ignore + class CohereConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/custom_provider_config.py b/python/databricks/bundles/model_serving_endpoints/_models/custom_provider_config.py index e50b61d00ff..bccc6d70876 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/custom_provider_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/custom_provider_config.py @@ -1,30 +1,79 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ( - ApiKeyAuth, - ApiKeyAuthParam, -) -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import ( - BearerTokenAuth, - BearerTokenAuthParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class CustomProviderConfig: """ Configs needed to create a custom provider model route. """ - custom_provider_url: VariableOr[str] """ This is a field to provide the URL of the custom provider API. @@ -43,11 +92,12 @@ class CustomProviderConfig: """ @classmethod - def from_dict(cls, value: "CustomProviderConfigDict") -> "Self": + def from_dict(cls, value: 'CustomProviderConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "CustomProviderConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'CustomProviderConfigDict': + return _transform_to_json_value(self) # type:ignore + class CustomProviderConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/databricks_model_serving_config.py b/python/databricks/bundles/model_serving_endpoints/_models/databricks_model_serving_config.py index 72f17e7c5de..72dc9af949c 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/databricks_model_serving_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/databricks_model_serving_config.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DatabricksModelServingConfig: """""" @@ -42,11 +100,12 @@ class DatabricksModelServingConfig: """ @classmethod - def from_dict(cls, value: "DatabricksModelServingConfigDict") -> "Self": + def from_dict(cls, value: 'DatabricksModelServingConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DatabricksModelServingConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DatabricksModelServingConfigDict': + return _transform_to_json_value(self) # type:ignore + class DatabricksModelServingConfigDict(TypedDict, total=False): @@ -79,6 +138,4 @@ class DatabricksModelServingConfigDict(TypedDict, total=False): """ -DatabricksModelServingConfigParam = ( - DatabricksModelServingConfigDict | DatabricksModelServingConfig -) +DatabricksModelServingConfigParam = DatabricksModelServingConfigDict | DatabricksModelServingConfig diff --git a/python/databricks/bundles/model_serving_endpoints/_models/email_notifications.py b/python/databricks/bundles/model_serving_endpoints/_models/email_notifications.py index 3eaeb0a8358..a422003de36 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/email_notifications.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/email_notifications.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EmailNotifications: """""" @@ -26,11 +84,12 @@ class EmailNotifications: """ @classmethod - def from_dict(cls, value: "EmailNotificationsDict") -> "Self": + def from_dict(cls, value: 'EmailNotificationsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EmailNotificationsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EmailNotificationsDict': + return _transform_to_json_value(self) # type:ignore + class EmailNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/endpoint_core_config_input.py b/python/databricks/bundles/model_serving_endpoints/_models/endpoint_core_config_input.py index b74a53e067d..62eb78c51eb 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/endpoint_core_config_input.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/endpoint_core_config_input.py @@ -1,32 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import ( - AutoCaptureConfigInput, - AutoCaptureConfigInputParam, -) -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ( - ServedEntityInput, - ServedEntityInputParam, -) -from databricks.bundles.model_serving_endpoints._models.served_model_input import ( - ServedModelInput, - ServedModelInputParam, -) -from databricks.bundles.model_serving_endpoints._models.traffic_config import ( - TrafficConfig, - TrafficConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EndpointCoreConfigInput: """""" @@ -55,11 +97,12 @@ class EndpointCoreConfigInput: """ @classmethod - def from_dict(cls, value: "EndpointCoreConfigInputDict") -> "Self": + def from_dict(cls, value: 'EndpointCoreConfigInputDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EndpointCoreConfigInputDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EndpointCoreConfigInputDict': + return _transform_to_json_value(self) # type:ignore + class EndpointCoreConfigInputDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/endpoint_tag.py b/python/databricks/bundles/model_serving_endpoints/_models/endpoint_tag.py index ee2bc8943d1..3ecd5c66208 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/endpoint_tag.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/endpoint_tag.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EndpointTag: """""" @@ -26,11 +84,12 @@ class EndpointTag: """ @classmethod - def from_dict(cls, value: "EndpointTagDict") -> "Self": + def from_dict(cls, value: 'EndpointTagDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EndpointTagDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EndpointTagDict': + return _transform_to_json_value(self) # type:ignore + class EndpointTagDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/external_model.py b/python/databricks/bundles/model_serving_endpoints/_models/external_model.py index 0f2fe2a6224..2fafeb97b09 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/external_model.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/external_model.py @@ -1,56 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import ( - Ai21LabsConfig, - Ai21LabsConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import ( - AmazonBedrockConfig, - AmazonBedrockConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.anthropic_config import ( - AnthropicConfig, - AnthropicConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.cohere_config import ( - CohereConfig, - CohereConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import ( - CustomProviderConfig, - CustomProviderConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import ( - DatabricksModelServingConfig, - DatabricksModelServingConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ( - ExternalModelProvider, - ExternalModelProviderParam, -) -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import ( - GoogleCloudVertexAiConfig, - GoogleCloudVertexAiConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.open_ai_config import ( - OpenAiConfig, - OpenAiConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import ( - PaLmConfig, - PaLmConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ExternalModel: """""" @@ -95,9 +113,7 @@ class ExternalModel: Custom Provider Config. Only required if the provider is 'custom'. """ - databricks_model_serving_config: VariableOrOptional[ - DatabricksModelServingConfig - ] = None + databricks_model_serving_config: VariableOrOptional[DatabricksModelServingConfig] = None """ Databricks Model Serving Config. Only required if the provider is 'databricks-model-serving'. """ @@ -118,11 +134,12 @@ class ExternalModel: """ @classmethod - def from_dict(cls, value: "ExternalModelDict") -> "Self": + def from_dict(cls, value: 'ExternalModelDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ExternalModelDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ExternalModelDict': + return _transform_to_json_value(self) # type:ignore + class ExternalModelDict(TypedDict, total=False): @@ -168,9 +185,7 @@ class ExternalModelDict(TypedDict, total=False): Custom Provider Config. Only required if the provider is 'custom'. """ - databricks_model_serving_config: VariableOrOptional[ - DatabricksModelServingConfigParam - ] + databricks_model_serving_config: VariableOrOptional[DatabricksModelServingConfigParam] """ Databricks Model Serving Config. Only required if the provider is 'databricks-model-serving'. """ diff --git a/python/databricks/bundles/model_serving_endpoints/_models/external_model_provider.py b/python/databricks/bundles/model_serving_endpoints/_models/external_model_provider.py index ee3ef3e0c08..706ff0f811c 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/external_model_provider.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/external_model_provider.py @@ -1,7 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class ExternalModelProvider(Enum): @@ -15,18 +76,4 @@ class ExternalModelProvider(Enum): PALM = "palm" CUSTOM = "custom" - -ExternalModelProviderParam = ( - Literal[ - "ai21labs", - "anthropic", - "amazon-bedrock", - "cohere", - "databricks-model-serving", - "google-cloud-vertex-ai", - "openai", - "palm", - "custom", - ] - | ExternalModelProvider -) +ExternalModelProviderParam = Literal["ai21labs", "anthropic", "amazon-bedrock", "cohere", "databricks-model-serving", "google-cloud-vertex-ai", "openai", "palm", "custom"] | ExternalModelProvider diff --git a/python/databricks/bundles/model_serving_endpoints/_models/fallback_config.py b/python/databricks/bundles/model_serving_endpoints/_models/fallback_config.py index c96493b0982..591e2042919 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/fallback_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/fallback_config.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FallbackConfig: """""" @@ -24,11 +82,12 @@ class FallbackConfig: """ @classmethod - def from_dict(cls, value: "FallbackConfigDict") -> "Self": + def from_dict(cls, value: 'FallbackConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FallbackConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FallbackConfigDict': + return _transform_to_json_value(self) # type:ignore + class FallbackConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/google_cloud_vertex_ai_config.py b/python/databricks/bundles/model_serving_endpoints/_models/google_cloud_vertex_ai_config.py index 5b8e3030f56..5e7c526dbd9 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/google_cloud_vertex_ai_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/google_cloud_vertex_ai_config.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GoogleCloudVertexAiConfig: """""" @@ -55,11 +113,12 @@ class GoogleCloudVertexAiConfig: """ @classmethod - def from_dict(cls, value: "GoogleCloudVertexAiConfigDict") -> "Self": + def from_dict(cls, value: 'GoogleCloudVertexAiConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GoogleCloudVertexAiConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GoogleCloudVertexAiConfigDict': + return _transform_to_json_value(self) # type:ignore + class GoogleCloudVertexAiConfigDict(TypedDict, total=False): @@ -105,6 +164,4 @@ class GoogleCloudVertexAiConfigDict(TypedDict, total=False): """ -GoogleCloudVertexAiConfigParam = ( - GoogleCloudVertexAiConfigDict | GoogleCloudVertexAiConfig -) +GoogleCloudVertexAiConfigParam = GoogleCloudVertexAiConfigDict | GoogleCloudVertexAiConfig diff --git a/python/databricks/bundles/model_serving_endpoints/_models/lifecycle.py b/python/databricks/bundles/model_serving_endpoints/_models/lifecycle.py index 697776a198e..6053b9dd0a1 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/lifecycle.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/lifecycle.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +79,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint.py b/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint.py index 64ecfbd87d1..3326583dda1 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint.py @@ -1,53 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import ( - AiGatewayConfig, - AiGatewayConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.email_notifications import ( - EmailNotifications, - EmailNotificationsParam, -) -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import ( - EndpointCoreConfigInput, - EndpointCoreConfigInputParam, -) -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import ( - EndpointTag, - EndpointTagParam, -) -from databricks.bundles.model_serving_endpoints._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ( - ModelServingEndpointPermission, - ModelServingEndpointPermissionParam, -) -from databricks.bundles.model_serving_endpoints._models.rate_limit import ( - RateLimit, - RateLimitParam, -) -from databricks.bundles.model_serving_endpoints._models.telemetry_config import ( - TelemetryConfig, - TelemetryConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelServingEndpoint(Resource): """""" @@ -85,9 +106,7 @@ class ModelServingEndpoint(Resource): Settings that control the deployment lifecycle of the resource, such as preventing it from being destroyed. """ - permissions: VariableOrList[ModelServingEndpointPermission] = field( - default_factory=list - ) + permissions: VariableOrList[ModelServingEndpointPermission] = field(default_factory=list) """ The permissions to apply to this resource. """ @@ -113,11 +132,12 @@ class ModelServingEndpoint(Resource): """ @classmethod - def from_dict(cls, value: "ModelServingEndpointDict") -> "Self": + def from_dict(cls, value: 'ModelServingEndpointDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServingEndpointDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServingEndpointDict': + return _transform_to_json_value(self) # type:ignore + class ModelServingEndpointDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint_permission.py b/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint_permission.py index 82e2fdc054a..93522ed346b 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint_permission.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint_permission.py @@ -1,20 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ( - ServingEndpointPermissionLevel, - ServingEndpointPermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelServingEndpointPermission: """""" @@ -40,11 +94,12 @@ class ModelServingEndpointPermission: """ @classmethod - def from_dict(cls, value: "ModelServingEndpointPermissionDict") -> "Self": + def from_dict(cls, value: 'ModelServingEndpointPermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelServingEndpointPermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelServingEndpointPermissionDict': + return _transform_to_json_value(self) # type:ignore + class ModelServingEndpointPermissionDict(TypedDict, total=False): @@ -71,6 +126,4 @@ class ModelServingEndpointPermissionDict(TypedDict, total=False): """ -ModelServingEndpointPermissionParam = ( - ModelServingEndpointPermissionDict | ModelServingEndpointPermission -) +ModelServingEndpointPermissionParam = ModelServingEndpointPermissionDict | ModelServingEndpointPermission diff --git a/python/databricks/bundles/model_serving_endpoints/_models/open_ai_config.py b/python/databricks/bundles/model_serving_endpoints/_models/open_ai_config.py index e19a4eac665..a371860d43b 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/open_ai_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/open_ai_config.py @@ -1,22 +1,79 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class OpenAiConfig: """ Configs needed to create an OpenAI model route. """ - microsoft_entra_client_id: VariableOrOptional[str] = None """ This field is only required for Azure AD OpenAI and is the Microsoft @@ -102,11 +159,12 @@ class OpenAiConfig: """ @classmethod - def from_dict(cls, value: "OpenAiConfigDict") -> "Self": + def from_dict(cls, value: 'OpenAiConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "OpenAiConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'OpenAiConfigDict': + return _transform_to_json_value(self) # type:ignore + class OpenAiConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/pa_lm_config.py b/python/databricks/bundles/model_serving_endpoints/_models/pa_lm_config.py index 98e5c572687..ce7e0a884d6 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/pa_lm_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/pa_lm_config.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PaLmConfig: """""" @@ -32,11 +90,12 @@ class PaLmConfig: """ @classmethod - def from_dict(cls, value: "PaLmConfigDict") -> "Self": + def from_dict(cls, value: 'PaLmConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PaLmConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PaLmConfigDict': + return _transform_to_json_value(self) # type:ignore + class PaLmConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit.py b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit.py index 5a0192df8ce..57c58ea9d20 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit.py @@ -1,30 +1,79 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import ( - RateLimitKey, - RateLimitKeyParam, -) -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import ( - RateLimitRenewalPeriod, - RateLimitRenewalPeriodParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RateLimit: """ - [DEPRECATED] + [DEPRECATED] """ - calls: VariableOr[int] """ Used to specify how many calls are allowed for a key within the renewal_period. @@ -41,11 +90,12 @@ class RateLimit: """ @classmethod - def from_dict(cls, value: "RateLimitDict") -> "Self": + def from_dict(cls, value: 'RateLimitDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RateLimitDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RateLimitDict': + return _transform_to_json_value(self) # type:ignore + class RateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_key.py b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_key.py index b6ebe9db1c4..4c329bcc14a 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_key.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_key.py @@ -1,16 +1,75 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class RateLimitKey(Enum): """ - [DEPRECATED] + [DEPRECATED] """ - USER = "user" ENDPOINT = "endpoint" - RateLimitKeyParam = Literal["user", "endpoint"] | RateLimitKey diff --git a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_renewal_period.py b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_renewal_period.py index 81f900bf39f..4ec5ff8cf02 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_renewal_period.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_renewal_period.py @@ -1,15 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class RateLimitRenewalPeriod(Enum): """ - [DEPRECATED] + [DEPRECATED] """ - MINUTE = "minute" - RateLimitRenewalPeriodParam = Literal["minute"] | RateLimitRenewalPeriod diff --git a/python/databricks/bundles/model_serving_endpoints/_models/route.py b/python/databricks/bundles/model_serving_endpoints/_models/route.py index eba4336b0fc..3796abc206f 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/route.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/route.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Route: """""" @@ -28,11 +86,12 @@ class Route: """ @classmethod - def from_dict(cls, value: "RouteDict") -> "Self": + def from_dict(cls, value: 'RouteDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RouteDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RouteDict': + return _transform_to_json_value(self) # type:ignore + class RouteDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/served_entity_input.py b/python/databricks/bundles/model_serving_endpoints/_models/served_entity_input.py index e40050ca3e6..1a8208b162d 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/served_entity_input.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/served_entity_input.py @@ -1,24 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrDict, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.external_model import ( - ExternalModel, - ExternalModelParam, -) -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ( - ServingModelWorkloadType, - ServingModelWorkloadTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ServedEntityInput: """""" @@ -98,11 +148,12 @@ class ServedEntityInput: """ @classmethod - def from_dict(cls, value: "ServedEntityInputDict") -> "Self": + def from_dict(cls, value: 'ServedEntityInputDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ServedEntityInputDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ServedEntityInputDict': + return _transform_to_json_value(self) # type:ignore + class ServedEntityInputDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/served_model_input.py b/python/databricks/bundles/model_serving_endpoints/_models/served_model_input.py index 90c98757af5..31a317ec704 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/served_model_input.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/served_model_input.py @@ -1,24 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrOptional, -) -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ( - ServedModelInputWorkloadType, - ServedModelInputWorkloadTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ServedModelInput: """""" @@ -90,11 +140,12 @@ class ServedModelInput: """ @classmethod - def from_dict(cls, value: "ServedModelInputDict") -> "Self": + def from_dict(cls, value: 'ServedModelInputDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ServedModelInputDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ServedModelInputDict': + return _transform_to_json_value(self) # type:ignore + class ServedModelInputDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/served_model_input_workload_type.py b/python/databricks/bundles/model_serving_endpoints/_models/served_model_input_workload_type.py index 87f62c41e34..c709f6185cb 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/served_model_input_workload_type.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/served_model_input_workload_type.py @@ -1,14 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class ServedModelInputWorkloadType(Enum): """ Please keep this in sync with workload types in InferenceEndpointEntities.scala. """ - CPU = "CPU" GPU_MEDIUM = "GPU_MEDIUM" GPU_SMALL = "GPU_SMALL" @@ -19,18 +79,4 @@ class ServedModelInputWorkloadType(Enum): GPU_XLARGE = "GPU_XLARGE" CPU_MEDIUM = "CPU_MEDIUM" - -ServedModelInputWorkloadTypeParam = ( - Literal[ - "CPU", - "GPU_MEDIUM", - "GPU_SMALL", - "GPU_LARGE", - "MULTIGPU_MEDIUM", - "CPU_LARGE", - "GPU_XLARGE_8", - "GPU_XLARGE", - "CPU_MEDIUM", - ] - | ServedModelInputWorkloadType -) +ServedModelInputWorkloadTypeParam = Literal["CPU", "GPU_MEDIUM", "GPU_SMALL", "GPU_LARGE", "MULTIGPU_MEDIUM", "CPU_LARGE", "GPU_XLARGE_8", "GPU_XLARGE", "CPU_MEDIUM"] | ServedModelInputWorkloadType diff --git a/python/databricks/bundles/model_serving_endpoints/_models/serving_endpoint_permission_level.py b/python/databricks/bundles/model_serving_endpoints/_models/serving_endpoint_permission_level.py index d9d75992c54..1cf68783544 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/serving_endpoint_permission_level.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/serving_endpoint_permission_level.py @@ -1,19 +1,76 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class ServingEndpointPermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_QUERY = "CAN_QUERY" CAN_VIEW = "CAN_VIEW" - -ServingEndpointPermissionLevelParam = ( - Literal["CAN_MANAGE", "CAN_QUERY", "CAN_VIEW"] | ServingEndpointPermissionLevel -) +ServingEndpointPermissionLevelParam = Literal["CAN_MANAGE", "CAN_QUERY", "CAN_VIEW"] | ServingEndpointPermissionLevel diff --git a/python/databricks/bundles/model_serving_endpoints/_models/serving_model_workload_type.py b/python/databricks/bundles/model_serving_endpoints/_models/serving_model_workload_type.py index 04b513e77a7..ded41bbf800 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/serving_model_workload_type.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/serving_model_workload_type.py @@ -1,14 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam class ServingModelWorkloadType(Enum): """ Please keep this in sync with workload types in InferenceEndpointEntities.scala. """ - CPU = "CPU" GPU_MEDIUM = "GPU_MEDIUM" GPU_SMALL = "GPU_SMALL" @@ -19,18 +79,4 @@ class ServingModelWorkloadType(Enum): GPU_XLARGE = "GPU_XLARGE" CPU_MEDIUM = "CPU_MEDIUM" - -ServingModelWorkloadTypeParam = ( - Literal[ - "CPU", - "GPU_MEDIUM", - "GPU_SMALL", - "GPU_LARGE", - "MULTIGPU_MEDIUM", - "CPU_LARGE", - "GPU_XLARGE_8", - "GPU_XLARGE", - "CPU_MEDIUM", - ] - | ServingModelWorkloadType -) +ServingModelWorkloadTypeParam = Literal["CPU", "GPU_MEDIUM", "GPU_SMALL", "GPU_LARGE", "MULTIGPU_MEDIUM", "CPU_LARGE", "GPU_XLARGE_8", "GPU_XLARGE", "CPU_MEDIUM"] | ServingModelWorkloadType diff --git a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_config.py b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_config.py index cffb10c2642..db4a4a68780 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_config.py @@ -1,35 +1,79 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import ( - TelemetryFeature, - TelemetryFeatureParam, -) -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import ( - TelemetryInferenceTableConfig, - TelemetryInferenceTableConfigParam, -) -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import ( - UnityCatalogTableNames, - UnityCatalogTableNamesParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TelemetryConfig: """""" - enabled_telemetry_features: VariableOrList[TelemetryFeature] = field( - default_factory=list - ) + enabled_telemetry_features: VariableOrList[TelemetryFeature] = field(default_factory=list) """ [Public Preview] The telemetry signals to enable for this endpoint. If empty or omitted, all signals are enabled; otherwise only the listed signals are enabled. @@ -53,11 +97,12 @@ class TelemetryConfig: """ @classmethod - def from_dict(cls, value: "TelemetryConfigDict") -> "Self": + def from_dict(cls, value: 'TelemetryConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TelemetryConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TelemetryConfigDict': + return _transform_to_json_value(self) # type:ignore + class TelemetryConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_feature.py b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_feature.py index 2d74f666c07..32629aeca9d 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_feature.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_feature.py @@ -1,7 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam class TelemetryFeature(Enum): @@ -9,19 +70,9 @@ class TelemetryFeature(Enum): A telemetry signal that a serving endpoint can export to Unity Catalog. Use these values to select which signals the endpoint exports. """ - TELEMETRY_FEATURE_LOGS = "TELEMETRY_FEATURE_LOGS" TELEMETRY_FEATURE_TRACES = "TELEMETRY_FEATURE_TRACES" TELEMETRY_FEATURE_METRICS = "TELEMETRY_FEATURE_METRICS" TELEMETRY_FEATURE_INFERENCE_TABLE = "TELEMETRY_FEATURE_INFERENCE_TABLE" - -TelemetryFeatureParam = ( - Literal[ - "TELEMETRY_FEATURE_LOGS", - "TELEMETRY_FEATURE_TRACES", - "TELEMETRY_FEATURE_METRICS", - "TELEMETRY_FEATURE_INFERENCE_TABLE", - ] - | TelemetryFeature -) +TelemetryFeatureParam = Literal["TELEMETRY_FEATURE_LOGS", "TELEMETRY_FEATURE_TRACES", "TELEMETRY_FEATURE_METRICS", "TELEMETRY_FEATURE_INFERENCE_TABLE"] | TelemetryFeature diff --git a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_inference_table_config.py b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_inference_table_config.py index 9153a4c49ac..44dc8bee8c5 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_inference_table_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_inference_table_config.py @@ -1,33 +1,91 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TelemetryInferenceTableConfig: """ Inference table payload logging configuration """ - sampling_fraction: VariableOrOptional[float] = None """ [Public Preview] Fraction of requests sampled for payload logging, in the range [0.0, 1.0], where 1.0 logs all requests. """ @classmethod - def from_dict(cls, value: "TelemetryInferenceTableConfigDict") -> "Self": + def from_dict(cls, value: 'TelemetryInferenceTableConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TelemetryInferenceTableConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TelemetryInferenceTableConfigDict': + return _transform_to_json_value(self) # type:ignore + class TelemetryInferenceTableConfigDict(TypedDict, total=False): @@ -39,6 +97,4 @@ class TelemetryInferenceTableConfigDict(TypedDict, total=False): """ -TelemetryInferenceTableConfigParam = ( - TelemetryInferenceTableConfigDict | TelemetryInferenceTableConfig -) +TelemetryInferenceTableConfigParam = TelemetryInferenceTableConfigDict | TelemetryInferenceTableConfig diff --git a/python/databricks/bundles/model_serving_endpoints/_models/traffic_config.py b/python/databricks/bundles/model_serving_endpoints/_models/traffic_config.py index d25a5c87fef..b4215ab2c29 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/traffic_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/traffic_config.py @@ -1,20 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList -from databricks.bundles.model_serving_endpoints._models.route import ( - Route, - RouteParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TrafficConfig: """""" @@ -25,11 +79,12 @@ class TrafficConfig: """ @classmethod - def from_dict(cls, value: "TrafficConfigDict") -> "Self": + def from_dict(cls, value: 'TrafficConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TrafficConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TrafficConfigDict': + return _transform_to_json_value(self) # type:ignore + class TrafficConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/unity_catalog_table_names.py b/python/databricks/bundles/model_serving_endpoints/_models/unity_catalog_table_names.py index 5cf8a32d821..3a6dcbbfb75 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/unity_catalog_table_names.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/unity_catalog_table_names.py @@ -1,16 +1,74 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam +from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam +from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam +from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam +from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam +from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam +from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam +from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam +from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam +from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam +from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam + +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam + +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam + +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam + +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam + +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class UnityCatalogTableNames: """""" @@ -40,11 +98,12 @@ class UnityCatalogTableNames: """ @classmethod - def from_dict(cls, value: "UnityCatalogTableNamesDict") -> "Self": + def from_dict(cls, value: 'UnityCatalogTableNamesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "UnityCatalogTableNamesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'UnityCatalogTableNamesDict': + return _transform_to_json_value(self) # type:ignore + class UnityCatalogTableNamesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/__init__.py b/python/databricks/bundles/models/__init__.py index a3469f1689e..673f479c47a 100644 --- a/python/databricks/bundles/models/__init__.py +++ b/python/databricks/bundles/models/__init__.py @@ -1,44 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "MlflowModel", - "MlflowModelDict", - "MlflowModelParam", - "MlflowModelPermission", - "MlflowModelPermissionDict", - "MlflowModelPermissionParam", - "ModelTag", - "ModelTagDict", - "ModelTagParam", - "RegisteredModelPermissionLevel", - "RegisteredModelPermissionLevelParam", + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'MlflowModel', + 'MlflowModelDict', + 'MlflowModelParam', + 'MlflowModelPermission', + 'MlflowModelPermissionDict', + 'MlflowModelPermissionParam', + 'ModelTag', + 'ModelTagDict', + 'ModelTagParam', + 'RegisteredModelPermissionLevel', + 'RegisteredModelPermissionLevelParam', ] -from databricks.bundles.models._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.models._models.mlflow_model import ( - MlflowModel, - MlflowModelDict, - MlflowModelParam, -) -from databricks.bundles.models._models.mlflow_model_permission import ( - MlflowModelPermission, - MlflowModelPermissionDict, - MlflowModelPermissionParam, -) -from databricks.bundles.models._models.model_tag import ( - ModelTag, - ModelTagDict, - ModelTagParam, -) -from databricks.bundles.models._models.registered_model_permission_level import ( - RegisteredModelPermissionLevel, - RegisteredModelPermissionLevelParam, -) +from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam +from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam +from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam +from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam + diff --git a/python/databricks/bundles/models/_models/lifecycle.py b/python/databricks/bundles/models/_models/lifecycle.py index 697776a198e..c174e277c77 100644 --- a/python/databricks/bundles/models/_models/lifecycle.py +++ b/python/databricks/bundles/models/_models/lifecycle.py @@ -1,16 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam +from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam +from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam +from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +28,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/_models/mlflow_model.py b/python/databricks/bundles/models/_models/mlflow_model.py index 390f1155f2c..e3242dbde2b 100644 --- a/python/databricks/bundles/models/_models/mlflow_model.py +++ b/python/databricks/bundles/models/_models/mlflow_model.py @@ -1,27 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.models._models.mlflow_model_permission import ( - MlflowModelPermission, - MlflowModelPermissionParam, -) -from databricks.bundles.models._models.model_tag import ModelTag, ModelTagParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam +from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam +from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MlflowModel(Resource): """""" @@ -52,11 +48,12 @@ class MlflowModel(Resource): """ @classmethod - def from_dict(cls, value: "MlflowModelDict") -> "Self": + def from_dict(cls, value: 'MlflowModelDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MlflowModelDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MlflowModelDict': + return _transform_to_json_value(self) # type:ignore + class MlflowModelDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/_models/mlflow_model_permission.py b/python/databricks/bundles/models/_models/mlflow_model_permission.py index 367cbfd4a3a..9163445ec6f 100644 --- a/python/databricks/bundles/models/_models/mlflow_model_permission.py +++ b/python/databricks/bundles/models/_models/mlflow_model_permission.py @@ -1,20 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.models._models.registered_model_permission_level import ( - RegisteredModelPermissionLevel, - RegisteredModelPermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam +from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam +from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MlflowModelPermission: """""" @@ -40,11 +43,12 @@ class MlflowModelPermission: """ @classmethod - def from_dict(cls, value: "MlflowModelPermissionDict") -> "Self": + def from_dict(cls, value: 'MlflowModelPermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MlflowModelPermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MlflowModelPermissionDict': + return _transform_to_json_value(self) # type:ignore + class MlflowModelPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/_models/model_tag.py b/python/databricks/bundles/models/_models/model_tag.py index b434ad05552..98b9885ae7c 100644 --- a/python/databricks/bundles/models/_models/model_tag.py +++ b/python/databricks/bundles/models/_models/model_tag.py @@ -1,22 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam +from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam +from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ModelTag: """ Tag for a registered model """ - key: VariableOrOptional[str] = None """ The tag key. @@ -28,11 +34,12 @@ class ModelTag: """ @classmethod - def from_dict(cls, value: "ModelTagDict") -> "Self": + def from_dict(cls, value: 'ModelTagDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ModelTagDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ModelTagDict': + return _transform_to_json_value(self) # type:ignore + class ModelTagDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/_models/registered_model_permission_level.py b/python/databricks/bundles/models/_models/registered_model_permission_level.py index 1344b5d5e48..56e37bb7ab9 100644 --- a/python/databricks/bundles/models/_models/registered_model_permission_level.py +++ b/python/databricks/bundles/models/_models/registered_model_permission_level.py @@ -1,28 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam +from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam +from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam class RegisteredModelPermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" CAN_MANAGE_PRODUCTION_VERSIONS = "CAN_MANAGE_PRODUCTION_VERSIONS" CAN_MANAGE_STAGING_VERSIONS = "CAN_MANAGE_STAGING_VERSIONS" CAN_EDIT = "CAN_EDIT" CAN_READ = "CAN_READ" - -RegisteredModelPermissionLevelParam = ( - Literal[ - "CAN_MANAGE", - "CAN_MANAGE_PRODUCTION_VERSIONS", - "CAN_MANAGE_STAGING_VERSIONS", - "CAN_EDIT", - "CAN_READ", - ] - | RegisteredModelPermissionLevel -) +RegisteredModelPermissionLevelParam = Literal["CAN_MANAGE", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_MANAGE_STAGING_VERSIONS", "CAN_EDIT", "CAN_READ"] | RegisteredModelPermissionLevel diff --git a/python/databricks/bundles/pipelines/__init__.py b/python/databricks/bundles/pipelines/__init__.py index 2b66e4f795c..34861c3afac 100644 --- a/python/databricks/bundles/pipelines/__init__.py +++ b/python/databricks/bundles/pipelines/__init__.py @@ -1,766 +1,414 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Adlsgen2Info", - "Adlsgen2InfoDict", - "Adlsgen2InfoParam", - "ApiSourceConnectorConfig", - "ApiSourceConnectorConfigDict", - "ApiSourceConnectorConfigParam", - "ApiSourceConnectorOptions", - "ApiSourceConnectorOptionsDict", - "ApiSourceConnectorOptionsParam", - "AutoFullRefreshPolicy", - "AutoFullRefreshPolicyDict", - "AutoFullRefreshPolicyParam", - "AvroTransformerOptions", - "AvroTransformerOptionsDict", - "AvroTransformerOptionsParam", - "AwsAttributes", - "AwsAttributesDict", - "AwsAttributesParam", - "AwsAvailability", - "AwsAvailabilityParam", - "AzureAttributes", - "AzureAttributesDict", - "AzureAttributesParam", - "AzureAvailability", - "AzureAvailabilityParam", - "ClusterLogConf", - "ClusterLogConfDict", - "ClusterLogConfParam", - "ConfidentialComputeType", - "ConfidentialComputeTypeParam", - "ConfluenceConnectorOptions", - "ConfluenceConnectorOptionsDict", - "ConfluenceConnectorOptionsParam", - "ConfluentSchemaRegistryOptions", - "ConfluentSchemaRegistryOptionsDict", - "ConfluentSchemaRegistryOptionsParam", - "ConnectionParameters", - "ConnectionParametersDict", - "ConnectionParametersParam", - "ConnectorOptions", - "ConnectorOptionsDict", - "ConnectorOptionsParam", - "ConnectorType", - "ConnectorTypeParam", - "DataStagingOptions", - "DataStagingOptionsDict", - "DataStagingOptionsParam", - "DayOfWeek", - "DayOfWeekParam", - "DbfsStorageInfo", - "DbfsStorageInfoDict", - "DbfsStorageInfoParam", - "EbsVolumeType", - "EbsVolumeTypeParam", - "EventLogSpec", - "EventLogSpecDict", - "EventLogSpecParam", - "FileFilter", - "FileFilterDict", - "FileFilterParam", - "FileIngestionOptions", - "FileIngestionOptionsDict", - "FileIngestionOptionsFileFormat", - "FileIngestionOptionsFileFormatParam", - "FileIngestionOptionsParam", - "FileIngestionOptionsSchemaEvolutionMode", - "FileIngestionOptionsSchemaEvolutionModeParam", - "FileLibrary", - "FileLibraryDict", - "FileLibraryParam", - "Filters", - "FiltersDict", - "FiltersParam", - "GcpAttributes", - "GcpAttributesDict", - "GcpAttributesParam", - "GcpAvailability", - "GcpAvailabilityParam", - "GcsStorageInfo", - "GcsStorageInfoDict", - "GcsStorageInfoParam", - "GoogleAdsConfig", - "GoogleAdsConfigDict", - "GoogleAdsConfigParam", - "GoogleAdsCustomReportOptions", - "GoogleAdsCustomReportOptionsDict", - "GoogleAdsCustomReportOptionsParam", - "GoogleAdsOptions", - "GoogleAdsOptionsDict", - "GoogleAdsOptionsParam", - "GoogleDriveOptions", - "GoogleDriveOptionsDict", - "GoogleDriveOptionsGoogleDriveEntityType", - "GoogleDriveOptionsGoogleDriveEntityTypeParam", - "GoogleDriveOptionsParam", - "IngestionConfig", - "IngestionConfigDict", - "IngestionConfigParam", - "IngestionGatewayPipelineDefinition", - "IngestionGatewayPipelineDefinitionDict", - "IngestionGatewayPipelineDefinitionParam", - "IngestionPipelineDefinition", - "IngestionPipelineDefinitionDict", - "IngestionPipelineDefinitionFanoutOptions", - "IngestionPipelineDefinitionFanoutOptionsDict", - "IngestionPipelineDefinitionFanoutOptionsParam", - "IngestionPipelineDefinitionParam", - "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig", - "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict", - "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam", - "IngestionPipelineDefinitionWorkdayReportParameters", - "IngestionPipelineDefinitionWorkdayReportParametersDict", - "IngestionPipelineDefinitionWorkdayReportParametersParam", - "InitScriptInfo", - "InitScriptInfoDict", - "InitScriptInfoParam", - "JiraConnectorOptions", - "JiraConnectorOptionsDict", - "JiraConnectorOptionsParam", - "JsonTransformerOptions", - "JsonTransformerOptionsDict", - "JsonTransformerOptionsParam", - "KafkaOptions", - "KafkaOptionsDict", - "KafkaOptionsParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "LinkedInAdsOptions", - "LinkedInAdsOptionsDict", - "LinkedInAdsOptionsLinkedInAdsCustomReportOptions", - "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict", - "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity", - "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam", - "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder", - "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam", - "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity", - "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam", - "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam", - "LinkedInAdsOptionsParam", - "LocalFileInfo", - "LocalFileInfoDict", - "LocalFileInfoParam", - "LogAnalyticsInfo", - "LogAnalyticsInfoDict", - "LogAnalyticsInfoParam", - "MarketoOptions", - "MarketoOptionsDict", - "MarketoOptionsParam", - "MavenLibrary", - "MavenLibraryDict", - "MavenLibraryParam", - "MetaMarketingOptions", - "MetaMarketingOptionsDict", - "MetaMarketingOptionsMetaMarketingCustomReportOptions", - "MetaMarketingOptionsMetaMarketingCustomReportOptionsDict", - "MetaMarketingOptionsMetaMarketingCustomReportOptionsParam", - "MetaMarketingOptionsParam", - "NotebookLibrary", - "NotebookLibraryDict", - "NotebookLibraryParam", - "Notifications", - "NotificationsDict", - "NotificationsParam", - "OperationTimeWindow", - "OperationTimeWindowDict", - "OperationTimeWindowParam", - "OutlookAttachmentMode", - "OutlookAttachmentModeParam", - "OutlookBodyFormat", - "OutlookBodyFormatParam", - "OutlookOptions", - "OutlookOptionsDict", - "OutlookOptionsParam", - "ParseMode", - "ParseModeParam", - "PathPattern", - "PathPatternDict", - "PathPatternParam", - "Pipeline", - "PipelineCluster", - "PipelineClusterAutoscale", - "PipelineClusterAutoscaleDict", - "PipelineClusterAutoscaleMode", - "PipelineClusterAutoscaleModeParam", - "PipelineClusterAutoscaleParam", - "PipelineClusterDict", - "PipelineClusterParam", - "PipelineDict", - "PipelineLibrary", - "PipelineLibraryDict", - "PipelineLibraryParam", - "PipelineParam", - "PipelinePermission", - "PipelinePermissionDict", - "PipelinePermissionLevel", - "PipelinePermissionLevelParam", - "PipelinePermissionParam", - "PipelinesEnvironment", - "PipelinesEnvironmentDict", - "PipelinesEnvironmentParam", - "PostgresCatalogConfig", - "PostgresCatalogConfigDict", - "PostgresCatalogConfigParam", - "PostgresSlotConfig", - "PostgresSlotConfigDict", - "PostgresSlotConfigParam", - "ProtobufTransformerOptions", - "ProtobufTransformerOptionsDict", - "ProtobufTransformerOptionsParam", - "RabbitmqOptions", - "RabbitmqOptionsDict", - "RabbitmqOptionsParam", - "RedditAdsOptions", - "RedditAdsOptionsDict", - "RedditAdsOptionsParam", - "RedditAdsOptionsRedditAdsCustomReportOptions", - "RedditAdsOptionsRedditAdsCustomReportOptionsDict", - "RedditAdsOptionsRedditAdsCustomReportOptionsParam", - "ReportSpec", - "ReportSpecDict", - "ReportSpecParam", - "RestartWindow", - "RestartWindowDict", - "RestartWindowParam", - "RunAs", - "RunAsDict", - "RunAsParam", - "S3StorageInfo", - "S3StorageInfoDict", - "S3StorageInfoParam", - "SchemaRegistryConfig", - "SchemaRegistryConfigDict", - "SchemaRegistryConfigParam", - "SchemaSpec", - "SchemaSpecDict", - "SchemaSpecParam", - "SharepointOptions", - "SharepointOptionsDict", - "SharepointOptionsParam", - "SharepointOptionsSharepointEntityType", - "SharepointOptionsSharepointEntityTypeParam", - "SmartsheetOptions", - "SmartsheetOptionsDict", - "SmartsheetOptionsParam", - "SourceCatalogConfig", - "SourceCatalogConfigDict", - "SourceCatalogConfigParam", - "SourceConfig", - "SourceConfigDict", - "SourceConfigParam", - "TableSpec", - "TableSpecDict", - "TableSpecParam", - "TableSpecificConfig", - "TableSpecificConfigDict", - "TableSpecificConfigParam", - "TableSpecificConfigScdType", - "TableSpecificConfigScdTypeParam", - "TikTokAdsOptions", - "TikTokAdsOptionsDict", - "TikTokAdsOptionsParam", - "TikTokAdsOptionsTikTokAdsCustomReportOptions", - "TikTokAdsOptionsTikTokAdsCustomReportOptionsDict", - "TikTokAdsOptionsTikTokAdsCustomReportOptionsParam", - "TikTokAdsOptionsTikTokDataLevel", - "TikTokAdsOptionsTikTokDataLevelParam", - "TikTokAdsOptionsTikTokReportType", - "TikTokAdsOptionsTikTokReportTypeParam", - "Transformer", - "TransformerDict", - "TransformerFormat", - "TransformerFormatParam", - "TransformerParam", - "VolumesStorageInfo", - "VolumesStorageInfoDict", - "VolumesStorageInfoParam", - "WorkspaceStorageInfo", - "WorkspaceStorageInfoDict", - "WorkspaceStorageInfoParam", - "ZendeskSupportOptions", - "ZendeskSupportOptionsDict", - "ZendeskSupportOptionsParam", + 'Adlsgen2Info', + 'Adlsgen2InfoDict', + 'Adlsgen2InfoParam', + 'ApiSourceConnectorConfig', + 'ApiSourceConnectorConfigDict', + 'ApiSourceConnectorConfigParam', + 'ApiSourceConnectorOptions', + 'ApiSourceConnectorOptionsDict', + 'ApiSourceConnectorOptionsParam', + 'AutoFullRefreshPolicy', + 'AutoFullRefreshPolicyDict', + 'AutoFullRefreshPolicyParam', + 'AvroTransformerOptions', + 'AvroTransformerOptionsDict', + 'AvroTransformerOptionsParam', + 'AwsAttributes', + 'AwsAttributesDict', + 'AwsAttributesParam', + 'AwsAvailability', + 'AwsAvailabilityParam', + 'AzureAttributes', + 'AzureAttributesDict', + 'AzureAttributesParam', + 'AzureAvailability', + 'AzureAvailabilityParam', + 'ClusterLogConf', + 'ClusterLogConfDict', + 'ClusterLogConfParam', + 'ConfidentialComputeType', + 'ConfidentialComputeTypeParam', + 'ConfluenceConnectorOptions', + 'ConfluenceConnectorOptionsDict', + 'ConfluenceConnectorOptionsParam', + 'ConfluentSchemaRegistryOptions', + 'ConfluentSchemaRegistryOptionsDict', + 'ConfluentSchemaRegistryOptionsParam', + 'ConnectionParameters', + 'ConnectionParametersDict', + 'ConnectionParametersParam', + 'ConnectorOptions', + 'ConnectorOptionsDict', + 'ConnectorOptionsParam', + 'ConnectorType', + 'ConnectorTypeParam', + 'DataStagingOptions', + 'DataStagingOptionsDict', + 'DataStagingOptionsParam', + 'DayOfWeek', + 'DayOfWeekParam', + 'DbfsStorageInfo', + 'DbfsStorageInfoDict', + 'DbfsStorageInfoParam', + 'EbsVolumeType', + 'EbsVolumeTypeParam', + 'EventLogSpec', + 'EventLogSpecDict', + 'EventLogSpecParam', + 'FileFilter', + 'FileFilterDict', + 'FileFilterParam', + 'FileIngestionOptions', + 'FileIngestionOptionsDict', + 'FileIngestionOptionsFileFormat', + 'FileIngestionOptionsFileFormatParam', + 'FileIngestionOptionsParam', + 'FileIngestionOptionsSchemaEvolutionMode', + 'FileIngestionOptionsSchemaEvolutionModeParam', + 'FileLibrary', + 'FileLibraryDict', + 'FileLibraryParam', + 'Filters', + 'FiltersDict', + 'FiltersParam', + 'GcpAttributes', + 'GcpAttributesDict', + 'GcpAttributesParam', + 'GcpAvailability', + 'GcpAvailabilityParam', + 'GcsStorageInfo', + 'GcsStorageInfoDict', + 'GcsStorageInfoParam', + 'GoogleAdsConfig', + 'GoogleAdsConfigDict', + 'GoogleAdsConfigParam', + 'GoogleAdsCustomReportOptions', + 'GoogleAdsCustomReportOptionsDict', + 'GoogleAdsCustomReportOptionsParam', + 'GoogleAdsOptions', + 'GoogleAdsOptionsDict', + 'GoogleAdsOptionsParam', + 'GoogleDriveOptions', + 'GoogleDriveOptionsDict', + 'GoogleDriveOptionsGoogleDriveEntityType', + 'GoogleDriveOptionsGoogleDriveEntityTypeParam', + 'GoogleDriveOptionsParam', + 'IngestionConfig', + 'IngestionConfigDict', + 'IngestionConfigParam', + 'IngestionGatewayPipelineDefinition', + 'IngestionGatewayPipelineDefinitionDict', + 'IngestionGatewayPipelineDefinitionParam', + 'IngestionPipelineDefinition', + 'IngestionPipelineDefinitionDict', + 'IngestionPipelineDefinitionFanoutOptions', + 'IngestionPipelineDefinitionFanoutOptionsDict', + 'IngestionPipelineDefinitionFanoutOptionsParam', + 'IngestionPipelineDefinitionParam', + 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig', + 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict', + 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam', + 'IngestionPipelineDefinitionWorkdayReportParameters', + 'IngestionPipelineDefinitionWorkdayReportParametersDict', + 'IngestionPipelineDefinitionWorkdayReportParametersParam', + 'InitScriptInfo', + 'InitScriptInfoDict', + 'InitScriptInfoParam', + 'JiraConnectorOptions', + 'JiraConnectorOptionsDict', + 'JiraConnectorOptionsParam', + 'JsonTransformerOptions', + 'JsonTransformerOptionsDict', + 'JsonTransformerOptionsParam', + 'KafkaOptions', + 'KafkaOptionsDict', + 'KafkaOptionsParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'LinkedInAdsOptions', + 'LinkedInAdsOptionsDict', + 'LinkedInAdsOptionsLinkedInAdsCustomReportOptions', + 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict', + 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity', + 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam', + 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder', + 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam', + 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity', + 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam', + 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam', + 'LinkedInAdsOptionsParam', + 'LocalFileInfo', + 'LocalFileInfoDict', + 'LocalFileInfoParam', + 'LogAnalyticsInfo', + 'LogAnalyticsInfoDict', + 'LogAnalyticsInfoParam', + 'MarketoOptions', + 'MarketoOptionsDict', + 'MarketoOptionsParam', + 'MavenLibrary', + 'MavenLibraryDict', + 'MavenLibraryParam', + 'MetaMarketingOptions', + 'MetaMarketingOptionsDict', + 'MetaMarketingOptionsMetaMarketingCustomReportOptions', + 'MetaMarketingOptionsMetaMarketingCustomReportOptionsDict', + 'MetaMarketingOptionsMetaMarketingCustomReportOptionsParam', + 'MetaMarketingOptionsParam', + 'NotebookLibrary', + 'NotebookLibraryDict', + 'NotebookLibraryParam', + 'Notifications', + 'NotificationsDict', + 'NotificationsParam', + 'OperationTimeWindow', + 'OperationTimeWindowDict', + 'OperationTimeWindowParam', + 'OutlookAttachmentMode', + 'OutlookAttachmentModeParam', + 'OutlookBodyFormat', + 'OutlookBodyFormatParam', + 'OutlookOptions', + 'OutlookOptionsDict', + 'OutlookOptionsParam', + 'ParseMode', + 'ParseModeParam', + 'PathPattern', + 'PathPatternDict', + 'PathPatternParam', + 'Pipeline', + 'PipelineCluster', + 'PipelineClusterAutoscale', + 'PipelineClusterAutoscaleDict', + 'PipelineClusterAutoscaleMode', + 'PipelineClusterAutoscaleModeParam', + 'PipelineClusterAutoscaleParam', + 'PipelineClusterDict', + 'PipelineClusterParam', + 'PipelineDict', + 'PipelineLibrary', + 'PipelineLibraryDict', + 'PipelineLibraryParam', + 'PipelineParam', + 'PipelinePermission', + 'PipelinePermissionDict', + 'PipelinePermissionLevel', + 'PipelinePermissionLevelParam', + 'PipelinePermissionParam', + 'PipelinesEnvironment', + 'PipelinesEnvironmentDict', + 'PipelinesEnvironmentParam', + 'PostgresCatalogConfig', + 'PostgresCatalogConfigDict', + 'PostgresCatalogConfigParam', + 'PostgresSlotConfig', + 'PostgresSlotConfigDict', + 'PostgresSlotConfigParam', + 'ProtobufTransformerOptions', + 'ProtobufTransformerOptionsDict', + 'ProtobufTransformerOptionsParam', + 'RabbitmqOptions', + 'RabbitmqOptionsDict', + 'RabbitmqOptionsParam', + 'RedditAdsOptions', + 'RedditAdsOptionsDict', + 'RedditAdsOptionsParam', + 'RedditAdsOptionsRedditAdsCustomReportOptions', + 'RedditAdsOptionsRedditAdsCustomReportOptionsDict', + 'RedditAdsOptionsRedditAdsCustomReportOptionsParam', + 'ReportSpec', + 'ReportSpecDict', + 'ReportSpecParam', + 'RestartWindow', + 'RestartWindowDict', + 'RestartWindowParam', + 'RunAs', + 'RunAsDict', + 'RunAsParam', + 'S3StorageInfo', + 'S3StorageInfoDict', + 'S3StorageInfoParam', + 'SchemaRegistryConfig', + 'SchemaRegistryConfigDict', + 'SchemaRegistryConfigParam', + 'SchemaSpec', + 'SchemaSpecDict', + 'SchemaSpecParam', + 'SharepointOptions', + 'SharepointOptionsDict', + 'SharepointOptionsParam', + 'SharepointOptionsSharepointEntityType', + 'SharepointOptionsSharepointEntityTypeParam', + 'SmartsheetOptions', + 'SmartsheetOptionsDict', + 'SmartsheetOptionsParam', + 'SourceCatalogConfig', + 'SourceCatalogConfigDict', + 'SourceCatalogConfigParam', + 'SourceConfig', + 'SourceConfigDict', + 'SourceConfigParam', + 'TableSpec', + 'TableSpecDict', + 'TableSpecParam', + 'TableSpecificConfig', + 'TableSpecificConfigDict', + 'TableSpecificConfigParam', + 'TableSpecificConfigScdType', + 'TableSpecificConfigScdTypeParam', + 'TikTokAdsOptions', + 'TikTokAdsOptionsDict', + 'TikTokAdsOptionsParam', + 'TikTokAdsOptionsTikTokAdsCustomReportOptions', + 'TikTokAdsOptionsTikTokAdsCustomReportOptionsDict', + 'TikTokAdsOptionsTikTokAdsCustomReportOptionsParam', + 'TikTokAdsOptionsTikTokDataLevel', + 'TikTokAdsOptionsTikTokDataLevelParam', + 'TikTokAdsOptionsTikTokReportType', + 'TikTokAdsOptionsTikTokReportTypeParam', + 'Transformer', + 'TransformerDict', + 'TransformerFormat', + 'TransformerFormatParam', + 'TransformerParam', + 'VolumesStorageInfo', + 'VolumesStorageInfoDict', + 'VolumesStorageInfoParam', + 'WorkspaceStorageInfo', + 'WorkspaceStorageInfoDict', + 'WorkspaceStorageInfoParam', + 'ZendeskSupportOptions', + 'ZendeskSupportOptionsDict', + 'ZendeskSupportOptionsParam', ] -from databricks.bundles.pipelines._models.adlsgen2_info import ( - Adlsgen2Info, - Adlsgen2InfoDict, - Adlsgen2InfoParam, -) -from databricks.bundles.pipelines._models.api_source_connector_config import ( - ApiSourceConnectorConfig, - ApiSourceConnectorConfigDict, - ApiSourceConnectorConfigParam, -) -from databricks.bundles.pipelines._models.api_source_connector_options import ( - ApiSourceConnectorOptions, - ApiSourceConnectorOptionsDict, - ApiSourceConnectorOptionsParam, -) -from databricks.bundles.pipelines._models.auto_full_refresh_policy import ( - AutoFullRefreshPolicy, - AutoFullRefreshPolicyDict, - AutoFullRefreshPolicyParam, -) -from databricks.bundles.pipelines._models.avro_transformer_options import ( - AvroTransformerOptions, - AvroTransformerOptionsDict, - AvroTransformerOptionsParam, -) -from databricks.bundles.pipelines._models.aws_attributes import ( - AwsAttributes, - AwsAttributesDict, - AwsAttributesParam, -) -from databricks.bundles.pipelines._models.aws_availability import ( - AwsAvailability, - AwsAvailabilityParam, -) -from databricks.bundles.pipelines._models.azure_attributes import ( - AzureAttributes, - AzureAttributesDict, - AzureAttributesParam, -) -from databricks.bundles.pipelines._models.azure_availability import ( - AzureAvailability, - AzureAvailabilityParam, -) -from databricks.bundles.pipelines._models.cluster_log_conf import ( - ClusterLogConf, - ClusterLogConfDict, - ClusterLogConfParam, -) -from databricks.bundles.pipelines._models.confidential_compute_type import ( - ConfidentialComputeType, - ConfidentialComputeTypeParam, -) -from databricks.bundles.pipelines._models.confluence_connector_options import ( - ConfluenceConnectorOptions, - ConfluenceConnectorOptionsDict, - ConfluenceConnectorOptionsParam, -) -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ( - ConfluentSchemaRegistryOptions, - ConfluentSchemaRegistryOptionsDict, - ConfluentSchemaRegistryOptionsParam, -) -from databricks.bundles.pipelines._models.connection_parameters import ( - ConnectionParameters, - ConnectionParametersDict, - ConnectionParametersParam, -) -from databricks.bundles.pipelines._models.connector_options import ( - ConnectorOptions, - ConnectorOptionsDict, - ConnectorOptionsParam, -) -from databricks.bundles.pipelines._models.connector_type import ( - ConnectorType, - ConnectorTypeParam, -) -from databricks.bundles.pipelines._models.data_staging_options import ( - DataStagingOptions, - DataStagingOptionsDict, - DataStagingOptionsParam, -) +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam -from databricks.bundles.pipelines._models.dbfs_storage_info import ( - DbfsStorageInfo, - DbfsStorageInfoDict, - DbfsStorageInfoParam, -) -from databricks.bundles.pipelines._models.ebs_volume_type import ( - EbsVolumeType, - EbsVolumeTypeParam, -) -from databricks.bundles.pipelines._models.event_log_spec import ( - EventLogSpec, - EventLogSpecDict, - EventLogSpecParam, -) -from databricks.bundles.pipelines._models.file_filter import ( - FileFilter, - FileFilterDict, - FileFilterParam, -) -from databricks.bundles.pipelines._models.file_ingestion_options import ( - FileIngestionOptions, - FileIngestionOptionsDict, - FileIngestionOptionsParam, -) -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import ( - FileIngestionOptionsFileFormat, - FileIngestionOptionsFileFormatParam, -) -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import ( - FileIngestionOptionsSchemaEvolutionMode, - FileIngestionOptionsSchemaEvolutionModeParam, -) -from databricks.bundles.pipelines._models.file_library import ( - FileLibrary, - FileLibraryDict, - FileLibraryParam, -) -from databricks.bundles.pipelines._models.filters import ( - Filters, - FiltersDict, - FiltersParam, -) -from databricks.bundles.pipelines._models.gcp_attributes import ( - GcpAttributes, - GcpAttributesDict, - GcpAttributesParam, -) -from databricks.bundles.pipelines._models.gcp_availability import ( - GcpAvailability, - GcpAvailabilityParam, -) -from databricks.bundles.pipelines._models.gcs_storage_info import ( - GcsStorageInfo, - GcsStorageInfoDict, - GcsStorageInfoParam, -) -from databricks.bundles.pipelines._models.google_ads_config import ( - GoogleAdsConfig, - GoogleAdsConfigDict, - GoogleAdsConfigParam, -) -from databricks.bundles.pipelines._models.google_ads_custom_report_options import ( - GoogleAdsCustomReportOptions, - GoogleAdsCustomReportOptionsDict, - GoogleAdsCustomReportOptionsParam, -) -from databricks.bundles.pipelines._models.google_ads_options import ( - GoogleAdsOptions, - GoogleAdsOptionsDict, - GoogleAdsOptionsParam, -) -from databricks.bundles.pipelines._models.google_drive_options import ( - GoogleDriveOptions, - GoogleDriveOptionsDict, - GoogleDriveOptionsParam, -) -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import ( - GoogleDriveOptionsGoogleDriveEntityType, - GoogleDriveOptionsGoogleDriveEntityTypeParam, -) -from databricks.bundles.pipelines._models.ingestion_config import ( - IngestionConfig, - IngestionConfigDict, - IngestionConfigParam, -) -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import ( - IngestionGatewayPipelineDefinition, - IngestionGatewayPipelineDefinitionDict, - IngestionGatewayPipelineDefinitionParam, -) -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import ( - IngestionPipelineDefinition, - IngestionPipelineDefinitionDict, - IngestionPipelineDefinitionParam, -) -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import ( - IngestionPipelineDefinitionFanoutOptions, - IngestionPipelineDefinitionFanoutOptionsDict, - IngestionPipelineDefinitionFanoutOptionsParam, -) -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import ( - IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, - IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, - IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam, -) -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import ( - IngestionPipelineDefinitionWorkdayReportParameters, - IngestionPipelineDefinitionWorkdayReportParametersDict, - IngestionPipelineDefinitionWorkdayReportParametersParam, -) -from databricks.bundles.pipelines._models.init_script_info import ( - InitScriptInfo, - InitScriptInfoDict, - InitScriptInfoParam, -) -from databricks.bundles.pipelines._models.jira_connector_options import ( - JiraConnectorOptions, - JiraConnectorOptionsDict, - JiraConnectorOptionsParam, -) -from databricks.bundles.pipelines._models.json_transformer_options import ( - JsonTransformerOptions, - JsonTransformerOptionsDict, - JsonTransformerOptionsParam, -) -from databricks.bundles.pipelines._models.kafka_options import ( - KafkaOptions, - KafkaOptionsDict, - KafkaOptionsParam, -) -from databricks.bundles.pipelines._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.pipelines._models.linked_in_ads_options import ( - LinkedInAdsOptions, - LinkedInAdsOptionsDict, - LinkedInAdsOptionsParam, -) -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import ( - LinkedInAdsOptionsLinkedInAdsCustomReportOptions, - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam, -) -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import ( - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam, -) -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import ( - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam, -) -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import ( - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam, -) -from databricks.bundles.pipelines._models.local_file_info import ( - LocalFileInfo, - LocalFileInfoDict, - LocalFileInfoParam, -) -from databricks.bundles.pipelines._models.log_analytics_info import ( - LogAnalyticsInfo, - LogAnalyticsInfoDict, - LogAnalyticsInfoParam, -) -from databricks.bundles.pipelines._models.marketo_options import ( - MarketoOptions, - MarketoOptionsDict, - MarketoOptionsParam, -) -from databricks.bundles.pipelines._models.maven_library import ( - MavenLibrary, - MavenLibraryDict, - MavenLibraryParam, -) -from databricks.bundles.pipelines._models.meta_marketing_options import ( - MetaMarketingOptions, - MetaMarketingOptionsDict, - MetaMarketingOptionsParam, -) -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import ( - MetaMarketingOptionsMetaMarketingCustomReportOptions, - MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, - MetaMarketingOptionsMetaMarketingCustomReportOptionsParam, -) -from databricks.bundles.pipelines._models.notebook_library import ( - NotebookLibrary, - NotebookLibraryDict, - NotebookLibraryParam, -) -from databricks.bundles.pipelines._models.notifications import ( - Notifications, - NotificationsDict, - NotificationsParam, -) -from databricks.bundles.pipelines._models.operation_time_window import ( - OperationTimeWindow, - OperationTimeWindowDict, - OperationTimeWindowParam, -) -from databricks.bundles.pipelines._models.outlook_attachment_mode import ( - OutlookAttachmentMode, - OutlookAttachmentModeParam, -) -from databricks.bundles.pipelines._models.outlook_body_format import ( - OutlookBodyFormat, - OutlookBodyFormatParam, -) -from databricks.bundles.pipelines._models.outlook_options import ( - OutlookOptions, - OutlookOptionsDict, - OutlookOptionsParam, -) + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam -from databricks.bundles.pipelines._models.path_pattern import ( - PathPattern, - PathPatternDict, - PathPatternParam, -) -from databricks.bundles.pipelines._models.pipeline import ( - Pipeline, - PipelineDict, - PipelineParam, -) -from databricks.bundles.pipelines._models.pipeline_cluster import ( - PipelineCluster, - PipelineClusterDict, - PipelineClusterParam, -) -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import ( - PipelineClusterAutoscale, - PipelineClusterAutoscaleDict, - PipelineClusterAutoscaleParam, -) -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import ( - PipelineClusterAutoscaleMode, - PipelineClusterAutoscaleModeParam, -) -from databricks.bundles.pipelines._models.pipeline_library import ( - PipelineLibrary, - PipelineLibraryDict, - PipelineLibraryParam, -) -from databricks.bundles.pipelines._models.pipeline_permission import ( - PipelinePermission, - PipelinePermissionDict, - PipelinePermissionParam, -) -from databricks.bundles.pipelines._models.pipeline_permission_level import ( - PipelinePermissionLevel, - PipelinePermissionLevelParam, -) -from databricks.bundles.pipelines._models.pipelines_environment import ( - PipelinesEnvironment, - PipelinesEnvironmentDict, - PipelinesEnvironmentParam, -) -from databricks.bundles.pipelines._models.postgres_catalog_config import ( - PostgresCatalogConfig, - PostgresCatalogConfigDict, - PostgresCatalogConfigParam, -) -from databricks.bundles.pipelines._models.postgres_slot_config import ( - PostgresSlotConfig, - PostgresSlotConfigDict, - PostgresSlotConfigParam, -) -from databricks.bundles.pipelines._models.protobuf_transformer_options import ( - ProtobufTransformerOptions, - ProtobufTransformerOptionsDict, - ProtobufTransformerOptionsParam, -) -from databricks.bundles.pipelines._models.rabbitmq_options import ( - RabbitmqOptions, - RabbitmqOptionsDict, - RabbitmqOptionsParam, -) -from databricks.bundles.pipelines._models.reddit_ads_options import ( - RedditAdsOptions, - RedditAdsOptionsDict, - RedditAdsOptionsParam, -) -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import ( - RedditAdsOptionsRedditAdsCustomReportOptions, - RedditAdsOptionsRedditAdsCustomReportOptionsDict, - RedditAdsOptionsRedditAdsCustomReportOptionsParam, -) -from databricks.bundles.pipelines._models.report_spec import ( - ReportSpec, - ReportSpecDict, - ReportSpecParam, -) -from databricks.bundles.pipelines._models.restart_window import ( - RestartWindow, - RestartWindowDict, - RestartWindowParam, -) -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.s3_storage_info import ( - S3StorageInfo, - S3StorageInfoDict, - S3StorageInfoParam, -) -from databricks.bundles.pipelines._models.schema_registry_config import ( - SchemaRegistryConfig, - SchemaRegistryConfigDict, - SchemaRegistryConfigParam, -) -from databricks.bundles.pipelines._models.schema_spec import ( - SchemaSpec, - SchemaSpecDict, - SchemaSpecParam, -) -from databricks.bundles.pipelines._models.sharepoint_options import ( - SharepointOptions, - SharepointOptionsDict, - SharepointOptionsParam, -) -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import ( - SharepointOptionsSharepointEntityType, - SharepointOptionsSharepointEntityTypeParam, -) -from databricks.bundles.pipelines._models.smartsheet_options import ( - SmartsheetOptions, - SmartsheetOptionsDict, - SmartsheetOptionsParam, -) -from databricks.bundles.pipelines._models.source_catalog_config import ( - SourceCatalogConfig, - SourceCatalogConfigDict, - SourceCatalogConfigParam, -) -from databricks.bundles.pipelines._models.source_config import ( - SourceConfig, - SourceConfigDict, - SourceConfigParam, -) -from databricks.bundles.pipelines._models.table_spec import ( - TableSpec, - TableSpecDict, - TableSpecParam, -) -from databricks.bundles.pipelines._models.table_specific_config import ( - TableSpecificConfig, - TableSpecificConfigDict, - TableSpecificConfigParam, -) -from databricks.bundles.pipelines._models.table_specific_config_scd_type import ( - TableSpecificConfigScdType, - TableSpecificConfigScdTypeParam, -) -from databricks.bundles.pipelines._models.tik_tok_ads_options import ( - TikTokAdsOptions, - TikTokAdsOptionsDict, - TikTokAdsOptionsParam, -) -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import ( - TikTokAdsOptionsTikTokAdsCustomReportOptions, - TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, - TikTokAdsOptionsTikTokAdsCustomReportOptionsParam, -) -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import ( - TikTokAdsOptionsTikTokDataLevel, - TikTokAdsOptionsTikTokDataLevelParam, -) -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import ( - TikTokAdsOptionsTikTokReportType, - TikTokAdsOptionsTikTokReportTypeParam, -) -from databricks.bundles.pipelines._models.transformer import ( - Transformer, - TransformerDict, - TransformerParam, -) -from databricks.bundles.pipelines._models.transformer_format import ( - TransformerFormat, - TransformerFormatParam, -) -from databricks.bundles.pipelines._models.volumes_storage_info import ( - VolumesStorageInfo, - VolumesStorageInfoDict, - VolumesStorageInfoParam, -) -from databricks.bundles.pipelines._models.workspace_storage_info import ( - WorkspaceStorageInfo, - WorkspaceStorageInfoDict, - WorkspaceStorageInfoParam, -) -from databricks.bundles.pipelines._models.zendesk_support_options import ( - ZendeskSupportOptions, - ZendeskSupportOptionsDict, - ZendeskSupportOptionsParam, -) + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + diff --git a/python/databricks/bundles/pipelines/_models/adlsgen2_info.py b/python/databricks/bundles/pipelines/_models/adlsgen2_info.py index 87c883adeb9..4d414ae5536 100644 --- a/python/databricks/bundles/pipelines/_models/adlsgen2_info.py +++ b/python/databricks/bundles/pipelines/_models/adlsgen2_info.py @@ -1,33 +1,159 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Adlsgen2Info: """ A storage location in Adls Gen2 """ - destination: VariableOr[str] """ abfss destination, e.g. `abfss://@.dfs.core.windows.net/`. """ @classmethod - def from_dict(cls, value: "Adlsgen2InfoDict") -> "Self": + def from_dict(cls, value: 'Adlsgen2InfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "Adlsgen2InfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'Adlsgen2InfoDict': + return _transform_to_json_value(self) # type:ignore + class Adlsgen2InfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/api_source_connector_config.py b/python/databricks/bundles/pipelines/_models/api_source_connector_config.py index 2ab533ab32f..a7188f32a28 100644 --- a/python/databricks/bundles/pipelines/_models/api_source_connector_config.py +++ b/python/databricks/bundles/pipelines/_models/api_source_connector_config.py @@ -1,24 +1,149 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrDict +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ApiSourceConnectorConfig: """ :meta private: [EXPERIMENTAL] - + Top-level configuration for API Source connectors with arbitrary configuration. """ - configs: VariableOrDict[str] = field(default_factory=dict) """ :meta private: [EXPERIMENTAL] @@ -27,11 +152,12 @@ class ApiSourceConnectorConfig: """ @classmethod - def from_dict(cls, value: "ApiSourceConnectorConfigDict") -> "Self": + def from_dict(cls, value: 'ApiSourceConnectorConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ApiSourceConnectorConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ApiSourceConnectorConfigDict': + return _transform_to_json_value(self) # type:ignore + class ApiSourceConnectorConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/api_source_connector_options.py b/python/databricks/bundles/pipelines/_models/api_source_connector_options.py index 2e0f23c0f03..ededf309db1 100644 --- a/python/databricks/bundles/pipelines/_models/api_source_connector_options.py +++ b/python/databricks/bundles/pipelines/_models/api_source_connector_options.py @@ -1,24 +1,149 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrDict +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ApiSourceConnectorOptions: """ :meta private: [EXPERIMENTAL] - + Options for API Source connectors with arbitrary configuration. """ - options: VariableOrDict[str] = field(default_factory=dict) """ :meta private: [EXPERIMENTAL] @@ -27,11 +152,12 @@ class ApiSourceConnectorOptions: """ @classmethod - def from_dict(cls, value: "ApiSourceConnectorOptionsDict") -> "Self": + def from_dict(cls, value: 'ApiSourceConnectorOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ApiSourceConnectorOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ApiSourceConnectorOptionsDict': + return _transform_to_json_value(self) # type:ignore + class ApiSourceConnectorOptionsDict(TypedDict, total=False): @@ -45,6 +171,4 @@ class ApiSourceConnectorOptionsDict(TypedDict, total=False): """ -ApiSourceConnectorOptionsParam = ( - ApiSourceConnectorOptionsDict | ApiSourceConnectorOptions -) +ApiSourceConnectorOptionsParam = ApiSourceConnectorOptionsDict | ApiSourceConnectorOptions diff --git a/python/databricks/bundles/pipelines/_models/auto_full_refresh_policy.py b/python/databricks/bundles/pipelines/_models/auto_full_refresh_policy.py index 2ee3ed2382a..dd8f87ae21f 100644 --- a/python/databricks/bundles/pipelines/_models/auto_full_refresh_policy.py +++ b/python/databricks/bundles/pipelines/_models/auto_full_refresh_policy.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AutoFullRefreshPolicy: """ Policy for auto full refresh. """ - enabled: VariableOr[bool] """ [Public Preview] (Required, Mutable) Whether to enable auto full refresh or not. @@ -30,11 +155,12 @@ class AutoFullRefreshPolicy: """ @classmethod - def from_dict(cls, value: "AutoFullRefreshPolicyDict") -> "Self": + def from_dict(cls, value: 'AutoFullRefreshPolicyDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AutoFullRefreshPolicyDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AutoFullRefreshPolicyDict': + return _transform_to_json_value(self) # type:ignore + class AutoFullRefreshPolicyDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/avro_transformer_options.py b/python/databricks/bundles/pipelines/_models/avro_transformer_options.py index 3dbfd7cf72b..fae21e3e9ea 100644 --- a/python/databricks/bundles/pipelines/_models/avro_transformer_options.py +++ b/python/databricks/bundles/pipelines/_models/avro_transformer_options.py @@ -1,21 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam -from databricks.bundles.pipelines._models.schema_registry_config import ( - SchemaRegistryConfig, - SchemaRegistryConfigParam, -) + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AvroTransformerOptions: """""" @@ -51,11 +172,12 @@ class AvroTransformerOptions: """ @classmethod - def from_dict(cls, value: "AvroTransformerOptionsDict") -> "Self": + def from_dict(cls, value: 'AvroTransformerOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AvroTransformerOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AvroTransformerOptionsDict': + return _transform_to_json_value(self) # type:ignore + class AvroTransformerOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/aws_attributes.py b/python/databricks/bundles/pipelines/_models/aws_attributes.py index 3308f3c6e96..9f63436ff27 100644 --- a/python/databricks/bundles/pipelines/_models/aws_attributes.py +++ b/python/databricks/bundles/pipelines/_models/aws_attributes.py @@ -1,30 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.aws_availability import ( - AwsAvailability, - AwsAvailabilityParam, -) -from databricks.bundles.pipelines._models.ebs_volume_type import ( - EbsVolumeType, - EbsVolumeTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AwsAttributes: """ Attributes set during cluster creation which are related to Amazon Web Services. """ - availability: VariableOrOptional[AwsAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -123,11 +240,12 @@ class AwsAttributes: """ @classmethod - def from_dict(cls, value: "AwsAttributesDict") -> "Self": + def from_dict(cls, value: 'AwsAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AwsAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AwsAttributesDict': + return _transform_to_json_value(self) # type:ignore + class AwsAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/aws_availability.py b/python/databricks/bundles/pipelines/_models/aws_availability.py index 4c3810f067e..55cc2ec69ec 100644 --- a/python/databricks/bundles/pipelines/_models/aws_availability.py +++ b/python/databricks/bundles/pipelines/_models/aws_availability.py @@ -1,21 +1,146 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class AwsAvailability(Enum): """ Availability type used for all subsequent nodes past the `first_on_demand` ones. - + Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ - SPOT = "SPOT" ON_DEMAND = "ON_DEMAND" SPOT_WITH_FALLBACK = "SPOT_WITH_FALLBACK" - -AwsAvailabilityParam = ( - Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability -) +AwsAvailabilityParam = Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability diff --git a/python/databricks/bundles/pipelines/_models/azure_attributes.py b/python/databricks/bundles/pipelines/_models/azure_attributes.py index 7caff90e481..d704fa90440 100644 --- a/python/databricks/bundles/pipelines/_models/azure_attributes.py +++ b/python/databricks/bundles/pipelines/_models/azure_attributes.py @@ -1,30 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.azure_availability import ( - AzureAvailability, - AzureAvailabilityParam, -) -from databricks.bundles.pipelines._models.log_analytics_info import ( - LogAnalyticsInfo, - LogAnalyticsInfoParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AzureAttributes: """ Attributes set during cluster creation which are related to Microsoft Azure. """ - availability: VariableOrOptional[AzureAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -72,11 +189,12 @@ class AzureAttributes: """ @classmethod - def from_dict(cls, value: "AzureAttributesDict") -> "Self": + def from_dict(cls, value: 'AzureAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AzureAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AzureAttributesDict': + return _transform_to_json_value(self) # type:ignore + class AzureAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/azure_availability.py b/python/databricks/bundles/pipelines/_models/azure_availability.py index a03cb1fdddf..bc43590fd87 100644 --- a/python/databricks/bundles/pipelines/_models/azure_availability.py +++ b/python/databricks/bundles/pipelines/_models/azure_availability.py @@ -1,7 +1,136 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class AzureAvailability(Enum): @@ -9,13 +138,8 @@ class AzureAvailability(Enum): Availability type used for all subsequent nodes past the `first_on_demand` ones. Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ - SPOT_AZURE = "SPOT_AZURE" ON_DEMAND_AZURE = "ON_DEMAND_AZURE" SPOT_WITH_FALLBACK_AZURE = "SPOT_WITH_FALLBACK_AZURE" - -AzureAvailabilityParam = ( - Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] - | AzureAvailability -) +AzureAvailabilityParam = Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] | AzureAvailability diff --git a/python/databricks/bundles/pipelines/_models/cluster_log_conf.py b/python/databricks/bundles/pipelines/_models/cluster_log_conf.py index 74c9dd37df4..d2153bfc0be 100644 --- a/python/databricks/bundles/pipelines/_models/cluster_log_conf.py +++ b/python/databricks/bundles/pipelines/_models/cluster_log_conf.py @@ -1,34 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.dbfs_storage_info import ( - DbfsStorageInfo, - DbfsStorageInfoParam, -) -from databricks.bundles.pipelines._models.s3_storage_info import ( - S3StorageInfo, - S3StorageInfoParam, -) -from databricks.bundles.pipelines._models.volumes_storage_info import ( - VolumesStorageInfo, - VolumesStorageInfoParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ClusterLogConf: """ Cluster log delivery config """ - dbfs: VariableOrOptional[DbfsStorageInfo] = None """ destination needs to be provided. e.g. @@ -50,11 +163,12 @@ class ClusterLogConf: """ @classmethod - def from_dict(cls, value: "ClusterLogConfDict") -> "Self": + def from_dict(cls, value: 'ClusterLogConfDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ClusterLogConfDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ClusterLogConfDict': + return _transform_to_json_value(self) # type:ignore + class ClusterLogConfDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/confidential_compute_type.py b/python/databricks/bundles/pipelines/_models/confidential_compute_type.py index e881295bd47..8c579046d06 100644 --- a/python/databricks/bundles/pipelines/_models/confidential_compute_type.py +++ b/python/databricks/bundles/pipelines/_models/confidential_compute_type.py @@ -1,23 +1,148 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class ConfidentialComputeType(Enum): """ :meta private: [EXPERIMENTAL] - + Confidential computing technology for GCP instances. Aligns with gcloud's --confidential-compute-type flag and the REST API's confidentialInstanceConfig.confidentialInstanceType field. See: https://cloud.google.com/confidential-computing/confidential-vm/docs/create-a-confidential-vm-instance """ - CONFIDENTIAL_COMPUTE_TYPE_NONE = "CONFIDENTIAL_COMPUTE_TYPE_NONE" SEV_SNP = "SEV_SNP" - -ConfidentialComputeTypeParam = ( - Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType -) +ConfidentialComputeTypeParam = Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType diff --git a/python/databricks/bundles/pipelines/_models/confluence_connector_options.py b/python/databricks/bundles/pipelines/_models/confluence_connector_options.py index 39f414edbd2..e73f41281a1 100644 --- a/python/databricks/bundles/pipelines/_models/confluence_connector_options.py +++ b/python/databricks/bundles/pipelines/_models/confluence_connector_options.py @@ -1,33 +1,159 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ConfluenceConnectorOptions: """ Confluence specific options for ingestion """ - include_confluence_spaces: VariableOrList[str] = field(default_factory=list) """ [Public Preview] (Optional) Spaces to filter Confluence data on """ @classmethod - def from_dict(cls, value: "ConfluenceConnectorOptionsDict") -> "Self": + def from_dict(cls, value: 'ConfluenceConnectorOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ConfluenceConnectorOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ConfluenceConnectorOptionsDict': + return _transform_to_json_value(self) # type:ignore + class ConfluenceConnectorOptionsDict(TypedDict, total=False): @@ -39,6 +165,4 @@ class ConfluenceConnectorOptionsDict(TypedDict, total=False): """ -ConfluenceConnectorOptionsParam = ( - ConfluenceConnectorOptionsDict | ConfluenceConnectorOptions -) +ConfluenceConnectorOptionsParam = ConfluenceConnectorOptionsDict | ConfluenceConnectorOptions diff --git a/python/databricks/bundles/pipelines/_models/confluent_schema_registry_options.py b/python/databricks/bundles/pipelines/_models/confluent_schema_registry_options.py index af82614ea7a..2e0b57bdd7d 100644 --- a/python/databricks/bundles/pipelines/_models/confluent_schema_registry_options.py +++ b/python/databricks/bundles/pipelines/_models/confluent_schema_registry_options.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ConfluentSchemaRegistryOptions: """""" @@ -23,11 +149,12 @@ class ConfluentSchemaRegistryOptions: """ @classmethod - def from_dict(cls, value: "ConfluentSchemaRegistryOptionsDict") -> "Self": + def from_dict(cls, value: 'ConfluentSchemaRegistryOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ConfluentSchemaRegistryOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ConfluentSchemaRegistryOptionsDict': + return _transform_to_json_value(self) # type:ignore + class ConfluentSchemaRegistryOptionsDict(TypedDict, total=False): @@ -41,6 +168,4 @@ class ConfluentSchemaRegistryOptionsDict(TypedDict, total=False): """ -ConfluentSchemaRegistryOptionsParam = ( - ConfluentSchemaRegistryOptionsDict | ConfluentSchemaRegistryOptions -) +ConfluentSchemaRegistryOptionsParam = ConfluentSchemaRegistryOptionsDict | ConfluentSchemaRegistryOptions diff --git a/python/databricks/bundles/pipelines/_models/connection_parameters.py b/python/databricks/bundles/pipelines/_models/connection_parameters.py index bb13f406413..e6057ee8909 100644 --- a/python/databricks/bundles/pipelines/_models/connection_parameters.py +++ b/python/databricks/bundles/pipelines/_models/connection_parameters.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ConnectionParameters: """ :meta private: [EXPERIMENTAL] """ - source_catalog: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -28,11 +153,12 @@ class ConnectionParameters: """ @classmethod - def from_dict(cls, value: "ConnectionParametersDict") -> "Self": + def from_dict(cls, value: 'ConnectionParametersDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ConnectionParametersDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ConnectionParametersDict': + return _transform_to_json_value(self) # type:ignore + class ConnectionParametersDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/connector_options.py b/python/databricks/bundles/pipelines/_models/connector_options.py index 945a5bab56a..cc98b4ad12a 100644 --- a/python/databricks/bundles/pipelines/_models/connector_options.py +++ b/python/databricks/bundles/pipelines/_models/connector_options.py @@ -1,86 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.api_source_connector_options import ( - ApiSourceConnectorOptions, - ApiSourceConnectorOptionsParam, -) -from databricks.bundles.pipelines._models.confluence_connector_options import ( - ConfluenceConnectorOptions, - ConfluenceConnectorOptionsParam, -) -from databricks.bundles.pipelines._models.google_ads_options import ( - GoogleAdsOptions, - GoogleAdsOptionsParam, -) -from databricks.bundles.pipelines._models.google_drive_options import ( - GoogleDriveOptions, - GoogleDriveOptionsParam, -) -from databricks.bundles.pipelines._models.jira_connector_options import ( - JiraConnectorOptions, - JiraConnectorOptionsParam, -) -from databricks.bundles.pipelines._models.kafka_options import ( - KafkaOptions, - KafkaOptionsParam, -) -from databricks.bundles.pipelines._models.linked_in_ads_options import ( - LinkedInAdsOptions, - LinkedInAdsOptionsParam, -) -from databricks.bundles.pipelines._models.marketo_options import ( - MarketoOptions, - MarketoOptionsParam, -) -from databricks.bundles.pipelines._models.meta_marketing_options import ( - MetaMarketingOptions, - MetaMarketingOptionsParam, -) -from databricks.bundles.pipelines._models.outlook_options import ( - OutlookOptions, - OutlookOptionsParam, -) -from databricks.bundles.pipelines._models.rabbitmq_options import ( - RabbitmqOptions, - RabbitmqOptionsParam, -) -from databricks.bundles.pipelines._models.reddit_ads_options import ( - RedditAdsOptions, - RedditAdsOptionsParam, -) -from databricks.bundles.pipelines._models.sharepoint_options import ( - SharepointOptions, - SharepointOptionsParam, -) -from databricks.bundles.pipelines._models.smartsheet_options import ( - SmartsheetOptions, - SmartsheetOptionsParam, -) -from databricks.bundles.pipelines._models.tik_tok_ads_options import ( - TikTokAdsOptions, - TikTokAdsOptionsParam, -) -from databricks.bundles.pipelines._models.zendesk_support_options import ( - ZendeskSupportOptions, - ZendeskSupportOptionsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ConnectorOptions: """ Wrapper message for source-specific options to support multiple connector types """ - api_source_connector_options: VariableOrOptional[ApiSourceConnectorOptions] = None """ :meta private: [EXPERIMENTAL] @@ -196,11 +257,12 @@ class ConnectorOptions: """ @classmethod - def from_dict(cls, value: "ConnectorOptionsDict") -> "Self": + def from_dict(cls, value: 'ConnectorOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ConnectorOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ConnectorOptionsDict': + return _transform_to_json_value(self) # type:ignore + class ConnectorOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/connector_type.py b/python/databricks/bundles/pipelines/_models/connector_type.py index e8c7fb15931..46a3ec1b670 100644 --- a/python/databricks/bundles/pipelines/_models/connector_type.py +++ b/python/databricks/bundles/pipelines/_models/connector_type.py @@ -1,7 +1,136 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class ConnectorType(Enum): @@ -10,9 +139,7 @@ class ConnectorType(Enum): ingestion, ConnectorType can bse used to convey the type of ingestion. If connection_name is provided for database sources, we default to Query Based ingestion """ - CDC = "CDC" QUERY_BASED = "QUERY_BASED" - ConnectorTypeParam = Literal["CDC", "QUERY_BASED"] | ConnectorType diff --git a/python/databricks/bundles/pipelines/_models/data_staging_options.py b/python/databricks/bundles/pipelines/_models/data_staging_options.py index 62672108bbd..a8e833c70a4 100644 --- a/python/databricks/bundles/pipelines/_models/data_staging_options.py +++ b/python/databricks/bundles/pipelines/_models/data_staging_options.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DataStagingOptions: """ Location of staged data storage """ - catalog_name: VariableOr[str] """ [Public Preview] (Required, Immutable) The name of the catalog for the connector's staging storage location. @@ -37,11 +162,12 @@ class DataStagingOptions: """ @classmethod - def from_dict(cls, value: "DataStagingOptionsDict") -> "Self": + def from_dict(cls, value: 'DataStagingOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DataStagingOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DataStagingOptionsDict': + return _transform_to_json_value(self) # type:ignore + class DataStagingOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/day_of_week.py b/python/databricks/bundles/pipelines/_models/day_of_week.py index 8ba5a7c1ab6..44674814837 100644 --- a/python/databricks/bundles/pipelines/_models/day_of_week.py +++ b/python/databricks/bundles/pipelines/_models/day_of_week.py @@ -1,7 +1,136 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class DayOfWeek(Enum): @@ -9,7 +138,6 @@ class DayOfWeek(Enum): Days of week in which the window is allowed to happen. If not specified all days of the week will be used. """ - MONDAY = "MONDAY" TUESDAY = "TUESDAY" WEDNESDAY = "WEDNESDAY" @@ -18,10 +146,4 @@ class DayOfWeek(Enum): SATURDAY = "SATURDAY" SUNDAY = "SUNDAY" - -DayOfWeekParam = ( - Literal[ - "MONDAY", "TUESDAY", "WEDNESDAY", "THURSDAY", "FRIDAY", "SATURDAY", "SUNDAY" - ] - | DayOfWeek -) +DayOfWeekParam = Literal["MONDAY", "TUESDAY", "WEDNESDAY", "THURSDAY", "FRIDAY", "SATURDAY", "SUNDAY"] | DayOfWeek diff --git a/python/databricks/bundles/pipelines/_models/dbfs_storage_info.py b/python/databricks/bundles/pipelines/_models/dbfs_storage_info.py index 29744acf26a..2a62fcd74ac 100644 --- a/python/databricks/bundles/pipelines/_models/dbfs_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/dbfs_storage_info.py @@ -1,33 +1,159 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DbfsStorageInfo: """ A storage location in DBFS """ - destination: VariableOr[str] """ dbfs destination, e.g. `dbfs:/my/path` """ @classmethod - def from_dict(cls, value: "DbfsStorageInfoDict") -> "Self": + def from_dict(cls, value: 'DbfsStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DbfsStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DbfsStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class DbfsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/ebs_volume_type.py b/python/databricks/bundles/pipelines/_models/ebs_volume_type.py index d8f3ef13f89..a4dc426f6e9 100644 --- a/python/databricks/bundles/pipelines/_models/ebs_volume_type.py +++ b/python/databricks/bundles/pipelines/_models/ebs_volume_type.py @@ -1,7 +1,136 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class EbsVolumeType(Enum): @@ -9,11 +138,7 @@ class EbsVolumeType(Enum): All EBS volume types that Databricks supports. See https://aws.amazon.com/ebs/details/ for details. """ - GENERAL_PURPOSE_SSD = "GENERAL_PURPOSE_SSD" THROUGHPUT_OPTIMIZED_HDD = "THROUGHPUT_OPTIMIZED_HDD" - -EbsVolumeTypeParam = ( - Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType -) +EbsVolumeTypeParam = Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType diff --git a/python/databricks/bundles/pipelines/_models/event_log_spec.py b/python/databricks/bundles/pipelines/_models/event_log_spec.py index d415ee83246..915bc04f7e2 100644 --- a/python/databricks/bundles/pipelines/_models/event_log_spec.py +++ b/python/databricks/bundles/pipelines/_models/event_log_spec.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EventLogSpec: """ Configurable event log parameters. """ - catalog: VariableOrOptional[str] = None """ The UC catalog the event log is published under. @@ -33,11 +158,12 @@ class EventLogSpec: """ @classmethod - def from_dict(cls, value: "EventLogSpecDict") -> "Self": + def from_dict(cls, value: 'EventLogSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EventLogSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EventLogSpecDict': + return _transform_to_json_value(self) # type:ignore + class EventLogSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/file_filter.py b/python/databricks/bundles/pipelines/_models/file_filter.py index 7e80f319c0f..73e7b08f5a2 100644 --- a/python/databricks/bundles/pipelines/_models/file_filter.py +++ b/python/databricks/bundles/pipelines/_models/file_filter.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FileFilter: """ :meta private: [EXPERIMENTAL] """ - modified_after: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -44,11 +169,12 @@ class FileFilter: """ @classmethod - def from_dict(cls, value: "FileFilterDict") -> "Self": + def from_dict(cls, value: 'FileFilterDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FileFilterDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FileFilterDict': + return _transform_to_json_value(self) # type:ignore + class FileFilterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/file_ingestion_options.py b/python/databricks/bundles/pipelines/_models/file_ingestion_options.py index 0e6d82d6684..0f460bdb54c 100644 --- a/python/databricks/bundles/pipelines/_models/file_ingestion_options.py +++ b/python/databricks/bundles/pipelines/_models/file_ingestion_options.py @@ -1,35 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import ( - FileIngestionOptionsFileFormat, - FileIngestionOptionsFileFormatParam, -) -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import ( - FileIngestionOptionsSchemaEvolutionMode, - FileIngestionOptionsSchemaEvolutionModeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FileIngestionOptions: """ :meta private: [EXPERIMENTAL] """ - corrupt_record_column: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -88,9 +200,7 @@ class FileIngestionOptions: [Private Preview] """ - schema_evolution_mode: VariableOrOptional[ - FileIngestionOptionsSchemaEvolutionMode - ] = None + schema_evolution_mode: VariableOrOptional[FileIngestionOptionsSchemaEvolutionMode] = None """ :meta private: [EXPERIMENTAL] @@ -113,11 +223,12 @@ class FileIngestionOptions: """ @classmethod - def from_dict(cls, value: "FileIngestionOptionsDict") -> "Self": + def from_dict(cls, value: 'FileIngestionOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FileIngestionOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FileIngestionOptionsDict': + return _transform_to_json_value(self) # type:ignore + class FileIngestionOptionsDict(TypedDict, total=False): @@ -181,9 +292,7 @@ class FileIngestionOptionsDict(TypedDict, total=False): [Private Preview] """ - schema_evolution_mode: VariableOrOptional[ - FileIngestionOptionsSchemaEvolutionModeParam - ] + schema_evolution_mode: VariableOrOptional[FileIngestionOptionsSchemaEvolutionModeParam] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/file_ingestion_options_file_format.py b/python/databricks/bundles/pipelines/_models/file_ingestion_options_file_format.py index 25838583512..f5aeae73670 100644 --- a/python/databricks/bundles/pipelines/_models/file_ingestion_options_file_format.py +++ b/python/databricks/bundles/pipelines/_models/file_ingestion_options_file_format.py @@ -1,14 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class FileIngestionOptionsFileFormat(Enum): """ :meta private: [EXPERIMENTAL] """ - BINARYFILE = "BINARYFILE" JSON = "JSON" CSV = "CSV" @@ -18,8 +146,4 @@ class FileIngestionOptionsFileFormat(Enum): AVRO = "AVRO" ORC = "ORC" - -FileIngestionOptionsFileFormatParam = ( - Literal["BINARYFILE", "JSON", "CSV", "XML", "EXCEL", "PARQUET", "AVRO", "ORC"] - | FileIngestionOptionsFileFormat -) +FileIngestionOptionsFileFormatParam = Literal["BINARYFILE", "JSON", "CSV", "XML", "EXCEL", "PARQUET", "AVRO", "ORC"] | FileIngestionOptionsFileFormat diff --git a/python/databricks/bundles/pipelines/_models/file_ingestion_options_schema_evolution_mode.py b/python/databricks/bundles/pipelines/_models/file_ingestion_options_schema_evolution_mode.py index 6d136255c5f..6bd172b5a16 100644 --- a/python/databricks/bundles/pipelines/_models/file_ingestion_options_schema_evolution_mode.py +++ b/python/databricks/bundles/pipelines/_models/file_ingestion_options_schema_evolution_mode.py @@ -1,28 +1,146 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class FileIngestionOptionsSchemaEvolutionMode(Enum): """ Based on https://docs.databricks.com/aws/en/ingestion/cloud-object-storage/auto-loader/schema#how-does-auto-loader-schema-evolution-work """ - ADD_NEW_COLUMNS_WITH_TYPE_WIDENING = "ADD_NEW_COLUMNS_WITH_TYPE_WIDENING" ADD_NEW_COLUMNS = "ADD_NEW_COLUMNS" RESCUE = "RESCUE" FAIL_ON_NEW_COLUMNS = "FAIL_ON_NEW_COLUMNS" NONE = "NONE" - -FileIngestionOptionsSchemaEvolutionModeParam = ( - Literal[ - "ADD_NEW_COLUMNS_WITH_TYPE_WIDENING", - "ADD_NEW_COLUMNS", - "RESCUE", - "FAIL_ON_NEW_COLUMNS", - "NONE", - ] - | FileIngestionOptionsSchemaEvolutionMode -) +FileIngestionOptionsSchemaEvolutionModeParam = Literal["ADD_NEW_COLUMNS_WITH_TYPE_WIDENING", "ADD_NEW_COLUMNS", "RESCUE", "FAIL_ON_NEW_COLUMNS", "NONE"] | FileIngestionOptionsSchemaEvolutionMode diff --git a/python/databricks/bundles/pipelines/_models/file_library.py b/python/databricks/bundles/pipelines/_models/file_library.py index 4fbf2a564af..96e31a7749e 100644 --- a/python/databricks/bundles/pipelines/_models/file_library.py +++ b/python/databricks/bundles/pipelines/_models/file_library.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class FileLibrary: """""" @@ -21,11 +147,12 @@ class FileLibrary: """ @classmethod - def from_dict(cls, value: "FileLibraryDict") -> "Self": + def from_dict(cls, value: 'FileLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FileLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FileLibraryDict': + return _transform_to_json_value(self) # type:ignore + class FileLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/filters.py b/python/databricks/bundles/pipelines/_models/filters.py index c9dc61c8c0c..b5460ed2d84 100644 --- a/python/databricks/bundles/pipelines/_models/filters.py +++ b/python/databricks/bundles/pipelines/_models/filters.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Filters: """""" @@ -26,11 +152,12 @@ class Filters: """ @classmethod - def from_dict(cls, value: "FiltersDict") -> "Self": + def from_dict(cls, value: 'FiltersDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "FiltersDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'FiltersDict': + return _transform_to_json_value(self) # type:ignore + class FiltersDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/gcp_attributes.py b/python/databricks/bundles/pipelines/_models/gcp_attributes.py index 70768dd4921..5c7a0bc130b 100644 --- a/python/databricks/bundles/pipelines/_models/gcp_attributes.py +++ b/python/databricks/bundles/pipelines/_models/gcp_attributes.py @@ -1,30 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.confidential_compute_type import ( - ConfidentialComputeType, - ConfidentialComputeTypeParam, -) -from databricks.bundles.pipelines._models.gcp_availability import ( - GcpAvailability, - GcpAvailabilityParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GcpAttributes: """ Attributes set during cluster creation which are related to GCP. """ - availability: VariableOrOptional[GcpAvailability] = None """ This field determines whether the spark executors will be scheduled to run on preemptible @@ -91,11 +208,12 @@ class GcpAttributes: """ @classmethod - def from_dict(cls, value: "GcpAttributesDict") -> "Self": + def from_dict(cls, value: 'GcpAttributesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GcpAttributesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GcpAttributesDict': + return _transform_to_json_value(self) # type:ignore + class GcpAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/gcp_availability.py b/python/databricks/bundles/pipelines/_models/gcp_availability.py index 0d391b87fe3..69a064121d9 100644 --- a/python/databricks/bundles/pipelines/_models/gcp_availability.py +++ b/python/databricks/bundles/pipelines/_models/gcp_availability.py @@ -1,7 +1,136 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class GcpAvailability(Enum): @@ -9,13 +138,8 @@ class GcpAvailability(Enum): This field determines whether the instance pool will contain preemptible VMs, on-demand VMs, or preemptible VMs with a fallback to on-demand VMs if the former is unavailable. """ - PREEMPTIBLE_GCP = "PREEMPTIBLE_GCP" ON_DEMAND_GCP = "ON_DEMAND_GCP" PREEMPTIBLE_WITH_FALLBACK_GCP = "PREEMPTIBLE_WITH_FALLBACK_GCP" - -GcpAvailabilityParam = ( - Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] - | GcpAvailability -) +GcpAvailabilityParam = Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] | GcpAvailability diff --git a/python/databricks/bundles/pipelines/_models/gcs_storage_info.py b/python/databricks/bundles/pipelines/_models/gcs_storage_info.py index e8f1059373b..52828712507 100644 --- a/python/databricks/bundles/pipelines/_models/gcs_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/gcs_storage_info.py @@ -1,33 +1,159 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GcsStorageInfo: """ A storage location in Google Cloud Platform's GCS """ - destination: VariableOr[str] """ GCS destination/URI, e.g. `gs://my-bucket/some-prefix` """ @classmethod - def from_dict(cls, value: "GcsStorageInfoDict") -> "Self": + def from_dict(cls, value: 'GcsStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GcsStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GcsStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class GcsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/google_ads_config.py b/python/databricks/bundles/pipelines/_models/google_ads_config.py index 3ddb52b9667..23d17863d03 100644 --- a/python/databricks/bundles/pipelines/_models/google_ads_config.py +++ b/python/databricks/bundles/pipelines/_models/google_ads_config.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GoogleAdsConfig: """ :meta private: [EXPERIMENTAL] """ - manager_account_id: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -29,11 +154,12 @@ class GoogleAdsConfig: """ @classmethod - def from_dict(cls, value: "GoogleAdsConfigDict") -> "Self": + def from_dict(cls, value: 'GoogleAdsConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GoogleAdsConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GoogleAdsConfigDict': + return _transform_to_json_value(self) # type:ignore + class GoogleAdsConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/google_ads_custom_report_options.py b/python/databricks/bundles/pipelines/_models/google_ads_custom_report_options.py index 441a8947342..096fcbf0414 100644 --- a/python/databricks/bundles/pipelines/_models/google_ads_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/google_ads_custom_report_options.py @@ -1,21 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GoogleAdsCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + User-defined custom report for the Google Ads connector. Mirrors the resource + fields + segments + metrics model that Google Ads GAQL exposes. The customer account this report runs against is supplied by the source @@ -25,7 +151,6 @@ class GoogleAdsCustomReportOptions: Only supported on table-type objects: a custom report requires a destination table, so it cannot be specified at the schema/source level. """ - resource: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -65,11 +190,12 @@ class GoogleAdsCustomReportOptions: """ @classmethod - def from_dict(cls, value: "GoogleAdsCustomReportOptionsDict") -> "Self": + def from_dict(cls, value: 'GoogleAdsCustomReportOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GoogleAdsCustomReportOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GoogleAdsCustomReportOptionsDict': + return _transform_to_json_value(self) # type:ignore + class GoogleAdsCustomReportOptionsDict(TypedDict, total=False): @@ -114,6 +240,4 @@ class GoogleAdsCustomReportOptionsDict(TypedDict, total=False): """ -GoogleAdsCustomReportOptionsParam = ( - GoogleAdsCustomReportOptionsDict | GoogleAdsCustomReportOptions -) +GoogleAdsCustomReportOptionsParam = GoogleAdsCustomReportOptionsDict | GoogleAdsCustomReportOptions diff --git a/python/databricks/bundles/pipelines/_models/google_ads_options.py b/python/databricks/bundles/pipelines/_models/google_ads_options.py index 33d94383803..3edf051c4c9 100644 --- a/python/databricks/bundles/pipelines/_models/google_ads_options.py +++ b/python/databricks/bundles/pipelines/_models/google_ads_options.py @@ -1,30 +1,151 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.pipelines._models.google_ads_custom_report_options import ( - GoogleAdsCustomReportOptions, - GoogleAdsCustomReportOptionsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GoogleAdsOptions: """ :meta private: [EXPERIMENTAL] - + Google Ads specific options for ingestion (object-level). When set, these values override the corresponding fields in GoogleAdsConfig (source_configurations). """ - manager_account_id: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -62,11 +183,12 @@ class GoogleAdsOptions: """ @classmethod - def from_dict(cls, value: "GoogleAdsOptionsDict") -> "Self": + def from_dict(cls, value: 'GoogleAdsOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GoogleAdsOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GoogleAdsOptionsDict': + return _transform_to_json_value(self) # type:ignore + class GoogleAdsOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/google_drive_options.py b/python/databricks/bundles/pipelines/_models/google_drive_options.py index e11b5048555..c4d3cfbfbdc 100644 --- a/python/databricks/bundles/pipelines/_models/google_drive_options.py +++ b/python/databricks/bundles/pipelines/_models/google_drive_options.py @@ -1,30 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.file_ingestion_options import ( - FileIngestionOptions, - FileIngestionOptionsParam, -) -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import ( - GoogleDriveOptionsGoogleDriveEntityType, - GoogleDriveOptionsGoogleDriveEntityTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class GoogleDriveOptions: """ :meta private: [EXPERIMENTAL] """ - entity_type: VariableOrOptional[GoogleDriveOptionsGoogleDriveEntityType] = None """ :meta private: [EXPERIMENTAL] @@ -47,11 +164,12 @@ class GoogleDriveOptions: """ @classmethod - def from_dict(cls, value: "GoogleDriveOptionsDict") -> "Self": + def from_dict(cls, value: 'GoogleDriveOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "GoogleDriveOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'GoogleDriveOptionsDict': + return _transform_to_json_value(self) # type:ignore + class GoogleDriveOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/google_drive_options_google_drive_entity_type.py b/python/databricks/bundles/pipelines/_models/google_drive_options_google_drive_entity_type.py index b3b9847c550..a6576134acd 100644 --- a/python/databricks/bundles/pipelines/_models/google_drive_options_google_drive_entity_type.py +++ b/python/databricks/bundles/pipelines/_models/google_drive_options_google_drive_entity_type.py @@ -1,20 +1,144 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class GoogleDriveOptionsGoogleDriveEntityType(Enum): """ :meta private: [EXPERIMENTAL] """ - FILE = "FILE" FILE_METADATA = "FILE_METADATA" PERMISSION = "PERMISSION" - -GoogleDriveOptionsGoogleDriveEntityTypeParam = ( - Literal["FILE", "FILE_METADATA", "PERMISSION"] - | GoogleDriveOptionsGoogleDriveEntityType -) +GoogleDriveOptionsGoogleDriveEntityTypeParam = Literal["FILE", "FILE_METADATA", "PERMISSION"] | GoogleDriveOptionsGoogleDriveEntityType diff --git a/python/databricks/bundles/pipelines/_models/ingestion_config.py b/python/databricks/bundles/pipelines/_models/ingestion_config.py index 1d5a14cceef..e9de551c2bf 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_config.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_config.py @@ -1,22 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecParam -from databricks.bundles.pipelines._models.schema_spec import ( - SchemaSpec, - SchemaSpecParam, -) -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class IngestionConfig: """""" @@ -37,11 +157,12 @@ class IngestionConfig: """ @classmethod - def from_dict(cls, value: "IngestionConfigDict") -> "Self": + def from_dict(cls, value: 'IngestionConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "IngestionConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'IngestionConfigDict': + return _transform_to_json_value(self) # type:ignore + class IngestionConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/ingestion_gateway_pipeline_definition.py b/python/databricks/bundles/pipelines/_models/ingestion_gateway_pipeline_definition.py index 56282a61fea..55374926b31 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_gateway_pipeline_definition.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_gateway_pipeline_definition.py @@ -1,26 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.pipelines._models.connection_parameters import ( - ConnectionParameters, - ConnectionParametersParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class IngestionGatewayPipelineDefinition: """ :meta private: [EXPERIMENTAL] """ - connection_name: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -59,11 +180,12 @@ class IngestionGatewayPipelineDefinition: """ @classmethod - def from_dict(cls, value: "IngestionGatewayPipelineDefinitionDict") -> "Self": + def from_dict(cls, value: 'IngestionGatewayPipelineDefinitionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "IngestionGatewayPipelineDefinitionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'IngestionGatewayPipelineDefinitionDict': + return _transform_to_json_value(self) # type:ignore + class IngestionGatewayPipelineDefinitionDict(TypedDict, total=False): @@ -107,6 +229,4 @@ class IngestionGatewayPipelineDefinitionDict(TypedDict, total=False): """ -IngestionGatewayPipelineDefinitionParam = ( - IngestionGatewayPipelineDefinitionDict | IngestionGatewayPipelineDefinition -) +IngestionGatewayPipelineDefinitionParam = IngestionGatewayPipelineDefinitionDict | IngestionGatewayPipelineDefinition diff --git a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition.py b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition.py index 605a119255e..382a227a495 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition.py @@ -1,40 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.pipelines._models.connector_type import ( - ConnectorType, - ConnectorTypeParam, -) -from databricks.bundles.pipelines._models.data_staging_options import ( - DataStagingOptions, - DataStagingOptionsParam, -) -from databricks.bundles.pipelines._models.ingestion_config import ( - IngestionConfig, - IngestionConfigParam, -) -from databricks.bundles.pipelines._models.operation_time_window import ( - OperationTimeWindow, - OperationTimeWindowParam, -) -from databricks.bundles.pipelines._models.source_config import ( - SourceConfig, - SourceConfigParam, -) -from databricks.bundles.pipelines._models.table_specific_config import ( - TableSpecificConfig, - TableSpecificConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class IngestionPipelineDefinition: """""" @@ -110,11 +212,12 @@ class IngestionPipelineDefinition: """ @classmethod - def from_dict(cls, value: "IngestionPipelineDefinitionDict") -> "Self": + def from_dict(cls, value: 'IngestionPipelineDefinitionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "IngestionPipelineDefinitionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'IngestionPipelineDefinitionDict': + return _transform_to_json_value(self) # type:ignore + class IngestionPipelineDefinitionDict(TypedDict, total=False): @@ -191,6 +294,4 @@ class IngestionPipelineDefinitionDict(TypedDict, total=False): """ -IngestionPipelineDefinitionParam = ( - IngestionPipelineDefinitionDict | IngestionPipelineDefinition -) +IngestionPipelineDefinitionParam = IngestionPipelineDefinitionDict | IngestionPipelineDefinition diff --git a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_fanout_options.py b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_fanout_options.py index 092b91581ed..f73140a7e37 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_fanout_options.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_fanout_options.py @@ -1,20 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.pipelines._models.transformer import ( - Transformer, - TransformerParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class IngestionPipelineDefinitionFanoutOptions: """ @@ -23,7 +145,6 @@ class IngestionPipelineDefinitionFanoutOptions: key derived from the record. The key value becomes the table name suffix: {destination_catalog}.{destination_schema}.{key_value}. """ - fanout_by: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -42,11 +163,12 @@ class IngestionPipelineDefinitionFanoutOptions: """ @classmethod - def from_dict(cls, value: "IngestionPipelineDefinitionFanoutOptionsDict") -> "Self": + def from_dict(cls, value: 'IngestionPipelineDefinitionFanoutOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "IngestionPipelineDefinitionFanoutOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'IngestionPipelineDefinitionFanoutOptionsDict': + return _transform_to_json_value(self) # type:ignore + class IngestionPipelineDefinitionFanoutOptionsDict(TypedDict, total=False): @@ -70,7 +192,4 @@ class IngestionPipelineDefinitionFanoutOptionsDict(TypedDict, total=False): """ -IngestionPipelineDefinitionFanoutOptionsParam = ( - IngestionPipelineDefinitionFanoutOptionsDict - | IngestionPipelineDefinitionFanoutOptions -) +IngestionPipelineDefinitionFanoutOptionsParam = IngestionPipelineDefinitionFanoutOptionsDict | IngestionPipelineDefinitionFanoutOptions diff --git a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_table_specific_config_query_based_connector_config.py b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_table_specific_config_query_based_connector_config.py index a784bb6e8b4..183a0e7e6b7 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_table_specific_config_query_based_connector_config.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_table_specific_config_query_based_connector_config.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig: """ Configurations that are only applicable for query-based ingestion connectors. """ - cursor_columns: VariableOrList[str] = field(default_factory=list) """ [Public Preview] The names of the monotonically increasing columns in the source table that are used to enable @@ -53,21 +178,15 @@ class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig: """ @classmethod - def from_dict( - cls, - value: "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict", - ) -> "Self": + def from_dict(cls, value: 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict') -> 'Self': return _transform(cls, value) - def as_dict( - self, - ) -> "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict': + return _transform_to_json_value(self) # type:ignore + -class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict( - TypedDict, total=False -): +class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict(TypedDict, total=False): """""" cursor_columns: VariableOrList[str] @@ -106,7 +225,4 @@ class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDic """ -IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam = ( - IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict - | IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig -) +IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam = IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict | IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig diff --git a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_workday_report_parameters.py b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_workday_report_parameters.py index a819adcf236..4452dc0d37b 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_workday_report_parameters.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_workday_report_parameters.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrDict +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class IngestionPipelineDefinitionWorkdayReportParameters: """ :meta private: [EXPERIMENTAL] """ - parameters: VariableOrDict[str] = field(default_factory=dict) """ :meta private: [EXPERIMENTAL] @@ -31,13 +156,12 @@ class IngestionPipelineDefinitionWorkdayReportParameters: """ @classmethod - def from_dict( - cls, value: "IngestionPipelineDefinitionWorkdayReportParametersDict" - ) -> "Self": + def from_dict(cls, value: 'IngestionPipelineDefinitionWorkdayReportParametersDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "IngestionPipelineDefinitionWorkdayReportParametersDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'IngestionPipelineDefinitionWorkdayReportParametersDict': + return _transform_to_json_value(self) # type:ignore + class IngestionPipelineDefinitionWorkdayReportParametersDict(TypedDict, total=False): @@ -57,7 +181,4 @@ class IngestionPipelineDefinitionWorkdayReportParametersDict(TypedDict, total=Fa """ -IngestionPipelineDefinitionWorkdayReportParametersParam = ( - IngestionPipelineDefinitionWorkdayReportParametersDict - | IngestionPipelineDefinitionWorkdayReportParameters -) +IngestionPipelineDefinitionWorkdayReportParametersParam = IngestionPipelineDefinitionWorkdayReportParametersDict | IngestionPipelineDefinitionWorkdayReportParameters diff --git a/python/databricks/bundles/pipelines/_models/init_script_info.py b/python/databricks/bundles/pipelines/_models/init_script_info.py index f6c696d11bd..0d8eda0cc7e 100644 --- a/python/databricks/bundles/pipelines/_models/init_script_info.py +++ b/python/databricks/bundles/pipelines/_models/init_script_info.py @@ -1,50 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.adlsgen2_info import ( - Adlsgen2Info, - Adlsgen2InfoParam, -) -from databricks.bundles.pipelines._models.dbfs_storage_info import ( - DbfsStorageInfo, - DbfsStorageInfoParam, -) -from databricks.bundles.pipelines._models.gcs_storage_info import ( - GcsStorageInfo, - GcsStorageInfoParam, -) -from databricks.bundles.pipelines._models.local_file_info import ( - LocalFileInfo, - LocalFileInfoParam, -) -from databricks.bundles.pipelines._models.s3_storage_info import ( - S3StorageInfo, - S3StorageInfoParam, -) -from databricks.bundles.pipelines._models.volumes_storage_info import ( - VolumesStorageInfo, - VolumesStorageInfoParam, -) -from databricks.bundles.pipelines._models.workspace_storage_info import ( - WorkspaceStorageInfo, - WorkspaceStorageInfoParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class InitScriptInfo: """ Config for an individual init script """ - abfss: VariableOrOptional[Adlsgen2Info] = None """ Contains the Azure Data Lake Storage destination path @@ -89,11 +186,12 @@ class InitScriptInfo: """ @classmethod - def from_dict(cls, value: "InitScriptInfoDict") -> "Self": + def from_dict(cls, value: 'InitScriptInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "InitScriptInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'InitScriptInfoDict': + return _transform_to_json_value(self) # type:ignore + class InitScriptInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/jira_connector_options.py b/python/databricks/bundles/pipelines/_models/jira_connector_options.py index ef7ba677587..5eff3ae1c58 100644 --- a/python/databricks/bundles/pipelines/_models/jira_connector_options.py +++ b/python/databricks/bundles/pipelines/_models/jira_connector_options.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JiraConnectorOptions: """ Jira specific options for ingestion """ - include_jira_spaces: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -25,11 +150,12 @@ class JiraConnectorOptions: """ @classmethod - def from_dict(cls, value: "JiraConnectorOptionsDict") -> "Self": + def from_dict(cls, value: 'JiraConnectorOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JiraConnectorOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JiraConnectorOptionsDict': + return _transform_to_json_value(self) # type:ignore + class JiraConnectorOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/json_transformer_options.py b/python/databricks/bundles/pipelines/_models/json_transformer_options.py index 8359f564eb2..aec0bd3b209 100644 --- a/python/databricks/bundles/pipelines/_models/json_transformer_options.py +++ b/python/databricks/bundles/pipelines/_models/json_transformer_options.py @@ -1,20 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import ( - FileIngestionOptionsSchemaEvolutionMode, - FileIngestionOptionsSchemaEvolutionModeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class JsonTransformerOptions: """""" @@ -33,9 +155,7 @@ class JsonTransformerOptions: [Beta] Inline schema string for JSON parsing (Spark DDL format). """ - schema_evolution_mode: VariableOrOptional[ - FileIngestionOptionsSchemaEvolutionMode - ] = None + schema_evolution_mode: VariableOrOptional[FileIngestionOptionsSchemaEvolutionMode] = None """ :meta private: [EXPERIMENTAL] @@ -57,11 +177,12 @@ class JsonTransformerOptions: """ @classmethod - def from_dict(cls, value: "JsonTransformerOptionsDict") -> "Self": + def from_dict(cls, value: 'JsonTransformerOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "JsonTransformerOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'JsonTransformerOptionsDict': + return _transform_to_json_value(self) # type:ignore + class JsonTransformerOptionsDict(TypedDict, total=False): @@ -81,9 +202,7 @@ class JsonTransformerOptionsDict(TypedDict, total=False): [Beta] Inline schema string for JSON parsing (Spark DDL format). """ - schema_evolution_mode: VariableOrOptional[ - FileIngestionOptionsSchemaEvolutionModeParam - ] + schema_evolution_mode: VariableOrOptional[FileIngestionOptionsSchemaEvolutionModeParam] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/kafka_options.py b/python/databricks/bundles/pipelines/_models/kafka_options.py index d42b585624d..30c4d017e3a 100644 --- a/python/databricks/bundles/pipelines/_models/kafka_options.py +++ b/python/databricks/bundles/pipelines/_models/kafka_options.py @@ -1,24 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.pipelines._models.transformer import ( - Transformer, - TransformerParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class KafkaOptions: """""" @@ -80,11 +198,12 @@ class KafkaOptions: """ @classmethod - def from_dict(cls, value: "KafkaOptionsDict") -> "Self": + def from_dict(cls, value: 'KafkaOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "KafkaOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'KafkaOptionsDict': + return _transform_to_json_value(self) # type:ignore + class KafkaOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/lifecycle.py b/python/databricks/bundles/pipelines/_models/lifecycle.py index 697776a198e..de0d6e7ddfc 100644 --- a/python/databricks/bundles/pipelines/_models/lifecycle.py +++ b/python/databricks/bundles/pipelines/_models/lifecycle.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +147,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options.py index 1d997abc919..e012d648407 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options.py @@ -1,34 +1,153 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import ( - LinkedInAdsOptionsLinkedInAdsCustomReportOptions, - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LinkedInAdsOptions: """ :meta private: [EXPERIMENTAL] - + LinkedIn Ads specific options for ingestion. sync_start_date and lookback_window_days apply to both the prebuilt analytics tables and custom reports. custom_report_options defines a custom (user-defined) adAnalytics report and is only valid on a table object. """ - - custom_report_options: VariableOrOptional[ - LinkedInAdsOptionsLinkedInAdsCustomReportOptions - ] = None + custom_report_options: VariableOrOptional[LinkedInAdsOptionsLinkedInAdsCustomReportOptions] = None """ :meta private: [EXPERIMENTAL] @@ -58,19 +177,18 @@ class LinkedInAdsOptions: """ @classmethod - def from_dict(cls, value: "LinkedInAdsOptionsDict") -> "Self": + def from_dict(cls, value: 'LinkedInAdsOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LinkedInAdsOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LinkedInAdsOptionsDict': + return _transform_to_json_value(self) # type:ignore + class LinkedInAdsOptionsDict(TypedDict, total=False): """""" - custom_report_options: VariableOrOptional[ - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam - ] + custom_report_options: VariableOrOptional[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options.py index 1420eb50d50..5abb7047fb2 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options.py @@ -1,55 +1,160 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import ( - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam, -) -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import ( - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam, -) -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import ( - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LinkedInAdsOptionsLinkedInAdsCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + User-defined custom report for the LinkedIn Ads connector. The destination table name comes from the enclosing TableSpec.destination_table, the start date from the enclosing LinkedInAdsOptions.sync_start_date, and the account it runs against from the source schema (namespace) -- none are repeated here. """ - - finder: VariableOr[ - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder - ] + finder: VariableOr[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder] """ :meta private: [EXPERIMENTAL] [Private Preview] (Required) adAnalytics finder. See LinkedInAdsFinder. """ - entity_granularity: VariableOrList[ - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity - ] = field(default_factory=list) + entity_granularity: VariableOrList[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -69,9 +174,7 @@ class LinkedInAdsOptionsLinkedInAdsCustomReportOptions: RevenueAttributionMetrics struct). """ - time_granularity: VariableOrOptional[ - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity - ] = None + time_granularity: VariableOrOptional[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity] = None """ :meta private: [EXPERIMENTAL] @@ -80,30 +183,25 @@ class LinkedInAdsOptionsLinkedInAdsCustomReportOptions: """ @classmethod - def from_dict( - cls, value: "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict" - ) -> "Self": + def from_dict(cls, value: 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict': + return _transform_to_json_value(self) # type:ignore + class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict(TypedDict, total=False): """""" - finder: VariableOr[ - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - ] + finder: VariableOr[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam] """ :meta private: [EXPERIMENTAL] [Private Preview] (Required) adAnalytics finder. See LinkedInAdsFinder. """ - entity_granularity: VariableOrList[ - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - ] + entity_granularity: VariableOrList[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam] """ :meta private: [EXPERIMENTAL] @@ -123,9 +221,7 @@ class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict(TypedDict, total=Fals RevenueAttributionMetrics struct). """ - time_granularity: VariableOrOptional[ - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - ] + time_granularity: VariableOrOptional[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam] """ :meta private: [EXPERIMENTAL] @@ -134,7 +230,4 @@ class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict(TypedDict, total=Fals """ -LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam = ( - LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict - | LinkedInAdsOptionsLinkedInAdsCustomReportOptions -) +LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam = LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict | LinkedInAdsOptionsLinkedInAdsCustomReportOptions diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity.py index ae32b0b83a4..e737ceaaa7f 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity.py @@ -1,24 +1,146 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict -class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity( - Enum -): +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + + +class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity(Enum): """ :meta private: [EXPERIMENTAL] - + Entity pivot to group by. """ - CAMPAIGN = "CAMPAIGN" CREATIVE = "CREATIVE" CAMPAIGN_GROUP = "CAMPAIGN_GROUP" - -LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam = ( - Literal["CAMPAIGN", "CREATIVE", "CAMPAIGN_GROUP"] - | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity -) +LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam = Literal["CAMPAIGN", "CREATIVE", "CAMPAIGN_GROUP"] | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder.py index 9abf3e9d7be..239eee96e80 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder.py @@ -1,23 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder(Enum): """ :meta private: [EXPERIMENTAL] - + adAnalytics finder. Determines call shape, valid pivots, and metric requirements. """ - ANALYTICS = "ANALYTICS" STATISTICS = "STATISTICS" ATTRIBUTED_REVENUE_METRICS = "ATTRIBUTED_REVENUE_METRICS" - -LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam = ( - Literal["ANALYTICS", "STATISTICS", "ATTRIBUTED_REVENUE_METRICS"] - | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder -) +LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam = Literal["ANALYTICS", "STATISTICS", "ATTRIBUTED_REVENUE_METRICS"] | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity.py index 256974843ea..5ef329b2449 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity.py @@ -1,24 +1,148 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity(Enum): """ :meta private: [EXPERIMENTAL] - + Time aggregation. Used by analytics/statistics; ignored for attributedRevenueMetrics. Defaults to DAILY when unspecified. """ - ALL = "ALL" DAILY = "DAILY" MONTHLY = "MONTHLY" YEARLY = "YEARLY" - -LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam = ( - Literal["ALL", "DAILY", "MONTHLY", "YEARLY"] - | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity -) +LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam = Literal["ALL", "DAILY", "MONTHLY", "YEARLY"] | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity diff --git a/python/databricks/bundles/pipelines/_models/local_file_info.py b/python/databricks/bundles/pipelines/_models/local_file_info.py index 875f2fe8353..09965373829 100644 --- a/python/databricks/bundles/pipelines/_models/local_file_info.py +++ b/python/databricks/bundles/pipelines/_models/local_file_info.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LocalFileInfo: """""" @@ -21,11 +147,12 @@ class LocalFileInfo: """ @classmethod - def from_dict(cls, value: "LocalFileInfoDict") -> "Self": + def from_dict(cls, value: 'LocalFileInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LocalFileInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LocalFileInfoDict': + return _transform_to_json_value(self) # type:ignore + class LocalFileInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/log_analytics_info.py b/python/databricks/bundles/pipelines/_models/log_analytics_info.py index 67cdf85b2b9..e4ab2bcbb48 100644 --- a/python/databricks/bundles/pipelines/_models/log_analytics_info.py +++ b/python/databricks/bundles/pipelines/_models/log_analytics_info.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LogAnalyticsInfo: """""" @@ -26,11 +152,12 @@ class LogAnalyticsInfo: """ @classmethod - def from_dict(cls, value: "LogAnalyticsInfoDict") -> "Self": + def from_dict(cls, value: 'LogAnalyticsInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LogAnalyticsInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LogAnalyticsInfoDict': + return _transform_to_json_value(self) # type:ignore + class LogAnalyticsInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/marketo_options.py b/python/databricks/bundles/pipelines/_models/marketo_options.py index 943dba4ff7b..14776f340fb 100644 --- a/python/databricks/bundles/pipelines/_models/marketo_options.py +++ b/python/databricks/bundles/pipelines/_models/marketo_options.py @@ -1,24 +1,149 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MarketoOptions: """ :meta private: [EXPERIMENTAL] - + Marketo specific options for ingestion """ - sync_start_date: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -29,11 +154,12 @@ class MarketoOptions: """ @classmethod - def from_dict(cls, value: "MarketoOptionsDict") -> "Self": + def from_dict(cls, value: 'MarketoOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MarketoOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MarketoOptionsDict': + return _transform_to_json_value(self) # type:ignore + class MarketoOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/maven_library.py b/python/databricks/bundles/pipelines/_models/maven_library.py index 25be1599c28..97368254302 100644 --- a/python/databricks/bundles/pipelines/_models/maven_library.py +++ b/python/databricks/bundles/pipelines/_models/maven_library.py @@ -1,20 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MavenLibrary: """""" @@ -39,11 +161,12 @@ class MavenLibrary: """ @classmethod - def from_dict(cls, value: "MavenLibraryDict") -> "Self": + def from_dict(cls, value: 'MavenLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MavenLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MavenLibraryDict': + return _transform_to_json_value(self) # type:ignore + class MavenLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/meta_marketing_options.py b/python/databricks/bundles/pipelines/_models/meta_marketing_options.py index 9a9f657bf5f..e87a1e8165c 100644 --- a/python/databricks/bundles/pipelines/_models/meta_marketing_options.py +++ b/python/databricks/bundles/pipelines/_models/meta_marketing_options.py @@ -1,26 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import ( - MetaMarketingOptionsMetaMarketingCustomReportOptions, - MetaMarketingOptionsMetaMarketingCustomReportOptionsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MetaMarketingOptions: """ Meta Marketing (Meta Ads) specific options for ingestion """ - action_attribution_windows: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -59,9 +180,7 @@ class MetaMarketingOptions: updated conversion data from the API, shared by prebuilt and custom reports. """ - custom_report_options: VariableOrOptional[ - MetaMarketingOptionsMetaMarketingCustomReportOptions - ] = None + custom_report_options: VariableOrOptional[MetaMarketingOptionsMetaMarketingCustomReportOptions] = None """ :meta private: [EXPERIMENTAL] @@ -95,11 +214,12 @@ class MetaMarketingOptions: """ @classmethod - def from_dict(cls, value: "MetaMarketingOptionsDict") -> "Self": + def from_dict(cls, value: 'MetaMarketingOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MetaMarketingOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MetaMarketingOptionsDict': + return _transform_to_json_value(self) # type:ignore + class MetaMarketingOptionsDict(TypedDict, total=False): @@ -143,9 +263,7 @@ class MetaMarketingOptionsDict(TypedDict, total=False): updated conversion data from the API, shared by prebuilt and custom reports. """ - custom_report_options: VariableOrOptional[ - MetaMarketingOptionsMetaMarketingCustomReportOptionsParam - ] + custom_report_options: VariableOrOptional[MetaMarketingOptionsMetaMarketingCustomReportOptionsParam] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/meta_marketing_options_meta_marketing_custom_report_options.py b/python/databricks/bundles/pipelines/_models/meta_marketing_options_meta_marketing_custom_report_options.py index a01a24ec45b..9d9d0fda95b 100644 --- a/python/databricks/bundles/pipelines/_models/meta_marketing_options_meta_marketing_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/meta_marketing_options_meta_marketing_custom_report_options.py @@ -1,26 +1,151 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MetaMarketingOptionsMetaMarketingCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + Defines the shape of a single Meta Ads custom report (one /insights call shape). start_date, custom_insights_lookback_window live on MetaMarketingOptions, not here. Metrics are not customer-selectable; the connector returns a fixed standard metric set. """ - action_attribution_windows: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -64,13 +189,12 @@ class MetaMarketingOptionsMetaMarketingCustomReportOptions: """ @classmethod - def from_dict( - cls, value: "MetaMarketingOptionsMetaMarketingCustomReportOptionsDict" - ) -> "Self": + def from_dict(cls, value: 'MetaMarketingOptionsMetaMarketingCustomReportOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MetaMarketingOptionsMetaMarketingCustomReportOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MetaMarketingOptionsMetaMarketingCustomReportOptionsDict': + return _transform_to_json_value(self) # type:ignore + class MetaMarketingOptionsMetaMarketingCustomReportOptionsDict(TypedDict, total=False): @@ -119,7 +243,4 @@ class MetaMarketingOptionsMetaMarketingCustomReportOptionsDict(TypedDict, total= """ -MetaMarketingOptionsMetaMarketingCustomReportOptionsParam = ( - MetaMarketingOptionsMetaMarketingCustomReportOptionsDict - | MetaMarketingOptionsMetaMarketingCustomReportOptions -) +MetaMarketingOptionsMetaMarketingCustomReportOptionsParam = MetaMarketingOptionsMetaMarketingCustomReportOptionsDict | MetaMarketingOptionsMetaMarketingCustomReportOptions diff --git a/python/databricks/bundles/pipelines/_models/notebook_library.py b/python/databricks/bundles/pipelines/_models/notebook_library.py index 72ded094151..503014c0709 100644 --- a/python/databricks/bundles/pipelines/_models/notebook_library.py +++ b/python/databricks/bundles/pipelines/_models/notebook_library.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class NotebookLibrary: """""" @@ -21,11 +147,12 @@ class NotebookLibrary: """ @classmethod - def from_dict(cls, value: "NotebookLibraryDict") -> "Self": + def from_dict(cls, value: 'NotebookLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "NotebookLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'NotebookLibraryDict': + return _transform_to_json_value(self) # type:ignore + class NotebookLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/notifications.py b/python/databricks/bundles/pipelines/_models/notifications.py index 271bb4c1fdb..47b12875b76 100644 --- a/python/databricks/bundles/pipelines/_models/notifications.py +++ b/python/databricks/bundles/pipelines/_models/notifications.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Notifications: """""" @@ -32,11 +158,12 @@ class Notifications: """ @classmethod - def from_dict(cls, value: "NotificationsDict") -> "Self": + def from_dict(cls, value: 'NotificationsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "NotificationsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'NotificationsDict': + return _transform_to_json_value(self) # type:ignore + class NotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/operation_time_window.py b/python/databricks/bundles/pipelines/_models/operation_time_window.py index cc3c315d6de..a7dd7dce6d1 100644 --- a/python/databricks/bundles/pipelines/_models/operation_time_window.py +++ b/python/databricks/bundles/pipelines/_models/operation_time_window.py @@ -1,27 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class OperationTimeWindow: """ Proto representing a window """ - start_hour: VariableOr[int] """ [Public Preview] An integer between 0 and 23 denoting the start hour for the window in the 24-hour day. @@ -40,11 +160,12 @@ class OperationTimeWindow: """ @classmethod - def from_dict(cls, value: "OperationTimeWindowDict") -> "Self": + def from_dict(cls, value: 'OperationTimeWindowDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "OperationTimeWindowDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'OperationTimeWindowDict': + return _transform_to_json_value(self) # type:ignore + class OperationTimeWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/outlook_attachment_mode.py b/python/databricks/bundles/pipelines/_models/outlook_attachment_mode.py index 5d3c5fa042f..98a0b153d21 100644 --- a/python/databricks/bundles/pipelines/_models/outlook_attachment_mode.py +++ b/python/databricks/bundles/pipelines/_models/outlook_attachment_mode.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class OutlookAttachmentMode(Enum): """ :meta private: [EXPERIMENTAL] - + Attachment behavior mode for Outlook ingestion """ - ALL = "ALL" NON_INLINE_ONLY = "NON_INLINE_ONLY" INLINE_ONLY = "INLINE_ONLY" NONE = "NONE" - -OutlookAttachmentModeParam = ( - Literal["ALL", "NON_INLINE_ONLY", "INLINE_ONLY", "NONE"] | OutlookAttachmentMode -) +OutlookAttachmentModeParam = Literal["ALL", "NON_INLINE_ONLY", "INLINE_ONLY", "NONE"] | OutlookAttachmentMode diff --git a/python/databricks/bundles/pipelines/_models/outlook_body_format.py b/python/databricks/bundles/pipelines/_models/outlook_body_format.py index e5f409e0412..c94bcf6e14e 100644 --- a/python/databricks/bundles/pipelines/_models/outlook_body_format.py +++ b/python/databricks/bundles/pipelines/_models/outlook_body_format.py @@ -1,18 +1,145 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class OutlookBodyFormat(Enum): """ :meta private: [EXPERIMENTAL] - + Body format for Outlook email content """ - TEXT_HTML = "TEXT_HTML" TEXT_PLAIN = "TEXT_PLAIN" - OutlookBodyFormatParam = Literal["TEXT_HTML", "TEXT_PLAIN"] | OutlookBodyFormat diff --git a/python/databricks/bundles/pipelines/_models/outlook_options.py b/python/databricks/bundles/pipelines/_models/outlook_options.py index 98e93a16a69..63247123aa9 100644 --- a/python/databricks/bundles/pipelines/_models/outlook_options.py +++ b/python/databricks/bundles/pipelines/_models/outlook_options.py @@ -1,32 +1,149 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.pipelines._models.outlook_attachment_mode import ( - OutlookAttachmentMode, - OutlookAttachmentModeParam, -) -from databricks.bundles.pipelines._models.outlook_body_format import ( - OutlookBodyFormat, - OutlookBodyFormatParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class OutlookOptions: """ :meta private: [EXPERIMENTAL] - + Outlook specific options for ingestion """ - attachment_mode: VariableOrOptional[OutlookAttachmentMode] = None """ :meta private: [EXPERIMENTAL] @@ -95,11 +212,12 @@ class OutlookOptions: """ @classmethod - def from_dict(cls, value: "OutlookOptionsDict") -> "Self": + def from_dict(cls, value: 'OutlookOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "OutlookOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'OutlookOptionsDict': + return _transform_to_json_value(self) # type:ignore + class OutlookOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/parse_mode.py b/python/databricks/bundles/pipelines/_models/parse_mode.py index 2e55f31226d..73726d967a7 100644 --- a/python/databricks/bundles/pipelines/_models/parse_mode.py +++ b/python/databricks/bundles/pipelines/_models/parse_mode.py @@ -1,16 +1,143 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class ParseMode(Enum): """ Determines how errors encountered while deserializing records are handled. """ - FAILFAST = "FAILFAST" PERMISSIVE = "PERMISSIVE" - ParseModeParam = Literal["FAILFAST", "PERMISSIVE"] | ParseMode diff --git a/python/databricks/bundles/pipelines/_models/path_pattern.py b/python/databricks/bundles/pipelines/_models/path_pattern.py index 6df8c4c5a3f..579e10436ec 100644 --- a/python/databricks/bundles/pipelines/_models/path_pattern.py +++ b/python/databricks/bundles/pipelines/_models/path_pattern.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PathPattern: """""" @@ -21,11 +147,12 @@ class PathPattern: """ @classmethod - def from_dict(cls, value: "PathPatternDict") -> "Self": + def from_dict(cls, value: 'PathPatternDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PathPatternDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PathPatternDict': + return _transform_to_json_value(self) # type:ignore + class PathPatternDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline.py b/python/databricks/bundles/pipelines/_models/pipeline.py index 5e4b04bb726..676a4f511fb 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline.py +++ b/python/databricks/bundles/pipelines/_models/pipeline.py @@ -1,63 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.pipelines._models.event_log_spec import ( - EventLogSpec, - EventLogSpecParam, -) -from databricks.bundles.pipelines._models.filters import ( - Filters, - FiltersParam, -) -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import ( - IngestionGatewayPipelineDefinition, - IngestionGatewayPipelineDefinitionParam, -) -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import ( - IngestionPipelineDefinition, - IngestionPipelineDefinitionParam, -) -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.pipelines._models.notifications import ( - Notifications, - NotificationsParam, -) -from databricks.bundles.pipelines._models.pipeline_cluster import ( - PipelineCluster, - PipelineClusterParam, -) -from databricks.bundles.pipelines._models.pipeline_library import ( - PipelineLibrary, - PipelineLibraryParam, -) -from databricks.bundles.pipelines._models.pipeline_permission import ( - PipelinePermission, - PipelinePermissionParam, -) -from databricks.bundles.pipelines._models.pipelines_environment import ( - PipelinesEnvironment, - PipelinesEnvironmentParam, -) -from databricks.bundles.pipelines._models.restart_window import ( - RestartWindow, - RestartWindowParam, -) -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Pipeline(Resource): """""" @@ -265,11 +344,12 @@ class Pipeline(Resource): """ @classmethod - def from_dict(cls, value: "PipelineDict") -> "Self": + def from_dict(cls, value: 'PipelineDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PipelineDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PipelineDict': + return _transform_to_json_value(self) # type:ignore + class PipelineDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_cluster.py b/python/databricks/bundles/pipelines/_models/pipeline_cluster.py index 6eda9e865a0..18d91182a27 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_cluster.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_cluster.py @@ -1,44 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.pipelines._models.aws_attributes import ( - AwsAttributes, - AwsAttributesParam, -) -from databricks.bundles.pipelines._models.azure_attributes import ( - AzureAttributes, - AzureAttributesParam, -) -from databricks.bundles.pipelines._models.cluster_log_conf import ( - ClusterLogConf, - ClusterLogConfParam, -) -from databricks.bundles.pipelines._models.gcp_attributes import ( - GcpAttributes, - GcpAttributesParam, -) -from databricks.bundles.pipelines._models.init_script_info import ( - InitScriptInfo, - InitScriptInfoParam, -) -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import ( - PipelineClusterAutoscale, - PipelineClusterAutoscaleParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PipelineCluster: """""" @@ -179,11 +277,12 @@ class PipelineCluster: """ @classmethod - def from_dict(cls, value: "PipelineClusterDict") -> "Self": + def from_dict(cls, value: 'PipelineClusterDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PipelineClusterDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PipelineClusterDict': + return _transform_to_json_value(self) # type:ignore + class PipelineClusterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale.py b/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale.py index d5ba9da3665..7ab741637fe 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale.py @@ -1,20 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import ( - PipelineClusterAutoscaleMode, - PipelineClusterAutoscaleModeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PipelineClusterAutoscale: """""" @@ -40,11 +162,12 @@ class PipelineClusterAutoscale: """ @classmethod - def from_dict(cls, value: "PipelineClusterAutoscaleDict") -> "Self": + def from_dict(cls, value: 'PipelineClusterAutoscaleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PipelineClusterAutoscaleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PipelineClusterAutoscaleDict': + return _transform_to_json_value(self) # type:ignore + class PipelineClusterAutoscaleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale_mode.py b/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale_mode.py index 52afa7d09d1..09bf9e1c806 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale_mode.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale_mode.py @@ -1,7 +1,136 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class PipelineClusterAutoscaleMode(Enum): @@ -12,11 +141,7 @@ class PipelineClusterAutoscaleMode(Enum): for `updates` clusters only. The legacy autoscaling feature is used for `maintenance` clusters. """ - ENHANCED = "ENHANCED" LEGACY = "LEGACY" - -PipelineClusterAutoscaleModeParam = ( - Literal["ENHANCED", "LEGACY"] | PipelineClusterAutoscaleMode -) +PipelineClusterAutoscaleModeParam = Literal["ENHANCED", "LEGACY"] | PipelineClusterAutoscaleMode diff --git a/python/databricks/bundles/pipelines/_models/pipeline_library.py b/python/databricks/bundles/pipelines/_models/pipeline_library.py index 5639f57fb2f..4f5fb7d4787 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_library.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_library.py @@ -1,32 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.file_library import ( - FileLibrary, - FileLibraryParam, -) -from databricks.bundles.pipelines._models.maven_library import ( - MavenLibrary, - MavenLibraryParam, -) -from databricks.bundles.pipelines._models.notebook_library import ( - NotebookLibrary, - NotebookLibraryParam, -) -from databricks.bundles.pipelines._models.path_pattern import ( - PathPattern, - PathPatternParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PipelineLibrary: """""" @@ -63,11 +173,12 @@ class PipelineLibrary: """ @classmethod - def from_dict(cls, value: "PipelineLibraryDict") -> "Self": + def from_dict(cls, value: 'PipelineLibraryDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PipelineLibraryDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PipelineLibraryDict': + return _transform_to_json_value(self) # type:ignore + class PipelineLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_permission.py b/python/databricks/bundles/pipelines/_models/pipeline_permission.py index e7b9c90f94e..ea2809a4879 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_permission.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_permission.py @@ -1,20 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.pipelines._models.pipeline_permission_level import ( - PipelinePermissionLevel, - PipelinePermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PipelinePermission: """""" @@ -40,11 +162,12 @@ class PipelinePermission: """ @classmethod - def from_dict(cls, value: "PipelinePermissionDict") -> "Self": + def from_dict(cls, value: 'PipelinePermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PipelinePermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PipelinePermissionDict': + return _transform_to_json_value(self) # type:ignore + class PipelinePermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_permission_level.py b/python/databricks/bundles/pipelines/_models/pipeline_permission_level.py index 5c2d68bfe57..02e1a4df38e 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_permission_level.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_permission_level.py @@ -1,20 +1,145 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class PipelinePermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" IS_OWNER = "IS_OWNER" CAN_RUN = "CAN_RUN" CAN_VIEW = "CAN_VIEW" - -PipelinePermissionLevelParam = ( - Literal["CAN_MANAGE", "IS_OWNER", "CAN_RUN", "CAN_VIEW"] | PipelinePermissionLevel -) +PipelinePermissionLevelParam = Literal["CAN_MANAGE", "IS_OWNER", "CAN_RUN", "CAN_VIEW"] | PipelinePermissionLevel diff --git a/python/databricks/bundles/pipelines/_models/pipelines_environment.py b/python/databricks/bundles/pipelines/_models/pipelines_environment.py index 0493c5d4fa7..14db0afc3dc 100644 --- a/python/databricks/bundles/pipelines/_models/pipelines_environment.py +++ b/python/databricks/bundles/pipelines/_models/pipelines_environment.py @@ -1,23 +1,148 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PipelinesEnvironment: """ The environment entity used to preserve serverless environment side panel, jobs' environment for non-notebook task, and SDP's environment for classic and serverless pipelines. In this minimal environment spec, only pip dependencies are supported. """ - dependencies: VariableOrList[str] = field(default_factory=list) """ [Public Preview] List of pip dependencies, as supported by the version of pip in this environment. @@ -40,11 +165,12 @@ class PipelinesEnvironment: """ @classmethod - def from_dict(cls, value: "PipelinesEnvironmentDict") -> "Self": + def from_dict(cls, value: 'PipelinesEnvironmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PipelinesEnvironmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PipelinesEnvironmentDict': + return _transform_to_json_value(self) # type:ignore + class PipelinesEnvironmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/postgres_catalog_config.py b/python/databricks/bundles/pipelines/_models/postgres_catalog_config.py index 0bc642e8a5a..ef3e7f7b64c 100644 --- a/python/databricks/bundles/pipelines/_models/postgres_catalog_config.py +++ b/python/databricks/bundles/pipelines/_models/postgres_catalog_config.py @@ -1,37 +1,159 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.postgres_slot_config import ( - PostgresSlotConfig, - PostgresSlotConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PostgresCatalogConfig: """ PG-specific catalog-level configuration parameters """ - slot_config: VariableOrOptional[PostgresSlotConfig] = None """ [Public Preview] Optional. The Postgres slot configuration to use for logical replication """ @classmethod - def from_dict(cls, value: "PostgresCatalogConfigDict") -> "Self": + def from_dict(cls, value: 'PostgresCatalogConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PostgresCatalogConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PostgresCatalogConfigDict': + return _transform_to_json_value(self) # type:ignore + class PostgresCatalogConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/postgres_slot_config.py b/python/databricks/bundles/pipelines/_models/postgres_slot_config.py index d6eaeacbb08..a629f1905c5 100644 --- a/python/databricks/bundles/pipelines/_models/postgres_slot_config.py +++ b/python/databricks/bundles/pipelines/_models/postgres_slot_config.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PostgresSlotConfig: """ PostgresSlotConfig contains the configuration for a Postgres logical replication slot """ - publication_name: VariableOrOptional[str] = None """ [Public Preview] The name of the publication to use for the Postgres source @@ -28,11 +153,12 @@ class PostgresSlotConfig: """ @classmethod - def from_dict(cls, value: "PostgresSlotConfigDict") -> "Self": + def from_dict(cls, value: 'PostgresSlotConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PostgresSlotConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PostgresSlotConfigDict': + return _transform_to_json_value(self) # type:ignore + class PostgresSlotConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/protobuf_transformer_options.py b/python/databricks/bundles/pipelines/_models/protobuf_transformer_options.py index cace688c4c9..3eb7538c530 100644 --- a/python/databricks/bundles/pipelines/_models/protobuf_transformer_options.py +++ b/python/databricks/bundles/pipelines/_models/protobuf_transformer_options.py @@ -1,21 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam -from databricks.bundles.pipelines._models.schema_registry_config import ( - SchemaRegistryConfig, - SchemaRegistryConfigParam, -) + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ProtobufTransformerOptions: """""" @@ -61,11 +182,12 @@ class ProtobufTransformerOptions: """ @classmethod - def from_dict(cls, value: "ProtobufTransformerOptionsDict") -> "Self": + def from_dict(cls, value: 'ProtobufTransformerOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ProtobufTransformerOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ProtobufTransformerOptionsDict': + return _transform_to_json_value(self) # type:ignore + class ProtobufTransformerOptionsDict(TypedDict, total=False): @@ -112,6 +234,4 @@ class ProtobufTransformerOptionsDict(TypedDict, total=False): """ -ProtobufTransformerOptionsParam = ( - ProtobufTransformerOptionsDict | ProtobufTransformerOptions -) +ProtobufTransformerOptionsParam = ProtobufTransformerOptionsDict | ProtobufTransformerOptions diff --git a/python/databricks/bundles/pipelines/_models/rabbitmq_options.py b/python/databricks/bundles/pipelines/_models/rabbitmq_options.py index a08a4a0613a..f2c31512535 100644 --- a/python/databricks/bundles/pipelines/_models/rabbitmq_options.py +++ b/python/databricks/bundles/pipelines/_models/rabbitmq_options.py @@ -1,16 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RabbitmqOptions: """ @@ -19,7 +145,6 @@ class RabbitmqOptions: are intentionally not exposed in the public API. The managed connector uses sensible defaults internally. These can be added later if user demand arises. """ - queue: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -28,11 +153,12 @@ class RabbitmqOptions: """ @classmethod - def from_dict(cls, value: "RabbitmqOptionsDict") -> "Self": + def from_dict(cls, value: 'RabbitmqOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RabbitmqOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RabbitmqOptionsDict': + return _transform_to_json_value(self) # type:ignore + class RabbitmqOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/reddit_ads_options.py b/python/databricks/bundles/pipelines/_models/reddit_ads_options.py index 3e39958b026..77a3afb6b63 100644 --- a/python/databricks/bundles/pipelines/_models/reddit_ads_options.py +++ b/python/databricks/bundles/pipelines/_models/reddit_ads_options.py @@ -1,31 +1,150 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import ( - RedditAdsOptionsRedditAdsCustomReportOptions, - RedditAdsOptionsRedditAdsCustomReportOptionsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RedditAdsOptions: """ :meta private: [EXPERIMENTAL] - + Reddit Ads specific options for ingestion """ - - custom_report_options: VariableOrOptional[ - RedditAdsOptionsRedditAdsCustomReportOptions - ] = None + custom_report_options: VariableOrOptional[RedditAdsOptionsRedditAdsCustomReportOptions] = None """ :meta private: [EXPERIMENTAL] @@ -53,19 +172,18 @@ class RedditAdsOptions: """ @classmethod - def from_dict(cls, value: "RedditAdsOptionsDict") -> "Self": + def from_dict(cls, value: 'RedditAdsOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RedditAdsOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RedditAdsOptionsDict': + return _transform_to_json_value(self) # type:ignore + class RedditAdsOptionsDict(TypedDict, total=False): """""" - custom_report_options: VariableOrOptional[ - RedditAdsOptionsRedditAdsCustomReportOptionsParam - ] + custom_report_options: VariableOrOptional[RedditAdsOptionsRedditAdsCustomReportOptionsParam] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/reddit_ads_options_reddit_ads_custom_report_options.py b/python/databricks/bundles/pipelines/_models/reddit_ads_options_reddit_ads_custom_report_options.py index 78063a7dcaf..32c383eb5f6 100644 --- a/python/databricks/bundles/pipelines/_models/reddit_ads_options_reddit_ads_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/reddit_ads_options_reddit_ads_custom_report_options.py @@ -1,25 +1,150 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RedditAdsOptionsRedditAdsCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + User-defined custom report for the Reddit Ads connector. Applies only to the custom_report table — prebuilt tables ignore this. """ - breakdowns: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -38,13 +163,12 @@ class RedditAdsOptionsRedditAdsCustomReportOptions: """ @classmethod - def from_dict( - cls, value: "RedditAdsOptionsRedditAdsCustomReportOptionsDict" - ) -> "Self": + def from_dict(cls, value: 'RedditAdsOptionsRedditAdsCustomReportOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RedditAdsOptionsRedditAdsCustomReportOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RedditAdsOptionsRedditAdsCustomReportOptionsDict': + return _transform_to_json_value(self) # type:ignore + class RedditAdsOptionsRedditAdsCustomReportOptionsDict(TypedDict, total=False): @@ -68,7 +192,4 @@ class RedditAdsOptionsRedditAdsCustomReportOptionsDict(TypedDict, total=False): """ -RedditAdsOptionsRedditAdsCustomReportOptionsParam = ( - RedditAdsOptionsRedditAdsCustomReportOptionsDict - | RedditAdsOptionsRedditAdsCustomReportOptions -) +RedditAdsOptionsRedditAdsCustomReportOptionsParam = RedditAdsOptionsRedditAdsCustomReportOptionsDict | RedditAdsOptionsRedditAdsCustomReportOptions diff --git a/python/databricks/bundles/pipelines/_models/report_spec.py b/python/databricks/bundles/pipelines/_models/report_spec.py index c1ae97d014b..90785f46cd8 100644 --- a/python/databricks/bundles/pipelines/_models/report_spec.py +++ b/python/databricks/bundles/pipelines/_models/report_spec.py @@ -1,20 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.pipelines._models.table_specific_config import ( - TableSpecificConfig, - TableSpecificConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ReportSpec: """""" @@ -45,11 +167,12 @@ class ReportSpec: """ @classmethod - def from_dict(cls, value: "ReportSpecDict") -> "Self": + def from_dict(cls, value: 'ReportSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ReportSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ReportSpecDict': + return _transform_to_json_value(self) # type:ignore + class ReportSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/restart_window.py b/python/databricks/bundles/pipelines/_models/restart_window.py index 9987bbe8ec7..ebd9b3a6672 100644 --- a/python/databricks/bundles/pipelines/_models/restart_window.py +++ b/python/databricks/bundles/pipelines/_models/restart_window.py @@ -1,27 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RestartWindow: """ :meta private: [EXPERIMENTAL] """ - start_hour: VariableOr[int] """ :meta private: [EXPERIMENTAL] @@ -47,11 +167,12 @@ class RestartWindow: """ @classmethod - def from_dict(cls, value: "RestartWindowDict") -> "Self": + def from_dict(cls, value: 'RestartWindowDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RestartWindowDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RestartWindowDict': + return _transform_to_json_value(self) # type:ignore + class RestartWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/run_as.py b/python/databricks/bundles/pipelines/_models/run_as.py index 50967e4e1e5..59512af8d53 100644 --- a/python/databricks/bundles/pipelines/_models/run_as.py +++ b/python/databricks/bundles/pipelines/_models/run_as.py @@ -1,24 +1,149 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RunAs: """ Write-only setting, available only in Create/Update calls. Specifies the user or service principal that the pipeline runs as. If not specified, the pipeline runs as the user who created the pipeline. - + Only `user_name` or `service_principal_name` can be specified. If both are specified, an error is thrown. """ - group_name: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -37,11 +162,12 @@ class RunAs: """ @classmethod - def from_dict(cls, value: "RunAsDict") -> "Self": + def from_dict(cls, value: 'RunAsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RunAsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RunAsDict': + return _transform_to_json_value(self) # type:ignore + class RunAsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/s3_storage_info.py b/python/databricks/bundles/pipelines/_models/s3_storage_info.py index 071abd5b6f3..be08638dc6c 100644 --- a/python/databricks/bundles/pipelines/_models/s3_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/s3_storage_info.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class S3StorageInfo: """ A storage location in Amazon S3 """ - destination: VariableOr[str] """ S3 destination, e.g. `s3://my-bucket/some-prefix` Note that logs will be delivered using @@ -64,11 +189,12 @@ class S3StorageInfo: """ @classmethod - def from_dict(cls, value: "S3StorageInfoDict") -> "Self": + def from_dict(cls, value: 'S3StorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "S3StorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'S3StorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class S3StorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/schema_registry_config.py b/python/databricks/bundles/pipelines/_models/schema_registry_config.py index d2617560d7a..ae8401fa98c 100644 --- a/python/databricks/bundles/pipelines/_models/schema_registry_config.py +++ b/python/databricks/bundles/pipelines/_models/schema_registry_config.py @@ -1,20 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ( - ConfluentSchemaRegistryOptions, - ConfluentSchemaRegistryOptionsParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SchemaRegistryConfig: """""" @@ -44,11 +166,12 @@ class SchemaRegistryConfig: """ @classmethod - def from_dict(cls, value: "SchemaRegistryConfigDict") -> "Self": + def from_dict(cls, value: 'SchemaRegistryConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SchemaRegistryConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SchemaRegistryConfigDict': + return _transform_to_json_value(self) # type:ignore + class SchemaRegistryConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/schema_spec.py b/python/databricks/bundles/pipelines/_models/schema_spec.py index d40e976ddcd..af5ac81b184 100644 --- a/python/databricks/bundles/pipelines/_models/schema_spec.py +++ b/python/databricks/bundles/pipelines/_models/schema_spec.py @@ -1,28 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.pipelines._models.connector_options import ( - ConnectorOptions, - ConnectorOptionsParam, -) -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import ( - IngestionPipelineDefinitionFanoutOptions, - IngestionPipelineDefinitionFanoutOptionsParam, -) -from databricks.bundles.pipelines._models.table_specific_config import ( - TableSpecificConfig, - TableSpecificConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SchemaSpec: """""" @@ -70,11 +184,12 @@ class SchemaSpec: """ @classmethod - def from_dict(cls, value: "SchemaSpecDict") -> "Self": + def from_dict(cls, value: 'SchemaSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SchemaSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SchemaSpecDict': + return _transform_to_json_value(self) # type:ignore + class SchemaSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/sharepoint_options.py b/python/databricks/bundles/pipelines/_models/sharepoint_options.py index 4da2754ee21..1178fd2877a 100644 --- a/python/databricks/bundles/pipelines/_models/sharepoint_options.py +++ b/python/databricks/bundles/pipelines/_models/sharepoint_options.py @@ -1,30 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.file_ingestion_options import ( - FileIngestionOptions, - FileIngestionOptionsParam, -) -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import ( - SharepointOptionsSharepointEntityType, - SharepointOptionsSharepointEntityTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SharepointOptions: """ :meta private: [EXPERIMENTAL] """ - entity_type: VariableOrOptional[SharepointOptionsSharepointEntityType] = None """ :meta private: [EXPERIMENTAL] @@ -48,11 +165,12 @@ class SharepointOptions: """ @classmethod - def from_dict(cls, value: "SharepointOptionsDict") -> "Self": + def from_dict(cls, value: 'SharepointOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SharepointOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SharepointOptionsDict': + return _transform_to_json_value(self) # type:ignore + class SharepointOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/sharepoint_options_sharepoint_entity_type.py b/python/databricks/bundles/pipelines/_models/sharepoint_options_sharepoint_entity_type.py index 3b0f8bf6403..280b5e277b8 100644 --- a/python/databricks/bundles/pipelines/_models/sharepoint_options_sharepoint_entity_type.py +++ b/python/databricks/bundles/pipelines/_models/sharepoint_options_sharepoint_entity_type.py @@ -1,21 +1,145 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class SharepointOptionsSharepointEntityType(Enum): """ :meta private: [EXPERIMENTAL] """ - FILE = "FILE" FILE_METADATA = "FILE_METADATA" PERMISSION = "PERMISSION" LIST = "LIST" - -SharepointOptionsSharepointEntityTypeParam = ( - Literal["FILE", "FILE_METADATA", "PERMISSION", "LIST"] - | SharepointOptionsSharepointEntityType -) +SharepointOptionsSharepointEntityTypeParam = Literal["FILE", "FILE_METADATA", "PERMISSION", "LIST"] | SharepointOptionsSharepointEntityType diff --git a/python/databricks/bundles/pipelines/_models/smartsheet_options.py b/python/databricks/bundles/pipelines/_models/smartsheet_options.py index 73691be256b..e12f7bf4026 100644 --- a/python/databricks/bundles/pipelines/_models/smartsheet_options.py +++ b/python/databricks/bundles/pipelines/_models/smartsheet_options.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SmartsheetOptions: """ Smartsheet specific options for ingestion """ - enforce_schema: VariableOrOptional[bool] = None """ [Public Preview] (Optional) When true, maps each column to its Smartsheet-declared type (Text/Number/Date/ @@ -27,11 +152,12 @@ class SmartsheetOptions: """ @classmethod - def from_dict(cls, value: "SmartsheetOptionsDict") -> "Self": + def from_dict(cls, value: 'SmartsheetOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SmartsheetOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SmartsheetOptionsDict': + return _transform_to_json_value(self) # type:ignore + class SmartsheetOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/source_catalog_config.py b/python/databricks/bundles/pipelines/_models/source_catalog_config.py index 096c7e971d7..dd90b20e46f 100644 --- a/python/databricks/bundles/pipelines/_models/source_catalog_config.py +++ b/python/databricks/bundles/pipelines/_models/source_catalog_config.py @@ -1,26 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.postgres_catalog_config import ( - PostgresCatalogConfig, - PostgresCatalogConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SourceCatalogConfig: """ SourceCatalogConfig contains catalog-level custom configuration parameters for each source """ - postgres: VariableOrOptional[PostgresCatalogConfig] = None """ [Public Preview] Postgres-specific catalog-level configuration parameters @@ -32,11 +153,12 @@ class SourceCatalogConfig: """ @classmethod - def from_dict(cls, value: "SourceCatalogConfigDict") -> "Self": + def from_dict(cls, value: 'SourceCatalogConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SourceCatalogConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SourceCatalogConfigDict': + return _transform_to_json_value(self) # type:ignore + class SourceCatalogConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/source_config.py b/python/databricks/bundles/pipelines/_models/source_config.py index 4695b153b40..33351d05d0b 100644 --- a/python/databricks/bundles/pipelines/_models/source_config.py +++ b/python/databricks/bundles/pipelines/_models/source_config.py @@ -1,28 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.api_source_connector_config import ( - ApiSourceConnectorConfig, - ApiSourceConnectorConfigParam, -) -from databricks.bundles.pipelines._models.google_ads_config import ( - GoogleAdsConfig, - GoogleAdsConfigParam, -) -from databricks.bundles.pipelines._models.source_catalog_config import ( - SourceCatalogConfig, - SourceCatalogConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SourceConfig: """""" @@ -47,11 +161,12 @@ class SourceConfig: """ @classmethod - def from_dict(cls, value: "SourceConfigDict") -> "Self": + def from_dict(cls, value: 'SourceConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SourceConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SourceConfigDict': + return _transform_to_json_value(self) # type:ignore + class SourceConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/table_spec.py b/python/databricks/bundles/pipelines/_models/table_spec.py index dc75d872c71..86c3c8445d1 100644 --- a/python/databricks/bundles/pipelines/_models/table_spec.py +++ b/python/databricks/bundles/pipelines/_models/table_spec.py @@ -1,24 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.pipelines._models.connector_options import ( - ConnectorOptions, - ConnectorOptionsParam, -) -from databricks.bundles.pipelines._models.table_specific_config import ( - TableSpecificConfig, - TableSpecificConfigParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TableSpec: """""" @@ -66,11 +184,12 @@ class TableSpec: """ @classmethod - def from_dict(cls, value: "TableSpecDict") -> "Self": + def from_dict(cls, value: 'TableSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TableSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TableSpecDict': + return _transform_to_json_value(self) # type:ignore + class TableSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/table_specific_config.py b/python/databricks/bundles/pipelines/_models/table_specific_config.py index 42773ae759e..fb29c052283 100644 --- a/python/databricks/bundles/pipelines/_models/table_specific_config.py +++ b/python/databricks/bundles/pipelines/_models/table_specific_config.py @@ -1,36 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.pipelines._models.auto_full_refresh_policy import ( - AutoFullRefreshPolicy, - AutoFullRefreshPolicyParam, -) -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import ( - IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, - IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam, -) -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import ( - IngestionPipelineDefinitionWorkdayReportParameters, - IngestionPipelineDefinitionWorkdayReportParametersParam, -) -from databricks.bundles.pipelines._models.table_specific_config_scd_type import ( - TableSpecificConfigScdType, - TableSpecificConfigScdTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TableSpecificConfig: """""" @@ -96,9 +202,7 @@ class TableSpecificConfig: [Public Preview] The primary key of the table used to apply changes. """ - query_based_connector_config: VariableOrOptional[ - IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig - ] = None + query_based_connector_config: VariableOrOptional[IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig] = None """ [Public Preview] Configurations that are only applicable for query-based ingestion connectors. """ @@ -145,9 +249,7 @@ class TableSpecificConfig: Note: table_properties in table specific configuration will override the table_properties of the pipeline definition. """ - workday_report_parameters: VariableOrOptional[ - IngestionPipelineDefinitionWorkdayReportParameters - ] = None + workday_report_parameters: VariableOrOptional[IngestionPipelineDefinitionWorkdayReportParameters] = None """ :meta private: [EXPERIMENTAL] @@ -155,11 +257,12 @@ class TableSpecificConfig: """ @classmethod - def from_dict(cls, value: "TableSpecificConfigDict") -> "Self": + def from_dict(cls, value: 'TableSpecificConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TableSpecificConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TableSpecificConfigDict': + return _transform_to_json_value(self) # type:ignore + class TableSpecificConfigDict(TypedDict, total=False): @@ -226,9 +329,7 @@ class TableSpecificConfigDict(TypedDict, total=False): [Public Preview] The primary key of the table used to apply changes. """ - query_based_connector_config: VariableOrOptional[ - IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam - ] + query_based_connector_config: VariableOrOptional[IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam] """ [Public Preview] Configurations that are only applicable for query-based ingestion connectors. """ @@ -275,9 +376,7 @@ class TableSpecificConfigDict(TypedDict, total=False): Note: table_properties in table specific configuration will override the table_properties of the pipeline definition. """ - workday_report_parameters: VariableOrOptional[ - IngestionPipelineDefinitionWorkdayReportParametersParam - ] + workday_report_parameters: VariableOrOptional[IngestionPipelineDefinitionWorkdayReportParametersParam] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/table_specific_config_scd_type.py b/python/databricks/bundles/pipelines/_models/table_specific_config_scd_type.py index 2d650e17111..fe3e10d7cbc 100644 --- a/python/databricks/bundles/pipelines/_models/table_specific_config_scd_type.py +++ b/python/databricks/bundles/pipelines/_models/table_specific_config_scd_type.py @@ -1,19 +1,144 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class TableSpecificConfigScdType(Enum): """ The SCD type to use to ingest the table. """ - SCD_TYPE_1 = "SCD_TYPE_1" SCD_TYPE_2 = "SCD_TYPE_2" APPEND_ONLY = "APPEND_ONLY" - -TableSpecificConfigScdTypeParam = ( - Literal["SCD_TYPE_1", "SCD_TYPE_2", "APPEND_ONLY"] | TableSpecificConfigScdType -) +TableSpecificConfigScdTypeParam = Literal["SCD_TYPE_1", "SCD_TYPE_2", "APPEND_ONLY"] | TableSpecificConfigScdType diff --git a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options.py b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options.py index 531f06cd434..1a5330b3e0c 100644 --- a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options.py +++ b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options.py @@ -1,37 +1,148 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import ( - TikTokAdsOptionsTikTokAdsCustomReportOptions, - TikTokAdsOptionsTikTokAdsCustomReportOptionsParam, -) -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import ( - TikTokAdsOptionsTikTokDataLevel, - TikTokAdsOptionsTikTokDataLevelParam, -) -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import ( - TikTokAdsOptionsTikTokReportType, - TikTokAdsOptionsTikTokReportTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TikTokAdsOptions: """ TikTok Ads specific options for ingestion """ - - custom_report_options: VariableOrOptional[ - TikTokAdsOptionsTikTokAdsCustomReportOptions - ] = None + custom_report_options: VariableOrOptional[TikTokAdsOptionsTikTokAdsCustomReportOptions] = None """ :meta private: [EXPERIMENTAL] @@ -94,19 +205,18 @@ class TikTokAdsOptions: """ @classmethod - def from_dict(cls, value: "TikTokAdsOptionsDict") -> "Self": + def from_dict(cls, value: 'TikTokAdsOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TikTokAdsOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TikTokAdsOptionsDict': + return _transform_to_json_value(self) # type:ignore + class TikTokAdsOptionsDict(TypedDict, total=False): """""" - custom_report_options: VariableOrOptional[ - TikTokAdsOptionsTikTokAdsCustomReportOptionsParam - ] + custom_report_options: VariableOrOptional[TikTokAdsOptionsTikTokAdsCustomReportOptionsParam] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_ads_custom_report_options.py b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_ads_custom_report_options.py index 6c2820a3bc2..4586a8359bd 100644 --- a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_ads_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_ads_custom_report_options.py @@ -1,34 +1,151 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import ( - TikTokAdsOptionsTikTokDataLevel, - TikTokAdsOptionsTikTokDataLevelParam, -) -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import ( - TikTokAdsOptionsTikTokReportType, - TikTokAdsOptionsTikTokReportTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class TikTokAdsOptionsTikTokAdsCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + User-defined custom report for the TikTok Ads connector. Groups the dimensions + metrics + report type + data level that define a TikTok Ads custom report request. """ - data_level: VariableOrOptional[TikTokAdsOptionsTikTokDataLevel] = None """ :meta private: [EXPERIMENTAL] @@ -71,13 +188,12 @@ class TikTokAdsOptionsTikTokAdsCustomReportOptions: """ @classmethod - def from_dict( - cls, value: "TikTokAdsOptionsTikTokAdsCustomReportOptionsDict" - ) -> "Self": + def from_dict(cls, value: 'TikTokAdsOptionsTikTokAdsCustomReportOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TikTokAdsOptionsTikTokAdsCustomReportOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TikTokAdsOptionsTikTokAdsCustomReportOptionsDict': + return _transform_to_json_value(self) # type:ignore + class TikTokAdsOptionsTikTokAdsCustomReportOptionsDict(TypedDict, total=False): @@ -125,7 +241,4 @@ class TikTokAdsOptionsTikTokAdsCustomReportOptionsDict(TypedDict, total=False): """ -TikTokAdsOptionsTikTokAdsCustomReportOptionsParam = ( - TikTokAdsOptionsTikTokAdsCustomReportOptionsDict - | TikTokAdsOptionsTikTokAdsCustomReportOptions -) +TikTokAdsOptionsTikTokAdsCustomReportOptionsParam = TikTokAdsOptionsTikTokAdsCustomReportOptionsDict | TikTokAdsOptionsTikTokAdsCustomReportOptions diff --git a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_data_level.py b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_data_level.py index 7795cd821b6..ddbe90f1ab9 100644 --- a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_data_level.py +++ b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_data_level.py @@ -1,21 +1,145 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class TikTokAdsOptionsTikTokDataLevel(Enum): """ Data level for TikTok Ads report aggregation. """ - AUCTION_ADVERTISER = "AUCTION_ADVERTISER" AUCTION_CAMPAIGN = "AUCTION_CAMPAIGN" AUCTION_ADGROUP = "AUCTION_ADGROUP" AUCTION_AD = "AUCTION_AD" - -TikTokAdsOptionsTikTokDataLevelParam = ( - Literal["AUCTION_ADVERTISER", "AUCTION_CAMPAIGN", "AUCTION_ADGROUP", "AUCTION_AD"] - | TikTokAdsOptionsTikTokDataLevel -) +TikTokAdsOptionsTikTokDataLevelParam = Literal["AUCTION_ADVERTISER", "AUCTION_CAMPAIGN", "AUCTION_ADGROUP", "AUCTION_AD"] | TikTokAdsOptionsTikTokDataLevel diff --git a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_report_type.py b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_report_type.py index c97e2ee93a5..1bc844137e9 100644 --- a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_report_type.py +++ b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_report_type.py @@ -1,14 +1,142 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam class TikTokAdsOptionsTikTokReportType(Enum): """ Report type for TikTok Ads API. """ - BASIC = "BASIC" AUDIENCE = "AUDIENCE" PLAYABLE_AD = "PLAYABLE_AD" @@ -16,8 +144,4 @@ class TikTokAdsOptionsTikTokReportType(Enum): BUSINESS_CENTER = "BUSINESS_CENTER" GMV_MAX = "GMV_MAX" - -TikTokAdsOptionsTikTokReportTypeParam = ( - Literal["BASIC", "AUDIENCE", "PLAYABLE_AD", "DSA", "BUSINESS_CENTER", "GMV_MAX"] - | TikTokAdsOptionsTikTokReportType -) +TikTokAdsOptionsTikTokReportTypeParam = Literal["BASIC", "AUDIENCE", "PLAYABLE_AD", "DSA", "BUSINESS_CENTER", "GMV_MAX"] | TikTokAdsOptionsTikTokReportType diff --git a/python/databricks/bundles/pipelines/_models/transformer.py b/python/databricks/bundles/pipelines/_models/transformer.py index b42fcef66f9..3c26cdc5ee9 100644 --- a/python/databricks/bundles/pipelines/_models/transformer.py +++ b/python/databricks/bundles/pipelines/_models/transformer.py @@ -1,38 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.pipelines._models.avro_transformer_options import ( - AvroTransformerOptions, - AvroTransformerOptionsParam, -) -from databricks.bundles.pipelines._models.json_transformer_options import ( - JsonTransformerOptions, - JsonTransformerOptionsParam, -) -from databricks.bundles.pipelines._models.protobuf_transformer_options import ( - ProtobufTransformerOptions, - ProtobufTransformerOptionsParam, -) -from databricks.bundles.pipelines._models.transformer_format import ( - TransformerFormat, - TransformerFormatParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Transformer: """ Specifies how to transform binary data into structured data. """ - avro_options: VariableOrOptional[AvroTransformerOptions] = None """ :meta private: [EXPERIMENTAL] @@ -78,11 +187,12 @@ class Transformer: """ @classmethod - def from_dict(cls, value: "TransformerDict") -> "Self": + def from_dict(cls, value: 'TransformerDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "TransformerDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'TransformerDict': + return _transform_to_json_value(self) # type:ignore + class TransformerDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/transformer_format.py b/python/databricks/bundles/pipelines/_models/transformer_format.py index 006855705ca..0b49977dc3f 100644 --- a/python/databricks/bundles/pipelines/_models/transformer_format.py +++ b/python/databricks/bundles/pipelines/_models/transformer_format.py @@ -1,7 +1,136 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam class TransformerFormat(Enum): @@ -10,7 +139,4 @@ class TransformerFormat(Enum): AVRO = "AVRO" PROTOBUF = "PROTOBUF" - -TransformerFormatParam = ( - Literal["STRING", "JSON", "AVRO", "PROTOBUF"] | TransformerFormat -) +TransformerFormatParam = Literal["STRING", "JSON", "AVRO", "PROTOBUF"] | TransformerFormat diff --git a/python/databricks/bundles/pipelines/_models/volumes_storage_info.py b/python/databricks/bundles/pipelines/_models/volumes_storage_info.py index 1eb4fff7bc2..1e0f05dbcf2 100644 --- a/python/databricks/bundles/pipelines/_models/volumes_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/volumes_storage_info.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class VolumesStorageInfo: """ A storage location back by UC Volumes. """ - destination: VariableOr[str] """ UC Volumes destination, e.g. `/Volumes/catalog/schema/vol1/init-scripts/setup-datadog.sh` @@ -24,11 +149,12 @@ class VolumesStorageInfo: """ @classmethod - def from_dict(cls, value: "VolumesStorageInfoDict") -> "Self": + def from_dict(cls, value: 'VolumesStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "VolumesStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'VolumesStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class VolumesStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/workspace_storage_info.py b/python/databricks/bundles/pipelines/_models/workspace_storage_info.py index 49aef9256c8..f222fd1f7ea 100644 --- a/python/databricks/bundles/pipelines/_models/workspace_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/workspace_storage_info.py @@ -1,33 +1,159 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class WorkspaceStorageInfo: """ A storage location in Workspace Filesystem (WSFS) """ - destination: VariableOr[str] """ wsfs destination, e.g. `workspace:/cluster-init-scripts/setup-datadog.sh` """ @classmethod - def from_dict(cls, value: "WorkspaceStorageInfoDict") -> "Self": + def from_dict(cls, value: 'WorkspaceStorageInfoDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "WorkspaceStorageInfoDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'WorkspaceStorageInfoDict': + return _transform_to_json_value(self) # type:ignore + class WorkspaceStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/zendesk_support_options.py b/python/databricks/bundles/pipelines/_models/zendesk_support_options.py index 23a03e8eab8..38b25e3c76f 100644 --- a/python/databricks/bundles/pipelines/_models/zendesk_support_options.py +++ b/python/databricks/bundles/pipelines/_models/zendesk_support_options.py @@ -1,22 +1,147 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam +from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam +from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam +from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam +from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam +from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam +from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam +from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam +from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam +from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam +from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam +from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam +from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam +from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam +from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam +from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam +from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam +from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam +from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam +from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam +from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam +from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam +from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam +from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam +from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam +from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam +from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam +from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam +from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam +from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam +from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam +from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam +from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam +from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam +from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam +from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam +from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam +from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam +from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam +from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam +from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam +from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam +from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam +from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam +from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam +from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam +from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam +from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam +from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam +from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam +from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam +from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam +from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam +from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam + +from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam + +from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam + +from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam + +from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam + +from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam + +from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam + +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam + +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam + +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + +from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam + +from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam + +from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam + +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam + +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam + +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam + +from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam + +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam + +from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class ZendeskSupportOptions: """ Zendesk Support specific options for ingestion """ - start_date: VariableOrOptional[str] = None """ [Public Preview] (Optional) Start date in YYYY-MM-DD format for the initial sync. @@ -24,11 +149,12 @@ class ZendeskSupportOptions: """ @classmethod - def from_dict(cls, value: "ZendeskSupportOptionsDict") -> "Self": + def from_dict(cls, value: 'ZendeskSupportOptionsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ZendeskSupportOptionsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ZendeskSupportOptionsDict': + return _transform_to_json_value(self) # type:ignore + class ZendeskSupportOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/__init__.py b/python/databricks/bundles/quality_monitors/__init__.py index ad0902bfcf6..5c4e3fed940 100644 --- a/python/databricks/bundles/quality_monitors/__init__.py +++ b/python/databricks/bundles/quality_monitors/__init__.py @@ -1,104 +1,58 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "MonitorCronSchedule", - "MonitorCronScheduleDict", - "MonitorCronScheduleParam", - "MonitorCronSchedulePauseStatus", - "MonitorCronSchedulePauseStatusParam", - "MonitorDataClassificationConfig", - "MonitorDataClassificationConfigDict", - "MonitorDataClassificationConfigParam", - "MonitorDestination", - "MonitorDestinationDict", - "MonitorDestinationParam", - "MonitorInferenceLog", - "MonitorInferenceLogDict", - "MonitorInferenceLogParam", - "MonitorInferenceLogProblemType", - "MonitorInferenceLogProblemTypeParam", - "MonitorMetric", - "MonitorMetricDict", - "MonitorMetricParam", - "MonitorMetricType", - "MonitorMetricTypeParam", - "MonitorNotifications", - "MonitorNotificationsDict", - "MonitorNotificationsParam", - "MonitorSnapshot", - "MonitorSnapshotDict", - "MonitorSnapshotParam", - "MonitorTimeSeries", - "MonitorTimeSeriesDict", - "MonitorTimeSeriesParam", - "QualityMonitor", - "QualityMonitorDict", - "QualityMonitorParam", + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'MonitorCronSchedule', + 'MonitorCronScheduleDict', + 'MonitorCronScheduleParam', + 'MonitorCronSchedulePauseStatus', + 'MonitorCronSchedulePauseStatusParam', + 'MonitorDataClassificationConfig', + 'MonitorDataClassificationConfigDict', + 'MonitorDataClassificationConfigParam', + 'MonitorDestination', + 'MonitorDestinationDict', + 'MonitorDestinationParam', + 'MonitorInferenceLog', + 'MonitorInferenceLogDict', + 'MonitorInferenceLogParam', + 'MonitorInferenceLogProblemType', + 'MonitorInferenceLogProblemTypeParam', + 'MonitorMetric', + 'MonitorMetricDict', + 'MonitorMetricParam', + 'MonitorMetricType', + 'MonitorMetricTypeParam', + 'MonitorNotifications', + 'MonitorNotificationsDict', + 'MonitorNotificationsParam', + 'MonitorSnapshot', + 'MonitorSnapshotDict', + 'MonitorSnapshotParam', + 'MonitorTimeSeries', + 'MonitorTimeSeriesDict', + 'MonitorTimeSeriesParam', + 'QualityMonitor', + 'QualityMonitorDict', + 'QualityMonitorParam', ] -from databricks.bundles.quality_monitors._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import ( - MonitorCronSchedule, - MonitorCronScheduleDict, - MonitorCronScheduleParam, -) -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import ( - MonitorCronSchedulePauseStatus, - MonitorCronSchedulePauseStatusParam, -) -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import ( - MonitorDataClassificationConfig, - MonitorDataClassificationConfigDict, - MonitorDataClassificationConfigParam, -) -from databricks.bundles.quality_monitors._models.monitor_destination import ( - MonitorDestination, - MonitorDestinationDict, - MonitorDestinationParam, -) -from databricks.bundles.quality_monitors._models.monitor_inference_log import ( - MonitorInferenceLog, - MonitorInferenceLogDict, - MonitorInferenceLogParam, -) -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import ( - MonitorInferenceLogProblemType, - MonitorInferenceLogProblemTypeParam, -) -from databricks.bundles.quality_monitors._models.monitor_metric import ( - MonitorMetric, - MonitorMetricDict, - MonitorMetricParam, -) -from databricks.bundles.quality_monitors._models.monitor_metric_type import ( - MonitorMetricType, - MonitorMetricTypeParam, -) -from databricks.bundles.quality_monitors._models.monitor_notifications import ( - MonitorNotifications, - MonitorNotificationsDict, - MonitorNotificationsParam, -) -from databricks.bundles.quality_monitors._models.monitor_snapshot import ( - MonitorSnapshot, - MonitorSnapshotDict, - MonitorSnapshotParam, -) -from databricks.bundles.quality_monitors._models.monitor_time_series import ( - MonitorTimeSeries, - MonitorTimeSeriesDict, - MonitorTimeSeriesParam, -) -from databricks.bundles.quality_monitors._models.quality_monitor import ( - QualityMonitor, - QualityMonitorDict, - QualityMonitorParam, -) +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + diff --git a/python/databricks/bundles/quality_monitors/_models/lifecycle.py b/python/databricks/bundles/quality_monitors/_models/lifecycle.py index 697776a198e..0de2fca97eb 100644 --- a/python/databricks/bundles/quality_monitors/_models/lifecycle.py +++ b/python/databricks/bundles/quality_monitors/_models/lifecycle.py @@ -1,16 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +38,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule.py b/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule.py index 02a8de80077..ff2c7d94a65 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule.py @@ -1,20 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import ( - MonitorCronSchedulePauseStatus, - MonitorCronSchedulePauseStatusParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MonitorCronSchedule: """""" @@ -35,11 +48,12 @@ class MonitorCronSchedule: """ @classmethod - def from_dict(cls, value: "MonitorCronScheduleDict") -> "Self": + def from_dict(cls, value: 'MonitorCronScheduleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MonitorCronScheduleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MonitorCronScheduleDict': + return _transform_to_json_value(self) # type:ignore + class MonitorCronScheduleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule_pause_status.py b/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule_pause_status.py index 69b5dd3666b..6994b0fca8d 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule_pause_status.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule_pause_status.py @@ -1,7 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam class MonitorCronSchedulePauseStatus(Enum): @@ -9,12 +29,8 @@ class MonitorCronSchedulePauseStatus(Enum): Source link: https://src.dev.databricks.com/databricks/universe/-/blob/elastic-spark-common/api/messages/schedule.proto Monitoring workflow schedule pause status. """ - UNSPECIFIED = "UNSPECIFIED" UNPAUSED = "UNPAUSED" PAUSED = "PAUSED" - -MonitorCronSchedulePauseStatusParam = ( - Literal["UNSPECIFIED", "UNPAUSED", "PAUSED"] | MonitorCronSchedulePauseStatus -) +MonitorCronSchedulePauseStatusParam = Literal["UNSPECIFIED", "UNPAUSED", "PAUSED"] | MonitorCronSchedulePauseStatus diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_data_classification_config.py b/python/databricks/bundles/quality_monitors/_models/monitor_data_classification_config.py index cda4e643683..be3a2f2d29a 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_data_classification_config.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_data_classification_config.py @@ -1,24 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MonitorDataClassificationConfig: """ :meta private: [EXPERIMENTAL] - + Data classification related configuration. """ - enabled: VariableOrOptional[bool] = None """ :meta private: [EXPERIMENTAL] @@ -27,11 +43,12 @@ class MonitorDataClassificationConfig: """ @classmethod - def from_dict(cls, value: "MonitorDataClassificationConfigDict") -> "Self": + def from_dict(cls, value: 'MonitorDataClassificationConfigDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MonitorDataClassificationConfigDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MonitorDataClassificationConfigDict': + return _transform_to_json_value(self) # type:ignore + class MonitorDataClassificationConfigDict(TypedDict, total=False): @@ -45,6 +62,4 @@ class MonitorDataClassificationConfigDict(TypedDict, total=False): """ -MonitorDataClassificationConfigParam = ( - MonitorDataClassificationConfigDict | MonitorDataClassificationConfig -) +MonitorDataClassificationConfigParam = MonitorDataClassificationConfigDict | MonitorDataClassificationConfig diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_destination.py b/python/databricks/bundles/quality_monitors/_models/monitor_destination.py index 2f4f5b1d37f..ca6decc05f7 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_destination.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_destination.py @@ -1,16 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MonitorDestination: """""" @@ -21,11 +38,12 @@ class MonitorDestination: """ @classmethod - def from_dict(cls, value: "MonitorDestinationDict") -> "Self": + def from_dict(cls, value: 'MonitorDestinationDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MonitorDestinationDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MonitorDestinationDict': + return _transform_to_json_value(self) # type:ignore + class MonitorDestinationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_inference_log.py b/python/databricks/bundles/quality_monitors/_models/monitor_inference_log.py index 24b6ea70e87..4804b87def3 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_inference_log.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_inference_log.py @@ -1,24 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import ( - MonitorInferenceLogProblemType, - MonitorInferenceLogProblemTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MonitorInferenceLog: """""" @@ -59,11 +68,12 @@ class MonitorInferenceLog: """ @classmethod - def from_dict(cls, value: "MonitorInferenceLogDict") -> "Self": + def from_dict(cls, value: 'MonitorInferenceLogDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MonitorInferenceLogDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MonitorInferenceLogDict': + return _transform_to_json_value(self) # type:ignore + class MonitorInferenceLogDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_inference_log_problem_type.py b/python/databricks/bundles/quality_monitors/_models/monitor_inference_log_problem_type.py index 42d8569619b..de722188952 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_inference_log_problem_type.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_inference_log_problem_type.py @@ -1,15 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam class MonitorInferenceLogProblemType(Enum): PROBLEM_TYPE_CLASSIFICATION = "PROBLEM_TYPE_CLASSIFICATION" PROBLEM_TYPE_REGRESSION = "PROBLEM_TYPE_REGRESSION" - -MonitorInferenceLogProblemTypeParam = ( - Literal["PROBLEM_TYPE_CLASSIFICATION", "PROBLEM_TYPE_REGRESSION"] - | MonitorInferenceLogProblemType -) +MonitorInferenceLogProblemTypeParam = Literal["PROBLEM_TYPE_CLASSIFICATION", "PROBLEM_TYPE_REGRESSION"] | MonitorInferenceLogProblemType diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_metric.py b/python/databricks/bundles/quality_monitors/_models/monitor_metric.py index cb7f876dae8..a22566cc818 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_metric.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_metric.py @@ -1,26 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrList -from databricks.bundles.quality_monitors._models.monitor_metric_type import ( - MonitorMetricType, - MonitorMetricTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MonitorMetric: """ Custom metric definition. """ - definition: VariableOr[str] """ Jinja template for a SQL expression that specifies how to compute the metric. See [create metric definition](https://docs.databricks.com/en/lakehouse-monitoring/custom-metrics.html#create-definition). @@ -54,11 +66,12 @@ class MonitorMetric: """ @classmethod - def from_dict(cls, value: "MonitorMetricDict") -> "Self": + def from_dict(cls, value: 'MonitorMetricDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MonitorMetricDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MonitorMetricDict': + return _transform_to_json_value(self) # type:ignore + class MonitorMetricDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_metric_type.py b/python/databricks/bundles/quality_monitors/_models/monitor_metric_type.py index 4c70c089a0f..abe31acfa9a 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_metric_type.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_metric_type.py @@ -1,7 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam class MonitorMetricType(Enum): @@ -14,17 +34,8 @@ class MonitorMetricType(Enum): - CUSTOM_METRIC_TYPE_DERIVED: depend on previously computed aggregate metrics - CUSTOM_METRIC_TYPE_DRIFT: depend on previously computed aggregate or derived metrics """ - CUSTOM_METRIC_TYPE_AGGREGATE = "CUSTOM_METRIC_TYPE_AGGREGATE" CUSTOM_METRIC_TYPE_DERIVED = "CUSTOM_METRIC_TYPE_DERIVED" CUSTOM_METRIC_TYPE_DRIFT = "CUSTOM_METRIC_TYPE_DRIFT" - -MonitorMetricTypeParam = ( - Literal[ - "CUSTOM_METRIC_TYPE_AGGREGATE", - "CUSTOM_METRIC_TYPE_DERIVED", - "CUSTOM_METRIC_TYPE_DRIFT", - ] - | MonitorMetricType -) +MonitorMetricTypeParam = Literal["CUSTOM_METRIC_TYPE_AGGREGATE", "CUSTOM_METRIC_TYPE_DERIVED", "CUSTOM_METRIC_TYPE_DRIFT"] | MonitorMetricType diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_notifications.py b/python/databricks/bundles/quality_monitors/_models/monitor_notifications.py index 007b3ae64e4..3801a91aafb 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_notifications.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_notifications.py @@ -1,20 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.quality_monitors._models.monitor_destination import ( - MonitorDestination, - MonitorDestinationParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MonitorNotifications: """""" @@ -32,11 +45,12 @@ class MonitorNotifications: """ @classmethod - def from_dict(cls, value: "MonitorNotificationsDict") -> "Self": + def from_dict(cls, value: 'MonitorNotificationsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MonitorNotificationsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MonitorNotificationsDict': + return _transform_to_json_value(self) # type:ignore + class MonitorNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_snapshot.py b/python/databricks/bundles/quality_monitors/_models/monitor_snapshot.py index 4cc39dcb9d0..e07bf168f4c 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_snapshot.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_snapshot.py @@ -1,27 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MonitorSnapshot: """ Snapshot analysis configuration """ - @classmethod - def from_dict(cls, value: "MonitorSnapshotDict") -> "Self": + def from_dict(cls, value: 'MonitorSnapshotDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MonitorSnapshotDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MonitorSnapshotDict': + return _transform_to_json_value(self) # type:ignore + class MonitorSnapshotDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_time_series.py b/python/databricks/bundles/quality_monitors/_models/monitor_time_series.py index 29f3b950a0e..f5bc7302386 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_time_series.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_time_series.py @@ -1,22 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrList +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class MonitorTimeSeries: """ Time series analysis configuration. """ - timestamp_col: VariableOr[str] """ Column for the timestamp. @@ -28,11 +44,12 @@ class MonitorTimeSeries: """ @classmethod - def from_dict(cls, value: "MonitorTimeSeriesDict") -> "Self": + def from_dict(cls, value: 'MonitorTimeSeriesDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "MonitorTimeSeriesDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'MonitorTimeSeriesDict': + return _transform_to_json_value(self) # type:ignore + class MonitorTimeSeriesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/quality_monitor.py b/python/databricks/bundles/quality_monitors/_models/quality_monitor.py index d2c22518503..7bd52cea946 100644 --- a/python/databricks/bundles/quality_monitors/_models/quality_monitor.py +++ b/python/databricks/bundles/quality_monitors/_models/quality_monitor.py @@ -1,53 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.quality_monitors._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import ( - MonitorCronSchedule, - MonitorCronScheduleParam, -) -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import ( - MonitorDataClassificationConfig, - MonitorDataClassificationConfigParam, -) -from databricks.bundles.quality_monitors._models.monitor_inference_log import ( - MonitorInferenceLog, - MonitorInferenceLogParam, -) -from databricks.bundles.quality_monitors._models.monitor_metric import ( - MonitorMetric, - MonitorMetricParam, -) -from databricks.bundles.quality_monitors._models.monitor_notifications import ( - MonitorNotifications, - MonitorNotificationsParam, -) -from databricks.bundles.quality_monitors._models.monitor_snapshot import ( - MonitorSnapshot, - MonitorSnapshotParam, -) -from databricks.bundles.quality_monitors._models.monitor_time_series import ( - MonitorTimeSeries, - MonitorTimeSeriesParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam +from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam +from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam +from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam +from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam + +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam + +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class QualityMonitor(Resource): """""" @@ -77,9 +57,7 @@ class QualityMonitor(Resource): [Create:OPT Update:OPT] Custom metrics. """ - data_classification_config: VariableOrOptional[MonitorDataClassificationConfig] = ( - None - ) + data_classification_config: VariableOrOptional[MonitorDataClassificationConfig] = None """ :meta private: [EXPERIMENTAL] @@ -140,11 +118,12 @@ class QualityMonitor(Resource): """ @classmethod - def from_dict(cls, value: "QualityMonitorDict") -> "Self": + def from_dict(cls, value: 'QualityMonitorDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "QualityMonitorDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'QualityMonitorDict': + return _transform_to_json_value(self) # type:ignore + class QualityMonitorDict(TypedDict, total=False): diff --git a/python/databricks/bundles/registered_models/__init__.py b/python/databricks/bundles/registered_models/__init__.py index a7ae3ba9ef3..cd283ff2cfa 100644 --- a/python/databricks/bundles/registered_models/__init__.py +++ b/python/databricks/bundles/registered_models/__init__.py @@ -1,44 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", - "RegisteredModel", - "RegisteredModelAlias", - "RegisteredModelAliasDict", - "RegisteredModelAliasParam", - "RegisteredModelDict", - "RegisteredModelParam", + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'RegisteredModel', + 'RegisteredModelAlias', + 'RegisteredModelAliasDict', + 'RegisteredModelAliasParam', + 'RegisteredModelDict', + 'RegisteredModelParam', ] -from databricks.bundles.registered_models._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.registered_models._models.privilege import ( - Privilege, - PrivilegeParam, -) -from databricks.bundles.registered_models._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) -from databricks.bundles.registered_models._models.registered_model import ( - RegisteredModel, - RegisteredModelDict, - RegisteredModelParam, -) -from databricks.bundles.registered_models._models.registered_model_alias import ( - RegisteredModelAlias, - RegisteredModelAliasDict, - RegisteredModelAliasParam, -) +from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam +from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam +from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam + diff --git a/python/databricks/bundles/registered_models/_models/lifecycle.py b/python/databricks/bundles/registered_models/_models/lifecycle.py index 697776a198e..c360ba26bfc 100644 --- a/python/databricks/bundles/registered_models/_models/lifecycle.py +++ b/python/databricks/bundles/registered_models/_models/lifecycle.py @@ -1,16 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam +from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam +from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +28,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/registered_models/_models/privilege.py b/python/databricks/bundles/registered_models/_models/privilege.py index df45211a76a..6b316c45dbc 100644 --- a/python/databricks/bundles/registered_models/_models/privilege.py +++ b/python/databricks/bundles/registered_models/_models/privilege.py @@ -1,8 +1,18 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam +from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam class Privilege(Enum): SELECT = "SELECT" @@ -58,61 +68,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/registered_models/_models/privilege_assignment.py b/python/databricks/bundles/registered_models/_models/privilege_assignment.py index 10ef2d094f8..a7b57d701c4 100644 --- a/python/databricks/bundles/registered_models/_models/privilege_assignment.py +++ b/python/databricks/bundles/registered_models/_models/privilege_assignment.py @@ -1,20 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.registered_models._models.privilege import ( - Privilege, - PrivilegeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam +from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam +from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -31,11 +34,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/registered_models/_models/registered_model.py b/python/databricks/bundles/registered_models/_models/registered_model.py index 08cf2ea45ae..e680d9f21de 100644 --- a/python/databricks/bundles/registered_models/_models/registered_model.py +++ b/python/databricks/bundles/registered_models/_models/registered_model.py @@ -1,29 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.registered_models._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.registered_models._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) -from databricks.bundles.registered_models._models.registered_model_alias import ( - RegisteredModelAlias, - RegisteredModelAliasParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam +from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RegisteredModel(Resource): """""" @@ -104,11 +98,12 @@ class RegisteredModel(Resource): """ @classmethod - def from_dict(cls, value: "RegisteredModelDict") -> "Self": + def from_dict(cls, value: 'RegisteredModelDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RegisteredModelDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RegisteredModelDict': + return _transform_to_json_value(self) # type:ignore + class RegisteredModelDict(TypedDict, total=False): diff --git a/python/databricks/bundles/registered_models/_models/registered_model_alias.py b/python/databricks/bundles/registered_models/_models/registered_model_alias.py index f4ea570a9a6..4842cf5d944 100644 --- a/python/databricks/bundles/registered_models/_models/registered_model_alias.py +++ b/python/databricks/bundles/registered_models/_models/registered_model_alias.py @@ -1,16 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam +from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class RegisteredModelAlias: """""" @@ -46,11 +53,12 @@ class RegisteredModelAlias: """ @classmethod - def from_dict(cls, value: "RegisteredModelAliasDict") -> "Self": + def from_dict(cls, value: 'RegisteredModelAliasDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "RegisteredModelAliasDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'RegisteredModelAliasDict': + return _transform_to_json_value(self) # type:ignore + class RegisteredModelAliasDict(TypedDict, total=False): diff --git a/python/databricks/bundles/schemas/__init__.py b/python/databricks/bundles/schemas/__init__.py index e891669012c..9f9373136dd 100644 --- a/python/databricks/bundles/schemas/__init__.py +++ b/python/databricks/bundles/schemas/__init__.py @@ -1,37 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", - "Schema", - "SchemaDict", - "SchemaGrant", - "SchemaGrantDict", - "SchemaGrantParam", - "SchemaGrantPrivilege", - "SchemaGrantPrivilegeParam", - "SchemaParam", + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'Schema', + 'SchemaDict', + 'SchemaParam', + 'SchemaGrant', + 'SchemaGrantDict', + 'SchemaGrantParam', + 'SchemaGrantPrivilege', + 'SchemaGrantPrivilegeParam', ] -from databricks.bundles.schemas._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.schemas._models.privilege import Privilege, PrivilegeParam -from databricks.bundles.schemas._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) +from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam +from databricks.bundles.schemas._models.privilege import Privilege, PrivilegeParam + SchemaGrant = PrivilegeAssignment SchemaGrantDict = PrivilegeAssignmentDict diff --git a/python/databricks/bundles/schemas/_models/lifecycle.py b/python/databricks/bundles/schemas/_models/lifecycle.py index 697776a198e..c5c99d0e4bf 100644 --- a/python/databricks/bundles/schemas/_models/lifecycle.py +++ b/python/databricks/bundles/schemas/_models/lifecycle.py @@ -1,16 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam +from databricks.bundles.schemas._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +27,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/schemas/_models/privilege.py b/python/databricks/bundles/schemas/_models/privilege.py index df45211a76a..cdd1f5348ac 100644 --- a/python/databricks/bundles/schemas/_models/privilege.py +++ b/python/databricks/bundles/schemas/_models/privilege.py @@ -1,8 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam class Privilege(Enum): SELECT = "SELECT" @@ -58,61 +67,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/schemas/_models/privilege_assignment.py b/python/databricks/bundles/schemas/_models/privilege_assignment.py index 012fe5b3b59..1155f70c633 100644 --- a/python/databricks/bundles/schemas/_models/privilege_assignment.py +++ b/python/databricks/bundles/schemas/_models/privilege_assignment.py @@ -1,17 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam from databricks.bundles.schemas._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -28,11 +33,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/schemas/_models/schema.py b/python/databricks/bundles/schemas/_models/schema.py index 32ff22c8227..fa5bd691fd8 100644 --- a/python/databricks/bundles/schemas/_models/schema.py +++ b/python/databricks/bundles/schemas/_models/schema.py @@ -1,27 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrDict, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.schemas._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.schemas._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Schema(Resource): """""" @@ -67,11 +62,12 @@ class Schema(Resource): """ @classmethod - def from_dict(cls, value: "SchemaDict") -> "Self": + def from_dict(cls, value: 'SchemaDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SchemaDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SchemaDict': + return _transform_to_json_value(self) # type:ignore + class SchemaDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secret_scopes/__init__.py b/python/databricks/bundles/secret_scopes/__init__.py index 1f3d9c9555d..c5882bdb65f 100644 --- a/python/databricks/bundles/secret_scopes/__init__.py +++ b/python/databricks/bundles/secret_scopes/__init__.py @@ -1,50 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "AzureKeyVaultSecretScopeMetadata", - "AzureKeyVaultSecretScopeMetadataDict", - "AzureKeyVaultSecretScopeMetadataParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "ScopeBackendType", - "ScopeBackendTypeParam", - "SecretScope", - "SecretScopeDict", - "SecretScopeParam", - "SecretScopePermission", - "SecretScopePermissionDict", - "SecretScopePermissionLevel", - "SecretScopePermissionLevelParam", - "SecretScopePermissionParam", + 'AzureKeyVaultSecretScopeMetadata', + 'AzureKeyVaultSecretScopeMetadataDict', + 'AzureKeyVaultSecretScopeMetadataParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'ScopeBackendType', + 'ScopeBackendTypeParam', + 'SecretScope', + 'SecretScopeDict', + 'SecretScopeParam', + 'SecretScopePermission', + 'SecretScopePermissionDict', + 'SecretScopePermissionLevel', + 'SecretScopePermissionLevelParam', + 'SecretScopePermissionParam', ] -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import ( - AzureKeyVaultSecretScopeMetadata, - AzureKeyVaultSecretScopeMetadataDict, - AzureKeyVaultSecretScopeMetadataParam, -) -from databricks.bundles.secret_scopes._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.secret_scopes._models.scope_backend_type import ( - ScopeBackendType, - ScopeBackendTypeParam, -) -from databricks.bundles.secret_scopes._models.secret_scope import ( - SecretScope, - SecretScopeDict, - SecretScopeParam, -) -from databricks.bundles.secret_scopes._models.secret_scope_permission import ( - SecretScopePermission, - SecretScopePermissionDict, - SecretScopePermissionParam, -) -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import ( - SecretScopePermissionLevel, - SecretScopePermissionLevelParam, -) +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam +from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam +from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam +from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam + +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam + diff --git a/python/databricks/bundles/secret_scopes/_models/azure_key_vault_secret_scope_metadata.py b/python/databricks/bundles/secret_scopes/_models/azure_key_vault_secret_scope_metadata.py index ceedb8b2ec4..8731cab043a 100644 --- a/python/databricks/bundles/secret_scopes/_models/azure_key_vault_secret_scope_metadata.py +++ b/python/databricks/bundles/secret_scopes/_models/azure_key_vault_secret_scope_metadata.py @@ -1,22 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam +from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam +from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam + +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class AzureKeyVaultSecretScopeMetadata: """ The metadata of the Azure KeyVault for a secret scope of type `AZURE_KEYVAULT` """ - dns_name: VariableOr[str] """ The DNS of the KeyVault @@ -28,11 +36,12 @@ class AzureKeyVaultSecretScopeMetadata: """ @classmethod - def from_dict(cls, value: "AzureKeyVaultSecretScopeMetadataDict") -> "Self": + def from_dict(cls, value: 'AzureKeyVaultSecretScopeMetadataDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "AzureKeyVaultSecretScopeMetadataDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'AzureKeyVaultSecretScopeMetadataDict': + return _transform_to_json_value(self) # type:ignore + class AzureKeyVaultSecretScopeMetadataDict(TypedDict, total=False): @@ -49,6 +58,4 @@ class AzureKeyVaultSecretScopeMetadataDict(TypedDict, total=False): """ -AzureKeyVaultSecretScopeMetadataParam = ( - AzureKeyVaultSecretScopeMetadataDict | AzureKeyVaultSecretScopeMetadata -) +AzureKeyVaultSecretScopeMetadataParam = AzureKeyVaultSecretScopeMetadataDict | AzureKeyVaultSecretScopeMetadata diff --git a/python/databricks/bundles/secret_scopes/_models/lifecycle.py b/python/databricks/bundles/secret_scopes/_models/lifecycle.py index 697776a198e..16e88694a25 100644 --- a/python/databricks/bundles/secret_scopes/_models/lifecycle.py +++ b/python/databricks/bundles/secret_scopes/_models/lifecycle.py @@ -1,16 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam +from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam +from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam +from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam + +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +30,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secret_scopes/_models/scope_backend_type.py b/python/databricks/bundles/secret_scopes/_models/scope_backend_type.py index d95f7022d33..9bc901f7b48 100644 --- a/python/databricks/bundles/secret_scopes/_models/scope_backend_type.py +++ b/python/databricks/bundles/secret_scopes/_models/scope_backend_type.py @@ -1,7 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam +from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam +from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam class ScopeBackendType(Enum): @@ -9,9 +21,7 @@ class ScopeBackendType(Enum): The types of secret scope backends in the Secret Manager. Azure KeyVault backed secret scopes will be supported in a later release. """ - DATABRICKS = "DATABRICKS" AZURE_KEYVAULT = "AZURE_KEYVAULT" - ScopeBackendTypeParam = Literal["DATABRICKS", "AZURE_KEYVAULT"] | ScopeBackendType diff --git a/python/databricks/bundles/secret_scopes/_models/secret_scope.py b/python/databricks/bundles/secret_scopes/_models/secret_scope.py index 5e436f1d247..fa36c384708 100644 --- a/python/databricks/bundles/secret_scopes/_models/secret_scope.py +++ b/python/databricks/bundles/secret_scopes/_models/secret_scope.py @@ -1,34 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import ( - AzureKeyVaultSecretScopeMetadata, - AzureKeyVaultSecretScopeMetadataParam, -) -from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.secret_scopes._models.scope_backend_type import ( - ScopeBackendType, - ScopeBackendTypeParam, -) -from databricks.bundles.secret_scopes._models.secret_scope_permission import ( - SecretScopePermission, - SecretScopePermissionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam +from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam +from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam + +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SecretScope(Resource): """""" @@ -59,11 +50,12 @@ class SecretScope(Resource): """ @classmethod - def from_dict(cls, value: "SecretScopeDict") -> "Self": + def from_dict(cls, value: 'SecretScopeDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SecretScopeDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SecretScopeDict': + return _transform_to_json_value(self) # type:ignore + class SecretScopeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secret_scopes/_models/secret_scope_permission.py b/python/databricks/bundles/secret_scopes/_models/secret_scope_permission.py index 3715039c411..47726469652 100644 --- a/python/databricks/bundles/secret_scopes/_models/secret_scope_permission.py +++ b/python/databricks/bundles/secret_scopes/_models/secret_scope_permission.py @@ -1,20 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import ( - SecretScopePermissionLevel, - SecretScopePermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam +from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam +from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam + +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SecretScopePermission: """""" @@ -40,11 +45,12 @@ class SecretScopePermission: """ @classmethod - def from_dict(cls, value: "SecretScopePermissionDict") -> "Self": + def from_dict(cls, value: 'SecretScopePermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SecretScopePermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SecretScopePermissionDict': + return _transform_to_json_value(self) # type:ignore + class SecretScopePermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secret_scopes/_models/secret_scope_permission_level.py b/python/databricks/bundles/secret_scopes/_models/secret_scope_permission_level.py index d7d84cd5854..373403cb375 100644 --- a/python/databricks/bundles/secret_scopes/_models/secret_scope_permission_level.py +++ b/python/databricks/bundles/secret_scopes/_models/secret_scope_permission_level.py @@ -1,7 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam +from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam +from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam +from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam class SecretScopePermissionLevel(Enum): @@ -9,7 +21,4 @@ class SecretScopePermissionLevel(Enum): WRITE = "WRITE" MANAGE = "MANAGE" - -SecretScopePermissionLevelParam = ( - Literal["READ", "WRITE", "MANAGE"] | SecretScopePermissionLevel -) +SecretScopePermissionLevelParam = Literal["READ", "WRITE", "MANAGE"] | SecretScopePermissionLevel diff --git a/python/databricks/bundles/secrets/__init__.py b/python/databricks/bundles/secrets/__init__.py index 6bc14c2a19d..feb447b97fc 100644 --- a/python/databricks/bundles/secrets/__init__.py +++ b/python/databricks/bundles/secrets/__init__.py @@ -1,29 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", - "Secret", - "SecretDict", - "SecretParam", + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'Secret', + 'SecretDict', + 'SecretParam', ] -from databricks.bundles.secrets._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.secrets._models.privilege import Privilege, PrivilegeParam -from databricks.bundles.secrets._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) +from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam from databricks.bundles.secrets._models.secret import Secret, SecretDict, SecretParam +from databricks.bundles.secrets._models.privilege import Privilege, PrivilegeParam + diff --git a/python/databricks/bundles/secrets/_models/lifecycle.py b/python/databricks/bundles/secrets/_models/lifecycle.py index 697776a198e..81ddf91d6b7 100644 --- a/python/databricks/bundles/secrets/_models/lifecycle.py +++ b/python/databricks/bundles/secrets/_models/lifecycle.py @@ -1,16 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.secrets._models.secret import Secret, SecretDict, SecretParam +from databricks.bundles.secrets._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +27,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secrets/_models/privilege.py b/python/databricks/bundles/secrets/_models/privilege.py index df45211a76a..543526c1089 100644 --- a/python/databricks/bundles/secrets/_models/privilege.py +++ b/python/databricks/bundles/secrets/_models/privilege.py @@ -1,8 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.secrets._models.secret import Secret, SecretDict, SecretParam class Privilege(Enum): SELECT = "SELECT" @@ -58,61 +67,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/secrets/_models/privilege_assignment.py b/python/databricks/bundles/secrets/_models/privilege_assignment.py index 443fa0120c7..4c2c7500501 100644 --- a/python/databricks/bundles/secrets/_models/privilege_assignment.py +++ b/python/databricks/bundles/secrets/_models/privilege_assignment.py @@ -1,17 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.secrets._models.secret import Secret, SecretDict, SecretParam from databricks.bundles.secrets._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -28,11 +33,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secrets/_models/secret.py b/python/databricks/bundles/secrets/_models/secret.py index b0db4c03ada..4bfbcbcf219 100644 --- a/python/databricks/bundles/secrets/_models/secret.py +++ b/python/databricks/bundles/secrets/_models/secret.py @@ -1,26 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.secrets._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.secrets._models.privilege import Privilege, PrivilegeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Secret(Resource): """ @@ -28,7 +24,6 @@ class Secret(Resource): (catalog.schema.secret) that securely store sensitive credential data such as passwords, tokens, and keys. """ - catalog_name: VariableOr[str] """ The name of the catalog where the schema and the secret reside. @@ -76,11 +71,12 @@ class Secret(Resource): """ @classmethod - def from_dict(cls, value: "SecretDict") -> "Self": + def from_dict(cls, value: 'SecretDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SecretDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SecretDict': + return _transform_to_json_value(self) # type:ignore + class SecretDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/__init__.py b/python/databricks/bundles/sql_warehouses/__init__.py index 4866d2ca86e..5a397048264 100644 --- a/python/databricks/bundles/sql_warehouses/__init__.py +++ b/python/databricks/bundles/sql_warehouses/__init__.py @@ -1,78 +1,46 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Channel", - "ChannelDict", - "ChannelName", - "ChannelNameParam", - "ChannelParam", - "CreateWarehouseRequestWarehouseType", - "CreateWarehouseRequestWarehouseTypeParam", - "EndpointTagPair", - "EndpointTagPairDict", - "EndpointTagPairParam", - "EndpointTags", - "EndpointTagsDict", - "EndpointTagsParam", - "LifecycleWithStarted", - "LifecycleWithStartedDict", - "LifecycleWithStartedParam", - "SpotInstancePolicy", - "SpotInstancePolicyParam", - "SqlWarehouse", - "SqlWarehouseDict", - "SqlWarehouseParam", - "SqlWarehousePermission", - "SqlWarehousePermissionDict", - "SqlWarehousePermissionParam", - "WarehousePermissionLevel", - "WarehousePermissionLevelParam", + 'Channel', + 'ChannelDict', + 'ChannelName', + 'ChannelNameParam', + 'ChannelParam', + 'CreateWarehouseRequestWarehouseType', + 'CreateWarehouseRequestWarehouseTypeParam', + 'EndpointTagPair', + 'EndpointTagPairDict', + 'EndpointTagPairParam', + 'EndpointTags', + 'EndpointTagsDict', + 'EndpointTagsParam', + 'LifecycleWithStarted', + 'LifecycleWithStartedDict', + 'LifecycleWithStartedParam', + 'SpotInstancePolicy', + 'SpotInstancePolicyParam', + 'SqlWarehouse', + 'SqlWarehouseDict', + 'SqlWarehouseParam', + 'SqlWarehousePermission', + 'SqlWarehousePermissionDict', + 'SqlWarehousePermissionParam', + 'WarehousePermissionLevel', + 'WarehousePermissionLevelParam', ] -from databricks.bundles.sql_warehouses._models.channel import ( - Channel, - ChannelDict, - ChannelParam, -) -from databricks.bundles.sql_warehouses._models.channel_name import ( - ChannelName, - ChannelNameParam, -) -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import ( - CreateWarehouseRequestWarehouseType, - CreateWarehouseRequestWarehouseTypeParam, -) -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import ( - EndpointTagPair, - EndpointTagPairDict, - EndpointTagPairParam, -) -from databricks.bundles.sql_warehouses._models.endpoint_tags import ( - EndpointTags, - EndpointTagsDict, - EndpointTagsParam, -) -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import ( - LifecycleWithStarted, - LifecycleWithStartedDict, - LifecycleWithStartedParam, -) -from databricks.bundles.sql_warehouses._models.spot_instance_policy import ( - SpotInstancePolicy, - SpotInstancePolicyParam, -) -from databricks.bundles.sql_warehouses._models.sql_warehouse import ( - SqlWarehouse, - SqlWarehouseDict, - SqlWarehouseParam, -) -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import ( - SqlWarehousePermission, - SqlWarehousePermissionDict, - SqlWarehousePermissionParam, -) -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import ( - WarehousePermissionLevel, - WarehousePermissionLevelParam, -) +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam + diff --git a/python/databricks/bundles/sql_warehouses/_models/channel.py b/python/databricks/bundles/sql_warehouses/_models/channel.py index f5893d7273f..367ed75dd55 100644 --- a/python/databricks/bundles/sql_warehouses/_models/channel.py +++ b/python/databricks/bundles/sql_warehouses/_models/channel.py @@ -1,36 +1,47 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional -from databricks.bundles.sql_warehouses._models.channel_name import ( - ChannelName, - ChannelNameParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Channel: """ Configures the channel name and DBSQL version of the warehouse. CHANNEL_NAME_CUSTOM should be chosen only when `dbsql_version` is specified. """ - dbsql_version: VariableOrOptional[str] = None name: VariableOrOptional[ChannelName] = None @classmethod - def from_dict(cls, value: "ChannelDict") -> "Self": + def from_dict(cls, value: 'ChannelDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "ChannelDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'ChannelDict': + return _transform_to_json_value(self) # type:ignore + class ChannelDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/channel_name.py b/python/databricks/bundles/sql_warehouses/_models/channel_name.py index a47e1c890f4..8d1f8fef0c3 100644 --- a/python/databricks/bundles/sql_warehouses/_models/channel_name.py +++ b/python/databricks/bundles/sql_warehouses/_models/channel_name.py @@ -1,7 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam class ChannelName(Enum): @@ -10,13 +28,4 @@ class ChannelName(Enum): CHANNEL_NAME_PREVIOUS = "CHANNEL_NAME_PREVIOUS" CHANNEL_NAME_CUSTOM = "CHANNEL_NAME_CUSTOM" - -ChannelNameParam = ( - Literal[ - "CHANNEL_NAME_PREVIEW", - "CHANNEL_NAME_CURRENT", - "CHANNEL_NAME_PREVIOUS", - "CHANNEL_NAME_CUSTOM", - ] - | ChannelName -) +ChannelNameParam = Literal["CHANNEL_NAME_PREVIEW", "CHANNEL_NAME_CURRENT", "CHANNEL_NAME_PREVIOUS", "CHANNEL_NAME_CUSTOM"] | ChannelName diff --git a/python/databricks/bundles/sql_warehouses/_models/create_warehouse_request_warehouse_type.py b/python/databricks/bundles/sql_warehouses/_models/create_warehouse_request_warehouse_type.py index feb20fbbbee..d4f26614cfa 100644 --- a/python/databricks/bundles/sql_warehouses/_models/create_warehouse_request_warehouse_type.py +++ b/python/databricks/bundles/sql_warehouses/_models/create_warehouse_request_warehouse_type.py @@ -1,7 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam class CreateWarehouseRequestWarehouseType(Enum): @@ -10,8 +28,4 @@ class CreateWarehouseRequestWarehouseType(Enum): PRO = "PRO" REALTIME = "REALTIME" - -CreateWarehouseRequestWarehouseTypeParam = ( - Literal["TYPE_UNSPECIFIED", "CLASSIC", "PRO", "REALTIME"] - | CreateWarehouseRequestWarehouseType -) +CreateWarehouseRequestWarehouseTypeParam = Literal["TYPE_UNSPECIFIED", "CLASSIC", "PRO", "REALTIME"] | CreateWarehouseRequestWarehouseType diff --git a/python/databricks/bundles/sql_warehouses/_models/endpoint_tag_pair.py b/python/databricks/bundles/sql_warehouses/_models/endpoint_tag_pair.py index 45f4d904d1e..b2ae77f8606 100644 --- a/python/databricks/bundles/sql_warehouses/_models/endpoint_tag_pair.py +++ b/python/databricks/bundles/sql_warehouses/_models/endpoint_tag_pair.py @@ -1,16 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EndpointTagPair: """""" @@ -20,11 +35,12 @@ class EndpointTagPair: value: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: "EndpointTagPairDict") -> "Self": + def from_dict(cls, value: 'EndpointTagPairDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EndpointTagPairDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EndpointTagPairDict': + return _transform_to_json_value(self) # type:ignore + class EndpointTagPairDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/endpoint_tags.py b/python/databricks/bundles/sql_warehouses/_models/endpoint_tags.py index 7d70c86199d..0d74becb5fd 100644 --- a/python/databricks/bundles/sql_warehouses/_models/endpoint_tags.py +++ b/python/databricks/bundles/sql_warehouses/_models/endpoint_tags.py @@ -1,20 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import ( - EndpointTagPair, - EndpointTagPairParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EndpointTags: """""" @@ -22,11 +33,12 @@ class EndpointTags: custom_tags: VariableOrList[EndpointTagPair] = field(default_factory=list) @classmethod - def from_dict(cls, value: "EndpointTagsDict") -> "Self": + def from_dict(cls, value: 'EndpointTagsDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EndpointTagsDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EndpointTagsDict': + return _transform_to_json_value(self) # type:ignore + class EndpointTagsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/lifecycle_with_started.py b/python/databricks/bundles/sql_warehouses/_models/lifecycle_with_started.py index f9cb9b8d49c..3092b166ce0 100644 --- a/python/databricks/bundles/sql_warehouses/_models/lifecycle_with_started.py +++ b/python/databricks/bundles/sql_warehouses/_models/lifecycle_with_started.py @@ -1,16 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class LifecycleWithStarted: """""" @@ -26,11 +41,12 @@ class LifecycleWithStarted: """ @classmethod - def from_dict(cls, value: "LifecycleWithStartedDict") -> "Self": + def from_dict(cls, value: 'LifecycleWithStartedDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleWithStartedDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleWithStartedDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleWithStartedDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/spot_instance_policy.py b/python/databricks/bundles/sql_warehouses/_models/spot_instance_policy.py index e0d5105a95f..6216f63086d 100644 --- a/python/databricks/bundles/sql_warehouses/_models/spot_instance_policy.py +++ b/python/databricks/bundles/sql_warehouses/_models/spot_instance_policy.py @@ -1,29 +1,42 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam class SpotInstancePolicy(Enum): """ EndpointSpotInstancePolicy configures whether the endpoint should use spot instances. - + The breakdown of how the EndpointSpotInstancePolicy converts to per cloud configurations is: - + - AWS, COST_OPTIMIZED: On Demand Driver with Spot Executors. - AWS, RELIABILITY_OPTIMIZED: On Demand Driver and Executors. - AZURE, COST_OPTIMIZED: On Demand Driver and Executors. - AZURE, RELIABILITY_OPTIMIZED: On Demand Driver and Executors. """ - POLICY_UNSPECIFIED = "POLICY_UNSPECIFIED" COST_OPTIMIZED = "COST_OPTIMIZED" RELIABILITY_OPTIMIZED = "RELIABILITY_OPTIMIZED" - -SpotInstancePolicyParam = ( - Literal["POLICY_UNSPECIFIED", "COST_OPTIMIZED", "RELIABILITY_OPTIMIZED"] - | SpotInstancePolicy -) +SpotInstancePolicyParam = Literal["POLICY_UNSPECIFIED", "COST_OPTIMIZED", "RELIABILITY_OPTIMIZED"] | SpotInstancePolicy diff --git a/python/databricks/bundles/sql_warehouses/_models/sql_warehouse.py b/python/databricks/bundles/sql_warehouses/_models/sql_warehouse.py index 3fe07aeadbe..caf96bb8b0e 100644 --- a/python/databricks/bundles/sql_warehouses/_models/sql_warehouse.py +++ b/python/databricks/bundles/sql_warehouses/_models/sql_warehouse.py @@ -1,44 +1,36 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelParam -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import ( - CreateWarehouseRequestWarehouseType, - CreateWarehouseRequestWarehouseTypeParam, -) -from databricks.bundles.sql_warehouses._models.endpoint_tags import ( - EndpointTags, - EndpointTagsParam, -) -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import ( - LifecycleWithStarted, - LifecycleWithStartedParam, -) -from databricks.bundles.sql_warehouses._models.spot_instance_policy import ( - SpotInstancePolicy, - SpotInstancePolicyParam, -) -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import ( - SqlWarehousePermission, - SqlWarehousePermissionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SqlWarehouse(Resource): """ Creates a new SQL warehouse. """ - auto_stop_mins: VariableOrOptional[int] = None """ The amount of time in minutes that a SQL warehouse must be idle (i.e., no @@ -103,6 +95,15 @@ class SqlWarehouse(Resource): Settings that control the deployment lifecycle of the resource, such as preventing it from being destroyed. """ + max_dbu: VariableOrOptional[int] = None + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] Maximum compute capacity of a real-time SQL warehouse, in DBU per hour. + Must be a multiple of 3 between 3 and 960. The effective maximum cannot be + lower than the warehouse's minimum capacity. + """ + max_num_clusters: VariableOrOptional[int] = None """ Maximum number of clusters that the autoscaler will create to handle @@ -144,6 +145,21 @@ class SqlWarehouse(Resource): The permissions to apply to this resource. """ + reyden_cluster_size_multiplier: VariableOrOptional[float] = None + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] Cluster size multiplier for Reyden warehouses. Scales the effective compute + capacity relative to the nominal virtual cluster size. Only meaningful when + the warehouse is of Reyden type. + + Accepted range: 1.0 to the workspace-configured maximum (decimals + allowed). The default maximum is 100.0. + - Values below 1.0 are normalized to 1.0. + - Values above the configured maximum are capped at that maximum. + - When absent, the warehouse is treated as having a multiplier of 1.0. + """ + spot_instance_policy: VariableOrOptional[SpotInstancePolicy] = None """ Configurations whether the endpoint should use spot instances. @@ -172,11 +188,12 @@ class SqlWarehouse(Resource): """ @classmethod - def from_dict(cls, value: "SqlWarehouseDict") -> "Self": + def from_dict(cls, value: 'SqlWarehouseDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SqlWarehouseDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SqlWarehouseDict': + return _transform_to_json_value(self) # type:ignore + class SqlWarehouseDict(TypedDict, total=False): @@ -246,6 +263,15 @@ class SqlWarehouseDict(TypedDict, total=False): Settings that control the deployment lifecycle of the resource, such as preventing it from being destroyed. """ + max_dbu: VariableOrOptional[int] + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] Maximum compute capacity of a real-time SQL warehouse, in DBU per hour. + Must be a multiple of 3 between 3 and 960. The effective maximum cannot be + lower than the warehouse's minimum capacity. + """ + max_num_clusters: VariableOrOptional[int] """ Maximum number of clusters that the autoscaler will create to handle @@ -287,6 +313,21 @@ class SqlWarehouseDict(TypedDict, total=False): The permissions to apply to this resource. """ + reyden_cluster_size_multiplier: VariableOrOptional[float] + """ + :meta private: [EXPERIMENTAL] + + [Private Preview] Cluster size multiplier for Reyden warehouses. Scales the effective compute + capacity relative to the nominal virtual cluster size. Only meaningful when + the warehouse is of Reyden type. + + Accepted range: 1.0 to the workspace-configured maximum (decimals + allowed). The default maximum is 100.0. + - Values below 1.0 are normalized to 1.0. + - Values above the configured maximum are capped at that maximum. + - When absent, the warehouse is treated as having a multiplier of 1.0. + """ + spot_instance_policy: VariableOrOptional[SpotInstancePolicyParam] """ Configurations whether the endpoint should use spot instances. diff --git a/python/databricks/bundles/sql_warehouses/_models/sql_warehouse_permission.py b/python/databricks/bundles/sql_warehouses/_models/sql_warehouse_permission.py index 2c9f284a2c0..559b5b9acf8 100644 --- a/python/databricks/bundles/sql_warehouses/_models/sql_warehouse_permission.py +++ b/python/databricks/bundles/sql_warehouses/_models/sql_warehouse_permission.py @@ -1,20 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import ( - WarehousePermissionLevel, - WarehousePermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam + +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SqlWarehousePermission: """""" @@ -40,11 +51,12 @@ class SqlWarehousePermission: """ @classmethod - def from_dict(cls, value: "SqlWarehousePermissionDict") -> "Self": + def from_dict(cls, value: 'SqlWarehousePermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SqlWarehousePermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SqlWarehousePermissionDict': + return _transform_to_json_value(self) # type:ignore + class SqlWarehousePermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/warehouse_permission_level.py b/python/databricks/bundles/sql_warehouses/_models/warehouse_permission_level.py index 5728daa0d12..385a19e3ff2 100644 --- a/python/databricks/bundles/sql_warehouses/_models/warehouse_permission_level.py +++ b/python/databricks/bundles/sql_warehouses/_models/warehouse_permission_level.py @@ -1,22 +1,35 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam +from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam +from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam +from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam + +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam + +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam class WarehousePermissionLevel(Enum): """ Permission level """ - CAN_MANAGE = "CAN_MANAGE" IS_OWNER = "IS_OWNER" CAN_USE = "CAN_USE" CAN_MONITOR = "CAN_MONITOR" CAN_VIEW = "CAN_VIEW" - -WarehousePermissionLevelParam = ( - Literal["CAN_MANAGE", "IS_OWNER", "CAN_USE", "CAN_MONITOR", "CAN_VIEW"] - | WarehousePermissionLevel -) +WarehousePermissionLevelParam = Literal["CAN_MANAGE", "IS_OWNER", "CAN_USE", "CAN_MONITOR", "CAN_VIEW"] | WarehousePermissionLevel diff --git a/python/databricks/bundles/synced_database_tables/__init__.py b/python/databricks/bundles/synced_database_tables/__init__.py index 0b3a240e460..a0729cab266 100644 --- a/python/databricks/bundles/synced_database_tables/__init__.py +++ b/python/databricks/bundles/synced_database_tables/__init__.py @@ -1,58 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "NewPipelineSpec", - "NewPipelineSpecDict", - "NewPipelineSpecParam", - "SyncedDatabaseTable", - "SyncedDatabaseTableDict", - "SyncedDatabaseTableParam", - "SyncedTableSchedulingPolicy", - "SyncedTableSchedulingPolicyParam", - "SyncedTableSpec", - "SyncedTableSpecDict", - "SyncedTableSpecParam", - "SyncedTableSpecPgSpecificType", - "SyncedTableSpecPgSpecificTypeParam", - "SyncedTableSpecTypeOverride", - "SyncedTableSpecTypeOverrideDict", - "SyncedTableSpecTypeOverrideParam", + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'NewPipelineSpec', + 'NewPipelineSpecDict', + 'NewPipelineSpecParam', + 'SyncedDatabaseTable', + 'SyncedDatabaseTableDict', + 'SyncedDatabaseTableParam', + 'SyncedTableSchedulingPolicy', + 'SyncedTableSchedulingPolicyParam', + 'SyncedTableSpec', + 'SyncedTableSpecDict', + 'SyncedTableSpecParam', + 'SyncedTableSpecPgSpecificType', + 'SyncedTableSpecPgSpecificTypeParam', + 'SyncedTableSpecTypeOverride', + 'SyncedTableSpecTypeOverrideDict', + 'SyncedTableSpecTypeOverrideParam', ] -from databricks.bundles.synced_database_tables._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.synced_database_tables._models.new_pipeline_spec import ( - NewPipelineSpec, - NewPipelineSpecDict, - NewPipelineSpecParam, -) -from databricks.bundles.synced_database_tables._models.synced_database_table import ( - SyncedDatabaseTable, - SyncedDatabaseTableDict, - SyncedDatabaseTableParam, -) -from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import ( - SyncedTableSchedulingPolicy, - SyncedTableSchedulingPolicyParam, -) -from databricks.bundles.synced_database_tables._models.synced_table_spec import ( - SyncedTableSpec, - SyncedTableSpecDict, - SyncedTableSpecParam, -) -from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import ( - SyncedTableSpecPgSpecificType, - SyncedTableSpecPgSpecificTypeParam, -) -from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import ( - SyncedTableSpecTypeOverride, - SyncedTableSpecTypeOverrideDict, - SyncedTableSpecTypeOverrideParam, -) +from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam +from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam +from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam + +from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam + diff --git a/python/databricks/bundles/synced_database_tables/_models/lifecycle.py b/python/databricks/bundles/synced_database_tables/_models/lifecycle.py index 697776a198e..cbfe5e2bb92 100644 --- a/python/databricks/bundles/synced_database_tables/_models/lifecycle.py +++ b/python/databricks/bundles/synced_database_tables/_models/lifecycle.py @@ -1,16 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam +from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam +from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam + +from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +31,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/synced_database_tables/_models/new_pipeline_spec.py b/python/databricks/bundles/synced_database_tables/_models/new_pipeline_spec.py index 9a47da58441..cf5896231ff 100644 --- a/python/databricks/bundles/synced_database_tables/_models/new_pipeline_spec.py +++ b/python/databricks/bundles/synced_database_tables/_models/new_pipeline_spec.py @@ -1,23 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam +from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam +from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam + +from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class NewPipelineSpec: """ Custom fields that user can set for pipeline while creating SyncedDatabaseTable. Note that other fields of pipeline are still inferred by table def internally """ - budget_policy_id: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -42,11 +51,12 @@ class NewPipelineSpec: """ @classmethod - def from_dict(cls, value: "NewPipelineSpecDict") -> "Self": + def from_dict(cls, value: 'NewPipelineSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "NewPipelineSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'NewPipelineSpecDict': + return _transform_to_json_value(self) # type:ignore + class NewPipelineSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_database_table.py b/python/databricks/bundles/synced_database_tables/_models/synced_database_table.py index c98b869b76e..0efeacfaec9 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_database_table.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_database_table.py @@ -1,25 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.synced_database_tables._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.synced_database_tables._models.synced_table_spec import ( - SyncedTableSpec, - SyncedTableSpecParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam +from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam + +from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SyncedDatabaseTable(Resource): """""" @@ -62,11 +63,12 @@ class SyncedDatabaseTable(Resource): """ @classmethod - def from_dict(cls, value: "SyncedDatabaseTableDict") -> "Self": + def from_dict(cls, value: 'SyncedDatabaseTableDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SyncedDatabaseTableDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SyncedDatabaseTableDict': + return _transform_to_json_value(self) # type:ignore + class SyncedDatabaseTableDict(TypedDict, total=False): diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_table_scheduling_policy.py b/python/databricks/bundles/synced_database_tables/_models/synced_table_scheduling_policy.py index 379d61c2c17..d2f6f64abb2 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_table_scheduling_policy.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_table_scheduling_policy.py @@ -1,7 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam +from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam +from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam class SyncedTableSchedulingPolicy(Enum): @@ -9,7 +22,4 @@ class SyncedTableSchedulingPolicy(Enum): TRIGGERED = "TRIGGERED" SNAPSHOT = "SNAPSHOT" - -SyncedTableSchedulingPolicyParam = ( - Literal["CONTINUOUS", "TRIGGERED", "SNAPSHOT"] | SyncedTableSchedulingPolicy -) +SyncedTableSchedulingPolicyParam = Literal["CONTINUOUS", "TRIGGERED", "SNAPSHOT"] | SyncedTableSchedulingPolicy diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec.py b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec.py index c6be97882ae..bd32c91936f 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec.py @@ -1,34 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.synced_database_tables._models.new_pipeline_spec import ( - NewPipelineSpec, - NewPipelineSpecParam, -) -from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import ( - SyncedTableSchedulingPolicy, - SyncedTableSchedulingPolicyParam, -) -from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import ( - SyncedTableSpecTypeOverride, - SyncedTableSpecTypeOverrideParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam +from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam +from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam + +from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SyncedTableSpec: """ Specification of a synced database table. """ - accelerated_sync: VariableOrOptional[bool] = None """ :meta private: [EXPERIMENTAL] @@ -83,9 +80,7 @@ class SyncedTableSpec: [Public Preview] Time series key to deduplicate (tie-break) rows with the same primary key. """ - type_overrides: VariableOrList[SyncedTableSpecTypeOverride] = field( - default_factory=list - ) + type_overrides: VariableOrList[SyncedTableSpecTypeOverride] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -94,11 +89,12 @@ class SyncedTableSpec: """ @classmethod - def from_dict(cls, value: "SyncedTableSpecDict") -> "Self": + def from_dict(cls, value: 'SyncedTableSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SyncedTableSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SyncedTableSpecDict': + return _transform_to_json_value(self) # type:ignore + class SyncedTableSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_pg_specific_type.py b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_pg_specific_type.py index 52c3b8dbe9a..e77b35923e7 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_pg_specific_type.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_pg_specific_type.py @@ -1,26 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam +from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam +from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam class SyncedTableSpecPgSpecificType(Enum): """ :meta private: [EXPERIMENTAL] - + PostgreSQL-specific target types that can override the default Delta-to-PG mapping. """ - PG_SPECIFIC_TYPE_VECTOR = "PG_SPECIFIC_TYPE_VECTOR" PG_SPECIFIC_TYPE_HALFVEC = "PG_SPECIFIC_TYPE_HALFVEC" PG_SPECIFIC_TYPE_VARCHAR = "PG_SPECIFIC_TYPE_VARCHAR" - -SyncedTableSpecPgSpecificTypeParam = ( - Literal[ - "PG_SPECIFIC_TYPE_VECTOR", - "PG_SPECIFIC_TYPE_HALFVEC", - "PG_SPECIFIC_TYPE_VARCHAR", - ] - | SyncedTableSpecPgSpecificType -) +SyncedTableSpecPgSpecificTypeParam = Literal["PG_SPECIFIC_TYPE_VECTOR", "PG_SPECIFIC_TYPE_HALFVEC", "PG_SPECIFIC_TYPE_VARCHAR"] | SyncedTableSpecPgSpecificType diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_type_override.py b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_type_override.py index 5cc3fa72720..63100d41d37 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_type_override.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_type_override.py @@ -1,28 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import ( - SyncedTableSpecPgSpecificType, - SyncedTableSpecPgSpecificTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam +from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam +from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam +from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam + +from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class SyncedTableSpecTypeOverride: """ :meta private: [EXPERIMENTAL] - + Overrides the default Delta-to-PostgreSQL type mapping for a single column. """ - column_name: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -46,11 +51,12 @@ class SyncedTableSpecTypeOverride: """ @classmethod - def from_dict(cls, value: "SyncedTableSpecTypeOverrideDict") -> "Self": + def from_dict(cls, value: 'SyncedTableSpecTypeOverrideDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "SyncedTableSpecTypeOverrideDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'SyncedTableSpecTypeOverrideDict': + return _transform_to_json_value(self) # type:ignore + class SyncedTableSpecTypeOverrideDict(TypedDict, total=False): @@ -79,6 +85,4 @@ class SyncedTableSpecTypeOverrideDict(TypedDict, total=False): """ -SyncedTableSpecTypeOverrideParam = ( - SyncedTableSpecTypeOverrideDict | SyncedTableSpecTypeOverride -) +SyncedTableSpecTypeOverrideParam = SyncedTableSpecTypeOverrideDict | SyncedTableSpecTypeOverride diff --git a/python/databricks/bundles/vector_search_endpoints/__init__.py b/python/databricks/bundles/vector_search_endpoints/__init__.py index aabc0081f5d..223e3cf67b2 100644 --- a/python/databricks/bundles/vector_search_endpoints/__init__.py +++ b/python/databricks/bundles/vector_search_endpoints/__init__.py @@ -1,42 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "EndpointType", - "EndpointTypeParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "VectorSearchEndpoint", - "VectorSearchEndpointDict", - "VectorSearchEndpointParam", - "VectorSearchEndpointPermission", - "VectorSearchEndpointPermissionDict", - "VectorSearchEndpointPermissionLevel", - "VectorSearchEndpointPermissionLevelParam", - "VectorSearchEndpointPermissionParam", + 'EndpointType', + 'EndpointTypeParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'VectorSearchEndpoint', + 'VectorSearchEndpointDict', + 'VectorSearchEndpointParam', + 'VectorSearchEndpointPermission', + 'VectorSearchEndpointPermissionDict', + 'VectorSearchEndpointPermissionLevel', + 'VectorSearchEndpointPermissionLevelParam', + 'VectorSearchEndpointPermissionParam', ] -from databricks.bundles.vector_search_endpoints._models.endpoint_type import ( - EndpointType, - EndpointTypeParam, -) -from databricks.bundles.vector_search_endpoints._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( - VectorSearchEndpoint, - VectorSearchEndpointDict, - VectorSearchEndpointParam, -) -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import ( - VectorSearchEndpointPermission, - VectorSearchEndpointPermissionDict, - VectorSearchEndpointPermissionParam, -) -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import ( - VectorSearchEndpointPermissionLevel, - VectorSearchEndpointPermissionLevelParam, -) +from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam +from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam + +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam + diff --git a/python/databricks/bundles/vector_search_endpoints/_models/endpoint_type.py b/python/databricks/bundles/vector_search_endpoints/_models/endpoint_type.py index 476295df34b..a26e8dadddf 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/endpoint_type.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/endpoint_type.py @@ -1,16 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam class EndpointType(Enum): """ Type of endpoint. """ - STORAGE_OPTIMIZED = "STORAGE_OPTIMIZED" STANDARD = "STANDARD" - EndpointTypeParam = Literal["STORAGE_OPTIMIZED", "STANDARD"] | EndpointType diff --git a/python/databricks/bundles/vector_search_endpoints/_models/lifecycle.py b/python/databricks/bundles/vector_search_endpoints/_models/lifecycle.py index 697776a198e..0d29e52955b 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/lifecycle.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/lifecycle.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam +from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam + +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +29,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint.py b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint.py index 8072329f47b..d01810ba304 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint.py @@ -1,33 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.vector_search_endpoints._models.endpoint_type import ( - EndpointType, - EndpointTypeParam, -) -from databricks.bundles.vector_search_endpoints._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import ( - VectorSearchEndpointPermission, - VectorSearchEndpointPermissionParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam +from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam + +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class VectorSearchEndpoint(Resource): """""" @@ -52,9 +43,7 @@ class VectorSearchEndpoint(Resource): Settings that control the deployment lifecycle of the resource, such as preventing it from being destroyed. """ - permissions: VariableOrList[VectorSearchEndpointPermission] = field( - default_factory=list - ) + permissions: VariableOrList[VectorSearchEndpointPermission] = field(default_factory=list) """ The permissions to apply to this resource. """ @@ -74,11 +63,12 @@ class VectorSearchEndpoint(Resource): """ @classmethod - def from_dict(cls, value: "VectorSearchEndpointDict") -> "Self": + def from_dict(cls, value: 'VectorSearchEndpointDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "VectorSearchEndpointDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'VectorSearchEndpointDict': + return _transform_to_json_value(self) # type:ignore + class VectorSearchEndpointDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission.py b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission.py index cfbdc1e31d6..7d779ac64f0 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission.py @@ -1,20 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import ( - VectorSearchEndpointPermissionLevel, - VectorSearchEndpointPermissionLevelParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam +from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam + +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class VectorSearchEndpointPermission: """""" @@ -28,11 +32,12 @@ class VectorSearchEndpointPermission: user_name: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: "VectorSearchEndpointPermissionDict") -> "Self": + def from_dict(cls, value: 'VectorSearchEndpointPermissionDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "VectorSearchEndpointPermissionDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'VectorSearchEndpointPermissionDict': + return _transform_to_json_value(self) # type:ignore + class VectorSearchEndpointPermissionDict(TypedDict, total=False): @@ -47,6 +52,4 @@ class VectorSearchEndpointPermissionDict(TypedDict, total=False): user_name: VariableOrOptional[str] -VectorSearchEndpointPermissionParam = ( - VectorSearchEndpointPermissionDict | VectorSearchEndpointPermission -) +VectorSearchEndpointPermissionParam = VectorSearchEndpointPermissionDict | VectorSearchEndpointPermission diff --git a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission_level.py b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission_level.py index d2618d39126..406d260f4d7 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission_level.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission_level.py @@ -1,19 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam +from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam class VectorSearchEndpointPermissionLevel(Enum): """ Permission level """ - CAN_CREATE = "CAN_CREATE" CAN_MANAGE = "CAN_MANAGE" CAN_USE = "CAN_USE" - -VectorSearchEndpointPermissionLevelParam = ( - Literal["CAN_CREATE", "CAN_MANAGE", "CAN_USE"] | VectorSearchEndpointPermissionLevel -) +VectorSearchEndpointPermissionLevelParam = Literal["CAN_CREATE", "CAN_MANAGE", "CAN_USE"] | VectorSearchEndpointPermissionLevel diff --git a/python/databricks/bundles/vector_search_indexes/__init__.py b/python/databricks/bundles/vector_search_indexes/__init__.py index 8e3c88eb018..678b3508272 100644 --- a/python/databricks/bundles/vector_search_indexes/__init__.py +++ b/python/databricks/bundles/vector_search_indexes/__init__.py @@ -1,86 +1,50 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "DeltaSyncVectorIndexSpecRequest", - "DeltaSyncVectorIndexSpecRequestDict", - "DeltaSyncVectorIndexSpecRequestParam", - "DirectAccessVectorIndexSpec", - "DirectAccessVectorIndexSpecDict", - "DirectAccessVectorIndexSpecParam", - "EmbeddingSourceColumn", - "EmbeddingSourceColumnDict", - "EmbeddingSourceColumnParam", - "EmbeddingVectorColumn", - "EmbeddingVectorColumnDict", - "EmbeddingVectorColumnParam", - "IndexSubtype", - "IndexSubtypeParam", - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "PipelineType", - "PipelineTypeParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", - "VectorIndexType", - "VectorIndexTypeParam", - "VectorSearchIndex", - "VectorSearchIndexDict", - "VectorSearchIndexParam", + 'DeltaSyncVectorIndexSpecRequest', + 'DeltaSyncVectorIndexSpecRequestDict', + 'DeltaSyncVectorIndexSpecRequestParam', + 'DirectAccessVectorIndexSpec', + 'DirectAccessVectorIndexSpecDict', + 'DirectAccessVectorIndexSpecParam', + 'EmbeddingSourceColumn', + 'EmbeddingSourceColumnDict', + 'EmbeddingSourceColumnParam', + 'EmbeddingVectorColumn', + 'EmbeddingVectorColumnDict', + 'EmbeddingVectorColumnParam', + 'IndexSubtype', + 'IndexSubtypeParam', + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'PipelineType', + 'PipelineTypeParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'VectorIndexType', + 'VectorIndexTypeParam', + 'VectorSearchIndex', + 'VectorSearchIndexDict', + 'VectorSearchIndexParam', ] -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import ( - DeltaSyncVectorIndexSpecRequest, - DeltaSyncVectorIndexSpecRequestDict, - DeltaSyncVectorIndexSpecRequestParam, -) -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import ( - DirectAccessVectorIndexSpec, - DirectAccessVectorIndexSpecDict, - DirectAccessVectorIndexSpecParam, -) -from databricks.bundles.vector_search_indexes._models.embedding_source_column import ( - EmbeddingSourceColumn, - EmbeddingSourceColumnDict, - EmbeddingSourceColumnParam, -) -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import ( - EmbeddingVectorColumn, - EmbeddingVectorColumnDict, - EmbeddingVectorColumnParam, -) -from databricks.bundles.vector_search_indexes._models.index_subtype import ( - IndexSubtype, - IndexSubtypeParam, -) -from databricks.bundles.vector_search_indexes._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.vector_search_indexes._models.pipeline_type import ( - PipelineType, - PipelineTypeParam, -) -from databricks.bundles.vector_search_indexes._models.privilege import ( - Privilege, - PrivilegeParam, -) -from databricks.bundles.vector_search_indexes._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) -from databricks.bundles.vector_search_indexes._models.vector_index_type import ( - VectorIndexType, - VectorIndexTypeParam, -) -from databricks.bundles.vector_search_indexes._models.vector_search_index import ( - VectorSearchIndex, - VectorSearchIndexDict, - VectorSearchIndexParam, -) +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam + diff --git a/python/databricks/bundles/vector_search_indexes/_models/delta_sync_vector_index_spec_request.py b/python/databricks/bundles/vector_search_indexes/_models/delta_sync_vector_index_spec_request.py index b7af851711b..ae2c5f6968d 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/delta_sync_vector_index_spec_request.py +++ b/python/databricks/bundles/vector_search_indexes/_models/delta_sync_vector_index_spec_request.py @@ -1,28 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.vector_search_indexes._models.embedding_source_column import ( - EmbeddingSourceColumn, - EmbeddingSourceColumnParam, -) -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import ( - EmbeddingVectorColumn, - EmbeddingVectorColumnParam, -) -from databricks.bundles.vector_search_indexes._models.pipeline_type import ( - PipelineType, - PipelineTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DeltaSyncVectorIndexSpecRequest: """""" @@ -42,16 +46,12 @@ class DeltaSyncVectorIndexSpecRequest: embedding vector column are always synced. """ - embedding_source_columns: VariableOrList[EmbeddingSourceColumn] = field( - default_factory=list - ) + embedding_source_columns: VariableOrList[EmbeddingSourceColumn] = field(default_factory=list) """ The columns that contain the embedding source. """ - embedding_vector_columns: VariableOrList[EmbeddingVectorColumn] = field( - default_factory=list - ) + embedding_vector_columns: VariableOrList[EmbeddingVectorColumn] = field(default_factory=list) """ The columns that contain the embedding vectors. """ @@ -74,11 +74,12 @@ class DeltaSyncVectorIndexSpecRequest: """ @classmethod - def from_dict(cls, value: "DeltaSyncVectorIndexSpecRequestDict") -> "Self": + def from_dict(cls, value: 'DeltaSyncVectorIndexSpecRequestDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DeltaSyncVectorIndexSpecRequestDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DeltaSyncVectorIndexSpecRequestDict': + return _transform_to_json_value(self) # type:ignore + class DeltaSyncVectorIndexSpecRequestDict(TypedDict, total=False): @@ -127,6 +128,4 @@ class DeltaSyncVectorIndexSpecRequestDict(TypedDict, total=False): """ -DeltaSyncVectorIndexSpecRequestParam = ( - DeltaSyncVectorIndexSpecRequestDict | DeltaSyncVectorIndexSpecRequest -) +DeltaSyncVectorIndexSpecRequestParam = DeltaSyncVectorIndexSpecRequestDict | DeltaSyncVectorIndexSpecRequest diff --git a/python/databricks/bundles/vector_search_indexes/_models/direct_access_vector_index_spec.py b/python/databricks/bundles/vector_search_indexes/_models/direct_access_vector_index_spec.py index 76ef31b4b30..426995a3b09 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/direct_access_vector_index_spec.py +++ b/python/databricks/bundles/vector_search_indexes/_models/direct_access_vector_index_spec.py @@ -1,38 +1,42 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.vector_search_indexes._models.embedding_source_column import ( - EmbeddingSourceColumn, - EmbeddingSourceColumnParam, -) -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import ( - EmbeddingVectorColumn, - EmbeddingVectorColumnParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class DirectAccessVectorIndexSpec: """""" - embedding_source_columns: VariableOrList[EmbeddingSourceColumn] = field( - default_factory=list - ) + embedding_source_columns: VariableOrList[EmbeddingSourceColumn] = field(default_factory=list) """ The columns that contain the embedding source. The format should be array[double]. """ - embedding_vector_columns: VariableOrList[EmbeddingVectorColumn] = field( - default_factory=list - ) + embedding_vector_columns: VariableOrList[EmbeddingVectorColumn] = field(default_factory=list) """ The columns that contain the embedding vectors. The format should be array[double]. """ @@ -45,11 +49,12 @@ class DirectAccessVectorIndexSpec: """ @classmethod - def from_dict(cls, value: "DirectAccessVectorIndexSpecDict") -> "Self": + def from_dict(cls, value: 'DirectAccessVectorIndexSpecDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "DirectAccessVectorIndexSpecDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'DirectAccessVectorIndexSpecDict': + return _transform_to_json_value(self) # type:ignore + class DirectAccessVectorIndexSpecDict(TypedDict, total=False): @@ -73,6 +78,4 @@ class DirectAccessVectorIndexSpecDict(TypedDict, total=False): """ -DirectAccessVectorIndexSpecParam = ( - DirectAccessVectorIndexSpecDict | DirectAccessVectorIndexSpec -) +DirectAccessVectorIndexSpecParam = DirectAccessVectorIndexSpecDict | DirectAccessVectorIndexSpec diff --git a/python/databricks/bundles/vector_search_indexes/_models/embedding_source_column.py b/python/databricks/bundles/vector_search_indexes/_models/embedding_source_column.py index 7debdc3f104..26371674396 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/embedding_source_column.py +++ b/python/databricks/bundles/vector_search_indexes/_models/embedding_source_column.py @@ -1,16 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EmbeddingSourceColumn: """""" @@ -31,11 +47,12 @@ class EmbeddingSourceColumn: """ @classmethod - def from_dict(cls, value: "EmbeddingSourceColumnDict") -> "Self": + def from_dict(cls, value: 'EmbeddingSourceColumnDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EmbeddingSourceColumnDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EmbeddingSourceColumnDict': + return _transform_to_json_value(self) # type:ignore + class EmbeddingSourceColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_indexes/_models/embedding_vector_column.py b/python/databricks/bundles/vector_search_indexes/_models/embedding_vector_column.py index b8efd6b4f0c..fd9076bc3ed 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/embedding_vector_column.py +++ b/python/databricks/bundles/vector_search_indexes/_models/embedding_vector_column.py @@ -1,16 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class EmbeddingVectorColumn: """""" @@ -26,11 +42,12 @@ class EmbeddingVectorColumn: """ @classmethod - def from_dict(cls, value: "EmbeddingVectorColumnDict") -> "Self": + def from_dict(cls, value: 'EmbeddingVectorColumnDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "EmbeddingVectorColumnDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'EmbeddingVectorColumnDict': + return _transform_to_json_value(self) # type:ignore + class EmbeddingVectorColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_indexes/_models/index_subtype.py b/python/databricks/bundles/vector_search_indexes/_models/index_subtype.py index 5485754d1e6..2dcf22c9d22 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/index_subtype.py +++ b/python/databricks/bundles/vector_search_indexes/_models/index_subtype.py @@ -1,7 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam class IndexSubtype(Enum): @@ -11,10 +30,8 @@ class IndexSubtype(Enum): - `FULL_TEXT`: An index that uses full-text search without vector embeddings. - `HYBRID`: An index that uses vector embeddings for similarity search and hybrid search. """ - VECTOR = "VECTOR" FULL_TEXT = "FULL_TEXT" HYBRID = "HYBRID" - IndexSubtypeParam = Literal["VECTOR", "FULL_TEXT", "HYBRID"] | IndexSubtype diff --git a/python/databricks/bundles/vector_search_indexes/_models/lifecycle.py b/python/databricks/bundles/vector_search_indexes/_models/lifecycle.py index 697776a198e..b090b45839d 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/lifecycle.py +++ b/python/databricks/bundles/vector_search_indexes/_models/lifecycle.py @@ -1,16 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +37,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_indexes/_models/pipeline_type.py b/python/databricks/bundles/vector_search_indexes/_models/pipeline_type.py index 6821dec812e..20898efdfb7 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/pipeline_type.py +++ b/python/databricks/bundles/vector_search_indexes/_models/pipeline_type.py @@ -1,7 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam class PipelineType(Enum): @@ -10,9 +29,7 @@ class PipelineType(Enum): - `TRIGGERED`: If the pipeline uses the triggered execution mode, the system stops processing after successfully refreshing the source table in the pipeline once, ensuring the table is updated based on the data available when the update started. - `CONTINUOUS`: If the pipeline uses continuous execution, the pipeline processes new data as it arrives in the source table to keep vector index fresh. """ - TRIGGERED = "TRIGGERED" CONTINUOUS = "CONTINUOUS" - PipelineTypeParam = Literal["TRIGGERED", "CONTINUOUS"] | PipelineType diff --git a/python/databricks/bundles/vector_search_indexes/_models/privilege.py b/python/databricks/bundles/vector_search_indexes/_models/privilege.py index df45211a76a..d6a59038647 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/privilege.py +++ b/python/databricks/bundles/vector_search_indexes/_models/privilege.py @@ -1,7 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam class Privilege(Enum): @@ -58,61 +77,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/vector_search_indexes/_models/privilege_assignment.py b/python/databricks/bundles/vector_search_indexes/_models/privilege_assignment.py index 55100240cc4..08748413bfd 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/privilege_assignment.py +++ b/python/databricks/bundles/vector_search_indexes/_models/privilege_assignment.py @@ -1,20 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional -from databricks.bundles.vector_search_indexes._models.privilege import ( - Privilege, - PrivilegeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -31,11 +43,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_indexes/_models/vector_index_type.py b/python/databricks/bundles/vector_search_indexes/_models/vector_index_type.py index 43b8550c8b2..32d9790a05e 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/vector_index_type.py +++ b/python/databricks/bundles/vector_search_indexes/_models/vector_index_type.py @@ -1,7 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam class VectorIndexType(Enum): @@ -10,9 +29,7 @@ class VectorIndexType(Enum): - `DELTA_SYNC`: An index that automatically syncs with a source Delta Table, automatically and incrementally updating the index as the underlying data in the Delta Table changes. - `DIRECT_ACCESS`: An index that supports direct read and write of vectors and metadata through our REST and SDK APIs. With this model, the user manages index updates. """ - DELTA_SYNC = "DELTA_SYNC" DIRECT_ACCESS = "DIRECT_ACCESS" - VectorIndexTypeParam = Literal["DELTA_SYNC", "DIRECT_ACCESS"] | VectorIndexType diff --git a/python/databricks/bundles/vector_search_indexes/_models/vector_search_index.py b/python/databricks/bundles/vector_search_indexes/_models/vector_search_index.py index dba21478b1f..1297e366dd2 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/vector_search_index.py +++ b/python/databricks/bundles/vector_search_indexes/_models/vector_search_index.py @@ -1,45 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import ( - DeltaSyncVectorIndexSpecRequest, - DeltaSyncVectorIndexSpecRequestParam, -) -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import ( - DirectAccessVectorIndexSpec, - DirectAccessVectorIndexSpecParam, -) -from databricks.bundles.vector_search_indexes._models.index_subtype import ( - IndexSubtype, - IndexSubtypeParam, -) -from databricks.bundles.vector_search_indexes._models.lifecycle import ( - Lifecycle, - LifecycleParam, -) -from databricks.bundles.vector_search_indexes._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) -from databricks.bundles.vector_search_indexes._models.vector_index_type import ( - VectorIndexType, - VectorIndexTypeParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam +from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam + +from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam + +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class VectorSearchIndex(Resource): """""" @@ -94,11 +81,12 @@ class VectorSearchIndex(Resource): """ @classmethod - def from_dict(cls, value: "VectorSearchIndexDict") -> "Self": + def from_dict(cls, value: 'VectorSearchIndexDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "VectorSearchIndexDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'VectorSearchIndexDict': + return _transform_to_json_value(self) # type:ignore + class VectorSearchIndexDict(TypedDict, total=False): diff --git a/python/databricks/bundles/volumes/__init__.py b/python/databricks/bundles/volumes/__init__.py index 97bc1d54bd7..38bf7b6db23 100644 --- a/python/databricks/bundles/volumes/__init__.py +++ b/python/databricks/bundles/volumes/__init__.py @@ -1,41 +1,35 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - "Lifecycle", - "LifecycleDict", - "LifecycleParam", - "Privilege", - "PrivilegeAssignment", - "PrivilegeAssignmentDict", - "PrivilegeAssignmentParam", - "PrivilegeParam", - "Volume", - "VolumeDict", - "VolumeGrant", - "VolumeGrantDict", - "VolumeGrantParam", - "VolumeGrantPrivilege", - "VolumeGrantPrivilegeParam", - "VolumeParam", - "VolumeType", - "VolumeTypeParam", + 'Lifecycle', + 'LifecycleDict', + 'LifecycleParam', + 'Privilege', + 'PrivilegeAssignment', + 'PrivilegeAssignmentDict', + 'PrivilegeAssignmentParam', + 'PrivilegeParam', + 'Volume', + 'VolumeDict', + 'VolumeParam', + 'VolumeType', + 'VolumeTypeParam', + 'VolumeGrant', + 'VolumeGrantDict', + 'VolumeGrantParam', + 'VolumeGrantPrivilege', + 'VolumeGrantPrivilegeParam', ] -from databricks.bundles.volumes._models.lifecycle import ( - Lifecycle, - LifecycleDict, - LifecycleParam, -) -from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam -from databricks.bundles.volumes._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentDict, - PrivilegeAssignmentParam, -) +from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam +from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam + from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam + VolumeGrant = PrivilegeAssignment VolumeGrantDict = PrivilegeAssignmentDict VolumeGrantParam = PrivilegeAssignmentParam diff --git a/python/databricks/bundles/volumes/_models/lifecycle.py b/python/databricks/bundles/volumes/_models/lifecycle.py index 697776a198e..1a826fe2134 100644 --- a/python/databricks/bundles/volumes/_models/lifecycle.py +++ b/python/databricks/bundles/volumes/_models/lifecycle.py @@ -1,16 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam +from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam + +from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Lifecycle: """""" @@ -21,11 +29,12 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: "LifecycleDict") -> "Self": + def from_dict(cls, value: 'LifecycleDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "LifecycleDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'LifecycleDict': + return _transform_to_json_value(self) # type:ignore + class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/volumes/_models/privilege.py b/python/databricks/bundles/volumes/_models/privilege.py index df45211a76a..88dbf104c93 100644 --- a/python/databricks/bundles/volumes/_models/privilege.py +++ b/python/databricks/bundles/volumes/_models/privilege.py @@ -1,7 +1,18 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam +from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam class Privilege(Enum): @@ -58,61 +69,4 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" - -PrivilegeParam = ( - Literal[ - "SELECT", - "READ_PRIVATE_FILES", - "WRITE_PRIVATE_FILES", - "CREATE", - "USAGE", - "USE_CATALOG", - "USE_SCHEMA", - "CREATE_SCHEMA", - "CREATE_VIEW", - "CREATE_EXTERNAL_TABLE", - "CREATE_MATERIALIZED_VIEW", - "CREATE_FUNCTION", - "CREATE_MODEL", - "CREATE_CATALOG", - "CREATE_MANAGED_STORAGE", - "CREATE_EXTERNAL_LOCATION", - "CREATE_STORAGE_CREDENTIAL", - "CREATE_SERVICE_CREDENTIAL", - "ACCESS", - "CREATE_SHARE", - "CREATE_RECIPIENT", - "CREATE_PROVIDER", - "USE_SHARE", - "USE_RECIPIENT", - "USE_PROVIDER", - "USE_MARKETPLACE_ASSETS", - "SET_SHARE_PERMISSION", - "MODIFY", - "REFRESH", - "EXECUTE", - "READ_FILES", - "WRITE_FILES", - "CREATE_TABLE", - "ALL_PRIVILEGES", - "CREATE_CONNECTION", - "USE_CONNECTION", - "APPLY_TAG", - "CREATE_FOREIGN_CATALOG", - "CREATE_FOREIGN_SECURABLE", - "MANAGE_ALLOWLIST", - "CREATE_VOLUME", - "CREATE_EXTERNAL_VOLUME", - "READ_VOLUME", - "WRITE_VOLUME", - "MANAGE", - "BROWSE", - "CREATE_CLEAN_ROOM", - "MODIFY_CLEAN_ROOM", - "EXECUTE_CLEAN_ROOM_TASK", - "EXTERNAL_USE_SCHEMA", - "READ_METADATA", - "EXTERNAL_USE_LOCATION", - ] - | Privilege -) +PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege diff --git a/python/databricks/bundles/volumes/_models/privilege_assignment.py b/python/databricks/bundles/volumes/_models/privilege_assignment.py index 612e9d85eb0..b09f932ef76 100644 --- a/python/databricks/bundles/volumes/_models/privilege_assignment.py +++ b/python/databricks/bundles/volumes/_models/privilege_assignment.py @@ -1,17 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field +from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam +from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam + + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -28,11 +35,12 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": + def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "PrivilegeAssignmentDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'PrivilegeAssignmentDict': + return _transform_to_json_value(self) # type:ignore + class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/volumes/_models/volume.py b/python/databricks/bundles/volumes/_models/volume.py index db248fbba94..79bed352640 100644 --- a/python/databricks/bundles/volumes/_models/volume.py +++ b/python/databricks/bundles/volumes/_models/volume.py @@ -1,27 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from dataclasses import dataclass, field -from typing import TYPE_CHECKING, TypedDict +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING +from enum import Enum +from dataclasses import dataclass, replace, field from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import ( - VariableOr, - VariableOrList, - VariableOrOptional, -) -from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleParam -from databricks.bundles.volumes._models.privilege_assignment import ( - PrivilegeAssignment, - PrivilegeAssignmentParam, -) +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam + from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam + if TYPE_CHECKING: from typing_extensions import Self - @dataclass(kw_only=True) class Volume(Resource): """""" @@ -69,11 +66,12 @@ class Volume(Resource): """ @classmethod - def from_dict(cls, value: "VolumeDict") -> "Self": + def from_dict(cls, value: 'VolumeDict') -> 'Self': return _transform(cls, value) - def as_dict(self) -> "VolumeDict": - return _transform_to_json_value(self) # type:ignore + def as_dict(self) -> 'VolumeDict': + return _transform_to_json_value(self) # type:ignore + class VolumeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/volumes/_models/volume_type.py b/python/databricks/bundles/volumes/_models/volume_type.py index 0caf77b1c33..42574bde7b2 100644 --- a/python/databricks/bundles/volumes/_models/volume_type.py +++ b/python/databricks/bundles/volumes/_models/volume_type.py @@ -1,12 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from typing import Literal +from dataclasses import dataclass, replace, field + +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._transform import _transform +from databricks.bundles.core._transform_to_json import _transform_to_json_value +from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict + +from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam +from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam +from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam class VolumeType(Enum): MANAGED = "MANAGED" EXTERNAL = "EXTERNAL" - VolumeTypeParam = Literal["MANAGED", "EXTERNAL"] | VolumeType diff --git a/python/databricks_tests/core/_generated/alerts.py b/python/databricks_tests/core/_generated/alerts.py index 2d8e117c99c..a0ef68845ad 100644 --- a/python/databricks_tests/core/_generated/alerts.py +++ b/python/databricks_tests/core/_generated/alerts.py @@ -1,57 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from databricks.bundles.core import Resources, alert_mutator +from databricks.bundles.core._generated.alerts import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.alerts._models.alert import Alert from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation -from databricks.bundles.alerts._models.alert_v2_operand_column import ( - AlertV2OperandColumn, -) +from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator from databricks.bundles.alerts._models.cron_schedule import CronSchedule from databricks.bundles.alerts._models.lifecycle import Lifecycle from databricks.bundles.alerts._models.permission import Permission from databricks.bundles.alerts._models.permission_level import PermissionLevel -from databricks.bundles.core import Resources, alert_mutator -from databricks.bundles.core._generated.alerts import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_alert, - dict_example={ - "display_name": "display_name", - "evaluation": { - "comparison_operator": "LESS_THAN", - "source": {"name": "name"}, - }, - "lifecycle": {}, - "permissions": [{"level": "CAN_MANAGE"}], - "query_text": "query_text", - "run_as": {}, - "schedule": { - "quartz_cron_schedule": "quartz_cron_schedule", - "timezone_id": "timezone_id", - }, - "warehouse_id": "warehouse_id", - }, - dataclass_example=Alert( - display_name="display_name", - evaluation=AlertV2Evaluation( - comparison_operator=ComparisonOperator.LESS_THAN, - source=AlertV2OperandColumn(name="name"), - ), - lifecycle=Lifecycle(), - permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], - query_text="query_text", - run_as=AlertV2RunAs(), - schedule=CronSchedule( - quartz_cron_schedule="quartz_cron_schedule", - timezone_id="timezone_id", - ), - warehouse_id="warehouse_id", - ), + dict_example={"display_name": "display_name", "evaluation": {"comparison_operator": "LESS_THAN", "source": {"name": "name"}}, "lifecycle": {}, "permissions": [{"level": "CAN_MANAGE"}], "query_text": "query_text", "run_as": {}, "schedule": {"quartz_cron_schedule": "quartz_cron_schedule", "timezone_id": "timezone_id"}, "warehouse_id": "warehouse_id"}, + dataclass_example=Alert(display_name="display_name", evaluation=AlertV2Evaluation(comparison_operator=ComparisonOperator.LESS_THAN, source=AlertV2OperandColumn(name="name")), lifecycle=Lifecycle(), permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], query_text="query_text", run_as=AlertV2RunAs(), schedule=CronSchedule(quartz_cron_schedule="quartz_cron_schedule", timezone_id="timezone_id"), warehouse_id="warehouse_id"), mutator=alert_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/apps.py b/python/databricks_tests/core/_generated/apps.py index 41518c8c2e5..79635665ea8 100644 --- a/python/databricks_tests/core/_generated/apps.py +++ b/python/databricks_tests/core/_generated/apps.py @@ -1,5 +1,8 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from databricks.bundles.core import Resources, app_mutator +from databricks.bundles.core._generated.apps import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.apps._models.app import App from databricks.bundles.apps._models.app_config import AppConfig from databricks.bundles.apps._models.app_permission import AppPermission @@ -9,42 +12,15 @@ from databricks.bundles.apps._models.git_repository import GitRepository from databricks.bundles.apps._models.git_source import GitSource from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted -from databricks.bundles.apps._models.telemetry_export_destination import ( - TelemetryExportDestination, -) -from databricks.bundles.core import Resources, app_mutator -from databricks.bundles.core._generated.apps import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_app, - dict_example={ - "compute_size": "MEDIUM", - "config": {}, - "git_repository": {"provider": "provider", "url": "url"}, - "git_source": {}, - "lifecycle": {}, - "name": "name", - "permissions": [{"level": "CAN_MANAGE"}], - "resources": [{"name": "name"}], - "telemetry_export_destinations": [{}], - "user_api_scopes": ["user_api_scopes"], - }, - dataclass_example=App( - compute_size=ComputeSize.MEDIUM, - config=AppConfig(), - git_repository=GitRepository(provider="provider", url="url"), - git_source=GitSource(), - lifecycle=LifecycleWithStarted(), - name="name", - permissions=[AppPermission(level=AppPermissionLevel.CAN_MANAGE)], - resources=[AppResource(name="name")], - telemetry_export_destinations=[TelemetryExportDestination()], - user_api_scopes=["user_api_scopes"], - ), + dict_example={"compute_size": "MEDIUM", "config": {}, "git_repository": {"provider": "provider", "url": "url"}, "git_source": {}, "lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_MANAGE"}], "resources": [{"name": "name"}], "telemetry_export_destinations": [{}], "user_api_scopes": ["user_api_scopes"]}, + dataclass_example=App(compute_size=ComputeSize.MEDIUM, config=AppConfig(), git_repository=GitRepository(provider="provider", url="url"), git_source=GitSource(), lifecycle=LifecycleWithStarted(), name="name", permissions=[AppPermission(level=AppPermissionLevel.CAN_MANAGE)], resources=[AppResource(name="name")], telemetry_export_destinations=[TelemetryExportDestination()], user_api_scopes=["user_api_scopes"]), mutator=app_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/catalogs.py b/python/databricks_tests/core/_generated/catalogs.py index 117d964071c..60e60d6790d 100644 --- a/python/databricks_tests/core/_generated/catalogs.py +++ b/python/databricks_tests/core/_generated/catalogs.py @@ -1,34 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from databricks.bundles.core import Resources, catalog_mutator +from databricks.bundles.core._generated.catalogs import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.catalogs._models.catalog import Catalog from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings from databricks.bundles.catalogs._models.lifecycle import Lifecycle from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment -from databricks.bundles.core import Resources, catalog_mutator -from databricks.bundles.core._generated.catalogs import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_catalog, - dict_example={ - "grants": [{}], - "lifecycle": {}, - "managed_encryption_settings": {}, - "name": "name", - "options": {"key": "value"}, - "properties": {"key": "value"}, - }, - dataclass_example=Catalog( - grants=[PrivilegeAssignment()], - lifecycle=Lifecycle(), - managed_encryption_settings=EncryptionSettings(), - name="name", - options={"key": "value"}, - properties={"key": "value"}, - ), + dict_example={"grants": [{}], "lifecycle": {}, "managed_encryption_settings": {}, "name": "name", "options": {"key": "value"}, "properties": {"key": "value"}}, + dataclass_example=Catalog(grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), managed_encryption_settings=EncryptionSettings(), name="name", options={"key": "value"}, properties={"key": "value"}), mutator=catalog_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/cluster_policies.py b/python/databricks_tests/core/_generated/cluster_policies.py index 72fd868be8a..9892dbdcd3a 100644 --- a/python/databricks_tests/core/_generated/cluster_policies.py +++ b/python/databricks_tests/core/_generated/cluster_policies.py @@ -1,35 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ( - ClusterPolicyPermission, -) -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ( - ClusterPolicyPermissionLevel, -) -from databricks.bundles.cluster_policies._models.library import Library -from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle from databricks.bundles.core import Resources, cluster_policy_mutator from databricks.bundles.core._generated.cluster_policies import _resource_type from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel +from databricks.bundles.cluster_policies._models.library import Library +from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_cluster_policy, - dict_example={ - "libraries": [{}], - "lifecycle": {}, - "permissions": [{"level": "CAN_USE"}], - }, - dataclass_example=ClusterPolicy( - libraries=[Library()], - lifecycle=Lifecycle(), - permissions=[ - ClusterPolicyPermission(level=ClusterPolicyPermissionLevel.CAN_USE) - ], - ), + dict_example={"libraries": [{}], "lifecycle": {}, "permissions": [{"level": "CAN_USE"}]}, + dataclass_example=ClusterPolicy(libraries=[Library()], lifecycle=Lifecycle(), permissions=[ClusterPolicyPermission(level=ClusterPolicyPermissionLevel.CAN_USE)]), mutator=cluster_policy_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/clusters.py b/python/databricks_tests/core/_generated/clusters.py index 2de29367b86..a3247c7e6b4 100644 --- a/python/databricks_tests/core/_generated/clusters.py +++ b/python/databricks_tests/core/_generated/clusters.py @@ -1,5 +1,8 @@ # Code generated by pydabs-codegen. DO NOT EDIT. +from databricks.bundles.core import Resources, cluster_mutator +from databricks.bundles.core._generated.clusters import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.clusters._models.auto_scale import AutoScale from databricks.bundles.clusters._models.aws_attributes import AwsAttributes from databricks.bundles.clusters._models.azure_attributes import AzureAttributes @@ -7,78 +10,25 @@ from databricks.bundles.clusters._models.cluster import Cluster from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf from databricks.bundles.clusters._models.cluster_permission import ClusterPermission -from databricks.bundles.clusters._models.cluster_permission_level import ( - ClusterPermissionLevel, -) +from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode from databricks.bundles.clusters._models.docker_image import DockerImage from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes from databricks.bundles.clusters._models.init_script_info import InitScriptInfo from databricks.bundles.clusters._models.kind import Kind from databricks.bundles.clusters._models.library import Library -from databricks.bundles.clusters._models.lifecycle_with_started import ( - LifecycleWithStarted, -) -from databricks.bundles.clusters._models.node_type_flexibility import ( - NodeTypeFlexibility, -) +from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted +from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine from databricks.bundles.clusters._models.workload_type import WorkloadType -from databricks.bundles.core import Resources, cluster_mutator -from databricks.bundles.core._generated.clusters import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_cluster, - dict_example={ - "autoscale": {}, - "aws_attributes": {}, - "azure_attributes": {}, - "cluster_log_conf": {}, - "custom_tags": {"key": "value"}, - "data_security_mode": "NONE", - "docker_image": {}, - "driver_node_type_flexibility": {}, - "gcp_attributes": {}, - "init_scripts": [{}], - "kind": "CLASSIC_PREVIEW", - "libraries": [{}], - "lifecycle": {}, - "permissions": [{"level": "CAN_MANAGE"}], - "runtime_engine": "NULL", - "spark_conf": {"key": "value"}, - "spark_env_vars": {"key": "value"}, - "ssh_public_keys": ["ssh_public_keys"], - "worker_node_type_flexibility": {}, - "workload_type": {"clients": {}}, - }, - dataclass_example=Cluster( - autoscale=AutoScale(), - aws_attributes=AwsAttributes(), - azure_attributes=AzureAttributes(), - cluster_log_conf=ClusterLogConf(), - custom_tags={"key": "value"}, - data_security_mode=DataSecurityMode.NONE, - docker_image=DockerImage(), - driver_node_type_flexibility=NodeTypeFlexibility(), - gcp_attributes=GcpAttributes(), - init_scripts=[InitScriptInfo()], - kind=Kind.CLASSIC_PREVIEW, - libraries=[Library()], - lifecycle=LifecycleWithStarted(), - permissions=[ - ClusterPermission(level=ClusterPermissionLevel.CAN_MANAGE) - ], - runtime_engine=RuntimeEngine.NULL, - spark_conf={"key": "value"}, - spark_env_vars={"key": "value"}, - ssh_public_keys=["ssh_public_keys"], - worker_node_type_flexibility=NodeTypeFlexibility(), - workload_type=WorkloadType(clients=ClientsTypes()), - ), + dict_example={"autoscale": {}, "aws_attributes": {}, "azure_attributes": {}, "cluster_log_conf": {}, "custom_tags": {"key": "value"}, "data_security_mode": "NONE", "docker_image": {}, "driver_node_type_flexibility": {}, "gcp_attributes": {}, "init_scripts": [{}], "kind": "CLASSIC_PREVIEW", "libraries": [{}], "lifecycle": {}, "permissions": [{"level": "CAN_MANAGE"}], "runtime_engine": "NULL", "spark_conf": {"key": "value"}, "spark_env_vars": {"key": "value"}, "ssh_public_keys": ["ssh_public_keys"], "worker_node_type_flexibility": {}, "workload_type": {"clients": {}}}, + dataclass_example=Cluster(autoscale=AutoScale(), aws_attributes=AwsAttributes(), azure_attributes=AzureAttributes(), cluster_log_conf=ClusterLogConf(), custom_tags={"key": "value"}, data_security_mode=DataSecurityMode.NONE, docker_image=DockerImage(), driver_node_type_flexibility=NodeTypeFlexibility(), gcp_attributes=GcpAttributes(), init_scripts=[InitScriptInfo()], kind=Kind.CLASSIC_PREVIEW, libraries=[Library()], lifecycle=LifecycleWithStarted(), permissions=[ClusterPermission(level=ClusterPermissionLevel.CAN_MANAGE)], runtime_engine=RuntimeEngine.NULL, spark_conf={"key": "value"}, spark_env_vars={"key": "value"}, ssh_public_keys=["ssh_public_keys"], worker_node_type_flexibility=NodeTypeFlexibility(), workload_type=WorkloadType(clients=ClientsTypes())), mutator=cluster_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/dashboards.py b/python/databricks_tests/core/_generated/dashboards.py index 3ec459fb4ba..49a1458dc93 100644 --- a/python/databricks_tests/core/_generated/dashboards.py +++ b/python/databricks_tests/core/_generated/dashboards.py @@ -2,28 +2,20 @@ from databricks.bundles.core import Resources, dashboard_mutator from databricks.bundles.core._generated.dashboards import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.dashboards._models.dashboard import Dashboard from databricks.bundles.dashboards._models.lifecycle import Lifecycle from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState from databricks.bundles.dashboards._models.permission import Permission from databricks.bundles.dashboards._models.permission_level import PermissionLevel -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_dashboard, - dict_example={ - "lifecycle": {}, - "lifecycle_state": "ACTIVE", - "permissions": [{"level": "CAN_MANAGE"}], - }, - dataclass_example=Dashboard( - lifecycle=Lifecycle(), - lifecycle_state=LifecycleState.ACTIVE, - permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], - ), + dict_example={"lifecycle": {}, "lifecycle_state": "ACTIVE", "permissions": [{"level": "CAN_MANAGE"}]}, + dataclass_example=Dashboard(lifecycle=Lifecycle(), lifecycle_state=LifecycleState.ACTIVE, permissions=[Permission(level=PermissionLevel.CAN_MANAGE)]), mutator=dashboard_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/database_catalogs.py b/python/databricks_tests/core/_generated/database_catalogs.py index 43919e9c7d9..c8ae49535fc 100644 --- a/python/databricks_tests/core/_generated/database_catalogs.py +++ b/python/databricks_tests/core/_generated/database_catalogs.py @@ -2,29 +2,17 @@ from databricks.bundles.core import Resources, database_catalog_mutator from databricks.bundles.core._generated.database_catalogs import _resource_type -from databricks.bundles.database_catalogs._models.database_catalog import ( - DatabaseCatalog, -) -from databricks.bundles.database_catalogs._models.lifecycle import Lifecycle from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog +from databricks.bundles.database_catalogs._models.lifecycle import Lifecycle def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_database_catalog, - dict_example={ - "database_instance_name": "database_instance_name", - "database_name": "database_name", - "lifecycle": {}, - "name": "name", - }, - dataclass_example=DatabaseCatalog( - database_instance_name="database_instance_name", - database_name="database_name", - lifecycle=Lifecycle(), - name="name", - ), + dict_example={"database_instance_name": "database_instance_name", "database_name": "database_name", "lifecycle": {}, "name": "name"}, + dataclass_example=DatabaseCatalog(database_instance_name="database_instance_name", database_name="database_name", lifecycle=Lifecycle(), name="name"), mutator=database_catalog_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/database_instances.py b/python/databricks_tests/core/_generated/database_instances.py index 9fb8c2d6962..bfbdd2fc10c 100644 --- a/python/databricks_tests/core/_generated/database_instances.py +++ b/python/databricks_tests/core/_generated/database_instances.py @@ -2,36 +2,20 @@ from databricks.bundles.core import Resources, database_instance_mutator from databricks.bundles.core._generated.database_instances import _resource_type -from databricks.bundles.database_instances._models.database_instance import ( - DatabaseInstance, -) -from databricks.bundles.database_instances._models.database_instance_ref import ( - DatabaseInstanceRef, -) +from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.database_instances._models.database_instance import DatabaseInstance +from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef from databricks.bundles.database_instances._models.lifecycle import Lifecycle from databricks.bundles.database_instances._models.permission import Permission -from databricks.bundles.database_instances._models.permission_level import ( - PermissionLevel, -) -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.database_instances._models.permission_level import PermissionLevel def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_database_instance, - dict_example={ - "lifecycle": {}, - "name": "name", - "parent_instance_ref": {}, - "permissions": [{"level": "CAN_MANAGE"}], - }, - dataclass_example=DatabaseInstance( - lifecycle=Lifecycle(), - name="name", - parent_instance_ref=DatabaseInstanceRef(), - permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], - ), + dict_example={"lifecycle": {}, "name": "name", "parent_instance_ref": {}, "permissions": [{"level": "CAN_MANAGE"}]}, + dataclass_example=DatabaseInstance(lifecycle=Lifecycle(), name="name", parent_instance_ref=DatabaseInstanceRef(), permissions=[Permission(level=PermissionLevel.CAN_MANAGE)]), mutator=database_instance_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/experiments.py b/python/databricks_tests/core/_generated/experiments.py index 229dec55646..b3fa64f5ba7 100644 --- a/python/databricks_tests/core/_generated/experiments.py +++ b/python/databricks_tests/core/_generated/experiments.py @@ -2,38 +2,20 @@ from databricks.bundles.core import Resources, mlflow_experiment_mutator from databricks.bundles.core._generated.experiments import _resource_type -from databricks.bundles.experiments._models.experiment_permission_level import ( - ExperimentPermissionLevel, -) +from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel from databricks.bundles.experiments._models.experiment_tag import ExperimentTag from databricks.bundles.experiments._models.lifecycle import Lifecycle from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment -from databricks.bundles.experiments._models.mlflow_experiment_permission import ( - MlflowExperimentPermission, -) -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_mlflow_experiment, - dict_example={ - "lifecycle": {}, - "name": "name", - "permissions": [{"level": "CAN_MANAGE"}], - "tags": [{}], - }, - dataclass_example=MlflowExperiment( - lifecycle=Lifecycle(), - name="name", - permissions=[ - MlflowExperimentPermission( - level=ExperimentPermissionLevel.CAN_MANAGE - ) - ], - tags=[ExperimentTag()], - ), + dict_example={"lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_MANAGE"}], "tags": [{}]}, + dataclass_example=MlflowExperiment(lifecycle=Lifecycle(), name="name", permissions=[MlflowExperimentPermission(level=ExperimentPermissionLevel.CAN_MANAGE)], tags=[ExperimentTag()]), mutator=mlflow_experiment_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/external_locations.py b/python/databricks_tests/core/_generated/external_locations.py index d2bb07e0644..37b7c8ba88c 100644 --- a/python/databricks_tests/core/_generated/external_locations.py +++ b/python/databricks_tests/core/_generated/external_locations.py @@ -2,44 +2,20 @@ from databricks.bundles.core import Resources, external_location_mutator from databricks.bundles.core._generated.external_locations import _resource_type -from databricks.bundles.external_locations._models.encryption_details import ( - EncryptionDetails, -) -from databricks.bundles.external_locations._models.external_location import ( - ExternalLocation, -) -from databricks.bundles.external_locations._models.file_event_queue import ( - FileEventQueue, -) -from databricks.bundles.external_locations._models.lifecycle import Lifecycle -from databricks.bundles.external_locations._models.privilege_assignment import ( - PrivilegeAssignment, -) from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails +from databricks.bundles.external_locations._models.external_location import ExternalLocation +from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue +from databricks.bundles.external_locations._models.lifecycle import Lifecycle +from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_external_location, - dict_example={ - "credential_name": "credential_name", - "encryption_details": {}, - "file_event_queue": {}, - "grants": [{}], - "lifecycle": {}, - "name": "name", - "url": "url", - }, - dataclass_example=ExternalLocation( - credential_name="credential_name", - encryption_details=EncryptionDetails(), - file_event_queue=FileEventQueue(), - grants=[PrivilegeAssignment()], - lifecycle=Lifecycle(), - name="name", - url="url", - ), + dict_example={"credential_name": "credential_name", "encryption_details": {}, "file_event_queue": {}, "grants": [{}], "lifecycle": {}, "name": "name", "url": "url"}, + dataclass_example=ExternalLocation(credential_name="credential_name", encryption_details=EncryptionDetails(), file_event_queue=FileEventQueue(), grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), name="name", url="url"), mutator=external_location_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/features.py b/python/databricks_tests/core/_generated/features.py index 19344ab0e56..60e2b777025 100644 --- a/python/databricks_tests/core/_generated/features.py +++ b/python/databricks_tests/core/_generated/features.py @@ -2,35 +2,21 @@ from databricks.bundles.core import Resources, feature_mutator from databricks.bundles.core._generated.features import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.features._models.data_source import DataSource from databricks.bundles.features._models.entity_column import EntityColumn from databricks.bundles.features._models.feature import Feature from databricks.bundles.features._models.function import Function from databricks.bundles.features._models.lifecycle import Lifecycle from databricks.bundles.features._models.timeseries_column import TimeseriesColumn -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_feature, - dict_example={ - "entities": [{"name": "name"}], - "full_name": "full_name", - "function": {}, - "lifecycle": {}, - "source": {}, - "timeseries_column": {"name": "name"}, - }, - dataclass_example=Feature( - entities=[EntityColumn(name="name")], - full_name="full_name", - function=Function(), - lifecycle=Lifecycle(), - source=DataSource(), - timeseries_column=TimeseriesColumn(name="name"), - ), + dict_example={"entities": [{"name": "name"}], "full_name": "full_name", "function": {}, "lifecycle": {}, "source": {}, "timeseries_column": {"name": "name"}}, + dataclass_example=Feature(entities=[EntityColumn(name="name")], full_name="full_name", function=Function(), lifecycle=Lifecycle(), source=DataSource(), timeseries_column=TimeseriesColumn(name="name")), mutator=feature_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/genie_spaces.py b/python/databricks_tests/core/_generated/genie_spaces.py index c219a22cda4..c2c35def27a 100644 --- a/python/databricks_tests/core/_generated/genie_spaces.py +++ b/python/databricks_tests/core/_generated/genie_spaces.py @@ -2,11 +2,11 @@ from databricks.bundles.core import Resources, genie_space_mutator from databricks.bundles.core._generated.genie_spaces import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.genie_spaces._models.genie_space import GenieSpace from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle from databricks.bundles.genie_spaces._models.permission import Permission from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): @@ -14,10 +14,7 @@ def _test_case(): ResourceTestCase( add_resource=Resources.add_genie_space, dict_example={"lifecycle": {}, "permissions": [{"level": "CAN_MANAGE"}]}, - dataclass_example=GenieSpace( - lifecycle=Lifecycle(), - permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], - ), + dataclass_example=GenieSpace(lifecycle=Lifecycle(), permissions=[Permission(level=PermissionLevel.CAN_MANAGE)]), mutator=genie_space_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/instance_pools.py b/python/databricks_tests/core/_generated/instance_pools.py index 585cdfaec8e..dd866e81fdb 100644 --- a/python/databricks_tests/core/_generated/instance_pools.py +++ b/python/databricks_tests/core/_generated/instance_pools.py @@ -2,65 +2,25 @@ from databricks.bundles.core import Resources, instance_pool_mutator from databricks.bundles.core._generated.instance_pools import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.instance_pools._models.disk_spec import DiskSpec from databricks.bundles.instance_pools._models.docker_image import DockerImage from databricks.bundles.instance_pools._models.instance_pool import InstancePool -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import ( - InstancePoolAwsAttributes, -) -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import ( - InstancePoolAzureAttributes, -) -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import ( - InstancePoolGcpAttributes, -) -from databricks.bundles.instance_pools._models.instance_pool_permission import ( - InstancePoolPermission, -) -from databricks.bundles.instance_pools._models.instance_pool_permission_level import ( - InstancePoolPermissionLevel, -) +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes +from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission +from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel from databricks.bundles.instance_pools._models.lifecycle import Lifecycle -from databricks.bundles.instance_pools._models.node_type_flexibility import ( - NodeTypeFlexibility, -) -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_instance_pool, - dict_example={ - "aws_attributes": {}, - "azure_attributes": {}, - "custom_tags": {"key": "value"}, - "disk_spec": {}, - "gcp_attributes": {}, - "instance_pool_name": "instance_pool_name", - "lifecycle": {}, - "node_type_flexibility": {}, - "node_type_id": "node_type_id", - "permissions": [{"level": "CAN_MANAGE"}], - "preloaded_docker_images": [{}], - "preloaded_spark_versions": ["preloaded_spark_versions"], - }, - dataclass_example=InstancePool( - aws_attributes=InstancePoolAwsAttributes(), - azure_attributes=InstancePoolAzureAttributes(), - custom_tags={"key": "value"}, - disk_spec=DiskSpec(), - gcp_attributes=InstancePoolGcpAttributes(), - instance_pool_name="instance_pool_name", - lifecycle=Lifecycle(), - node_type_flexibility=NodeTypeFlexibility(), - node_type_id="node_type_id", - permissions=[ - InstancePoolPermission(level=InstancePoolPermissionLevel.CAN_MANAGE) - ], - preloaded_docker_images=[DockerImage()], - preloaded_spark_versions=["preloaded_spark_versions"], - ), + dict_example={"aws_attributes": {}, "azure_attributes": {}, "custom_tags": {"key": "value"}, "disk_spec": {}, "gcp_attributes": {}, "instance_pool_name": "instance_pool_name", "lifecycle": {}, "node_type_flexibility": {}, "node_type_id": "node_type_id", "permissions": [{"level": "CAN_MANAGE"}], "preloaded_docker_images": [{}], "preloaded_spark_versions": ["preloaded_spark_versions"]}, + dataclass_example=InstancePool(aws_attributes=InstancePoolAwsAttributes(), azure_attributes=InstancePoolAzureAttributes(), custom_tags={"key": "value"}, disk_spec=DiskSpec(), gcp_attributes=InstancePoolGcpAttributes(), instance_pool_name="instance_pool_name", lifecycle=Lifecycle(), node_type_flexibility=NodeTypeFlexibility(), node_type_id="node_type_id", permissions=[InstancePoolPermission(level=InstancePoolPermissionLevel.CAN_MANAGE)], preloaded_docker_images=[DockerImage()], preloaded_spark_versions=["preloaded_spark_versions"]), mutator=instance_pool_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/job_runs.py b/python/databricks_tests/core/_generated/job_runs.py index 09b1c10f6eb..d0aeea28131 100644 --- a/python/databricks_tests/core/_generated/job_runs.py +++ b/python/databricks_tests/core/_generated/job_runs.py @@ -2,36 +2,20 @@ from databricks.bundles.core import Resources, job_run_mutator from databricks.bundles.core._generated.job_runs import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.job_runs._models.job_run import JobRun from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle from databricks.bundles.job_runs._models.performance_target import PerformanceTarget from databricks.bundles.job_runs._models.pipeline_params import PipelineParams from databricks.bundles.job_runs._models.queue_settings import QueueSettings -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_job_run, - dict_example={ - "job_id": 0, - "job_parameters": {"key": "value"}, - "lifecycle": {}, - "only": ["only"], - "performance_target": "PERFORMANCE_OPTIMIZED", - "pipeline_params": {}, - "queue": {"enabled": True}, - }, - dataclass_example=JobRun( - job_id=0, - job_parameters={"key": "value"}, - lifecycle=JobRunLifecycle(), - only=["only"], - performance_target=PerformanceTarget.PERFORMANCE_OPTIMIZED, - pipeline_params=PipelineParams(), - queue=QueueSettings(enabled=True), - ), + dict_example={"job_id": 0, "job_parameters": {"key": "value"}, "lifecycle": {}, "only": ["only"], "performance_target": "PERFORMANCE_OPTIMIZED", "pipeline_params": {}, "queue": {"enabled": True}}, + dataclass_example=JobRun(job_id=0, job_parameters={"key": "value"}, lifecycle=JobRunLifecycle(), only=["only"], performance_target=PerformanceTarget.PERFORMANCE_OPTIMIZED, pipeline_params=PipelineParams(), queue=QueueSettings(enabled=True)), mutator=job_run_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/jobs.py b/python/databricks_tests/core/_generated/jobs.py index 3ed344b3811..4c41941111a 100644 --- a/python/databricks_tests/core/_generated/jobs.py +++ b/python/databricks_tests/core/_generated/jobs.py @@ -2,22 +2,17 @@ from databricks.bundles.core import Resources, job_mutator from databricks.bundles.core._generated.jobs import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.jobs._models.continuous import Continuous from databricks.bundles.jobs._models.cron_schedule import CronSchedule from databricks.bundles.jobs._models.git_provider import GitProvider from databricks.bundles.jobs._models.git_source import GitSource from databricks.bundles.jobs._models.job import Job from databricks.bundles.jobs._models.job_cluster import JobCluster -from databricks.bundles.jobs._models.job_email_notifications import ( - JobEmailNotifications, -) +from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications from databricks.bundles.jobs._models.job_environment import JobEnvironment -from databricks.bundles.jobs._models.job_notification_settings import ( - JobNotificationSettings, -) -from databricks.bundles.jobs._models.job_parameter_definition import ( - JobParameterDefinition, -) +from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings +from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition from databricks.bundles.jobs._models.job_permission import JobPermission from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel from databricks.bundles.jobs._models.job_run_as import JobRunAs @@ -28,61 +23,14 @@ from databricks.bundles.jobs._models.task import Task from databricks.bundles.jobs._models.trigger_settings import TriggerSettings from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_job, - dict_example={ - "continuous": {}, - "email_notifications": {}, - "environments": [{"environment_key": "environment_key"}], - "git_source": {"git_provider": "gitHub", "git_url": "git_url"}, - "health": {}, - "job_clusters": [{"job_cluster_key": "job_cluster_key"}], - "lifecycle": {}, - "notification_settings": {}, - "parameters": [{"default": "default", "name": "name"}], - "performance_target": "PERFORMANCE_OPTIMIZED", - "permissions": [{"level": "CAN_MANAGE"}], - "queue": {"enabled": True}, - "run_as": {}, - "schedule": { - "quartz_cron_expression": "quartz_cron_expression", - "timezone_id": "timezone_id", - }, - "tags": {"key": "value"}, - "tasks": [{"task_key": "task_key"}], - "trigger": {}, - "webhook_notifications": {}, - }, - dataclass_example=Job( - continuous=Continuous(), - email_notifications=JobEmailNotifications(), - environments=[JobEnvironment(environment_key="environment_key")], - git_source=GitSource( - git_provider=GitProvider.GIT_HUB, git_url="git_url" - ), - health=JobsHealthRules(), - job_clusters=[JobCluster(job_cluster_key="job_cluster_key")], - lifecycle=Lifecycle(), - notification_settings=JobNotificationSettings(), - parameters=[JobParameterDefinition(default="default", name="name")], - performance_target=PerformanceTarget.PERFORMANCE_OPTIMIZED, - permissions=[JobPermission(level=JobPermissionLevel.CAN_MANAGE)], - queue=QueueSettings(enabled=True), - run_as=JobRunAs(), - schedule=CronSchedule( - quartz_cron_expression="quartz_cron_expression", - timezone_id="timezone_id", - ), - tags={"key": "value"}, - tasks=[Task(task_key="task_key")], - trigger=TriggerSettings(), - webhook_notifications=WebhookNotifications(), - ), + dict_example={"continuous": {}, "email_notifications": {}, "environments": [{"environment_key": "environment_key"}], "git_source": {"git_provider": "gitHub", "git_url": "git_url"}, "health": {}, "job_clusters": [{"job_cluster_key": "job_cluster_key"}], "lifecycle": {}, "notification_settings": {}, "parameters": [{"default": "default", "name": "name"}], "performance_target": "PERFORMANCE_OPTIMIZED", "permissions": [{"level": "CAN_MANAGE"}], "queue": {"enabled": True}, "run_as": {}, "schedule": {"quartz_cron_expression": "quartz_cron_expression", "timezone_id": "timezone_id"}, "tags": {"key": "value"}, "tasks": [{"task_key": "task_key"}], "trigger": {}, "webhook_notifications": {}}, + dataclass_example=Job(continuous=Continuous(), email_notifications=JobEmailNotifications(), environments=[JobEnvironment(environment_key="environment_key")], git_source=GitSource(git_provider=GitProvider.GIT_HUB, git_url="git_url"), health=JobsHealthRules(), job_clusters=[JobCluster(job_cluster_key="job_cluster_key")], lifecycle=Lifecycle(), notification_settings=JobNotificationSettings(), parameters=[JobParameterDefinition(default="default", name="name")], performance_target=PerformanceTarget.PERFORMANCE_OPTIMIZED, permissions=[JobPermission(level=JobPermissionLevel.CAN_MANAGE)], queue=QueueSettings(enabled=True), run_as=JobRunAs(), schedule=CronSchedule(quartz_cron_expression="quartz_cron_expression", timezone_id="timezone_id"), tags={"key": "value"}, tasks=[Task(task_key="task_key")], trigger=TriggerSettings(), webhook_notifications=WebhookNotifications()), mutator=job_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/mcp_services.py b/python/databricks_tests/core/_generated/mcp_services.py index ca1247a300d..9e5ab45845e 100644 --- a/python/databricks_tests/core/_generated/mcp_services.py +++ b/python/databricks_tests/core/_generated/mcp_services.py @@ -2,33 +2,19 @@ from databricks.bundles.core import Resources, mcp_service_mutator from databricks.bundles.core._generated.mcp_services import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.mcp_services._models.lifecycle import Lifecycle from databricks.bundles.mcp_services._models.mcp_service import McpService from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig -from databricks.bundles.mcp_services._models.privilege_assignment import ( - PrivilegeAssignment, -) -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_mcp_service, - dict_example={ - "config": {}, - "grants": [{}], - "lifecycle": {}, - "mcp_service_id": "mcp_service_id", - "parent": "parent", - }, - dataclass_example=McpService( - config=McpServiceConfig(), - grants=[PrivilegeAssignment()], - lifecycle=Lifecycle(), - mcp_service_id="mcp_service_id", - parent="parent", - ), + dict_example={"config": {}, "grants": [{}], "lifecycle": {}, "mcp_service_id": "mcp_service_id", "parent": "parent"}, + dataclass_example=McpService(config=McpServiceConfig(), grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), mcp_service_id="mcp_service_id", parent="parent"), mutator=mcp_service_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/model_provider_services.py b/python/databricks_tests/core/_generated/model_provider_services.py index 994624aeee8..bc933b54a92 100644 --- a/python/databricks_tests/core/_generated/model_provider_services.py +++ b/python/databricks_tests/core/_generated/model_provider_services.py @@ -2,37 +2,19 @@ from databricks.bundles.core import Resources, model_provider_service_mutator from databricks.bundles.core._generated.model_provider_services import _resource_type -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle -from databricks.bundles.model_provider_services._models.model_provider_service import ( - ModelProviderService, -) -from databricks.bundles.model_provider_services._models.model_provider_service_config import ( - ModelProviderServiceConfig, -) -from databricks.bundles.model_provider_services._models.privilege_assignment import ( - PrivilegeAssignment, -) from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle +from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService +from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig +from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_model_provider_service, - dict_example={ - "config": {}, - "grants": [{}], - "lifecycle": {}, - "model_provider_service_id": "model_provider_service_id", - "parent": "parent", - }, - dataclass_example=ModelProviderService( - config=ModelProviderServiceConfig(), - grants=[PrivilegeAssignment()], - lifecycle=Lifecycle(), - model_provider_service_id="model_provider_service_id", - parent="parent", - ), + dict_example={"config": {}, "grants": [{}], "lifecycle": {}, "model_provider_service_id": "model_provider_service_id", "parent": "parent"}, + dataclass_example=ModelProviderService(config=ModelProviderServiceConfig(), grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), model_provider_service_id="model_provider_service_id", parent="parent"), mutator=model_provider_service_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/model_services.py b/python/databricks_tests/core/_generated/model_services.py index 5d301ae0203..f188ac17721 100644 --- a/python/databricks_tests/core/_generated/model_services.py +++ b/python/databricks_tests/core/_generated/model_services.py @@ -2,35 +2,19 @@ from databricks.bundles.core import Resources, model_service_mutator from databricks.bundles.core._generated.model_services import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.model_services._models.lifecycle import Lifecycle from databricks.bundles.model_services._models.model_service import ModelService -from databricks.bundles.model_services._models.model_service_config import ( - ModelServiceConfig, -) -from databricks.bundles.model_services._models.privilege_assignment import ( - PrivilegeAssignment, -) -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig +from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_model_service, - dict_example={ - "config": {}, - "grants": [{}], - "lifecycle": {}, - "model_service_id": "model_service_id", - "parent": "parent", - }, - dataclass_example=ModelService( - config=ModelServiceConfig(), - grants=[PrivilegeAssignment()], - lifecycle=Lifecycle(), - model_service_id="model_service_id", - parent="parent", - ), + dict_example={"config": {}, "grants": [{}], "lifecycle": {}, "model_service_id": "model_service_id", "parent": "parent"}, + dataclass_example=ModelService(config=ModelServiceConfig(), grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), model_service_id="model_service_id", parent="parent"), mutator=model_service_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/model_serving_endpoints.py b/python/databricks_tests/core/_generated/model_serving_endpoints.py index f658d74e899..af99b59b3cc 100644 --- a/python/databricks_tests/core/_generated/model_serving_endpoints.py +++ b/python/databricks_tests/core/_generated/model_serving_endpoints.py @@ -2,60 +2,24 @@ from databricks.bundles.core import Resources, model_serving_endpoint_mutator from databricks.bundles.core._generated.model_serving_endpoints import _resource_type -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import ( - AiGatewayConfig, -) -from databricks.bundles.model_serving_endpoints._models.email_notifications import ( - EmailNotifications, -) -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import ( - EndpointCoreConfigInput, -) +from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig +from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( - ModelServingEndpoint, -) -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ( - ModelServingEndpointPermission, -) -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ( - ServingEndpointPermissionLevel, -) -from databricks.bundles.model_serving_endpoints._models.telemetry_config import ( - TelemetryConfig, -) -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel +from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_model_serving_endpoint, - dict_example={ - "ai_gateway": {}, - "config": {}, - "email_notifications": {}, - "lifecycle": {}, - "name": "name", - "permissions": [{"level": "CAN_MANAGE"}], - "tags": [{"key": "key"}], - "telemetry_config": {}, - }, - dataclass_example=ModelServingEndpoint( - ai_gateway=AiGatewayConfig(), - config=EndpointCoreConfigInput(), - email_notifications=EmailNotifications(), - lifecycle=Lifecycle(), - name="name", - permissions=[ - ModelServingEndpointPermission( - level=ServingEndpointPermissionLevel.CAN_MANAGE - ) - ], - tags=[EndpointTag(key="key")], - telemetry_config=TelemetryConfig(), - ), + dict_example={"ai_gateway": {}, "config": {}, "email_notifications": {}, "lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_MANAGE"}], "tags": [{"key": "key"}], "telemetry_config": {}}, + dataclass_example=ModelServingEndpoint(ai_gateway=AiGatewayConfig(), config=EndpointCoreConfigInput(), email_notifications=EmailNotifications(), lifecycle=Lifecycle(), name="name", permissions=[ModelServingEndpointPermission(level=ServingEndpointPermissionLevel.CAN_MANAGE)], tags=[EndpointTag(key="key")], telemetry_config=TelemetryConfig()), mutator=model_serving_endpoint_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/models.py b/python/databricks_tests/core/_generated/models.py index e1c3c16deea..6facdc94b7d 100644 --- a/python/databricks_tests/core/_generated/models.py +++ b/python/databricks_tests/core/_generated/models.py @@ -2,38 +2,20 @@ from databricks.bundles.core import Resources, mlflow_model_mutator from databricks.bundles.core._generated.models import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.models._models.lifecycle import Lifecycle from databricks.bundles.models._models.mlflow_model import MlflowModel -from databricks.bundles.models._models.mlflow_model_permission import ( - MlflowModelPermission, -) +from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission from databricks.bundles.models._models.model_tag import ModelTag -from databricks.bundles.models._models.registered_model_permission_level import ( - RegisteredModelPermissionLevel, -) -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_mlflow_model, - dict_example={ - "lifecycle": {}, - "name": "name", - "permissions": [{"level": "CAN_MANAGE"}], - "tags": [{}], - }, - dataclass_example=MlflowModel( - lifecycle=Lifecycle(), - name="name", - permissions=[ - MlflowModelPermission( - level=RegisteredModelPermissionLevel.CAN_MANAGE - ) - ], - tags=[ModelTag()], - ), + dict_example={"lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_MANAGE"}], "tags": [{}]}, + dataclass_example=MlflowModel(lifecycle=Lifecycle(), name="name", permissions=[MlflowModelPermission(level=RegisteredModelPermissionLevel.CAN_MANAGE)], tags=[ModelTag()]), mutator=mlflow_model_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/pipelines.py b/python/databricks_tests/core/_generated/pipelines.py index 46ecc044b1f..99ac31bf846 100644 --- a/python/databricks_tests/core/_generated/pipelines.py +++ b/python/databricks_tests/core/_generated/pipelines.py @@ -2,61 +2,27 @@ from databricks.bundles.core import Resources, pipeline_mutator from databricks.bundles.core._generated.pipelines import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec from databricks.bundles.pipelines._models.filters import Filters -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import ( - IngestionPipelineDefinition, -) +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition from databricks.bundles.pipelines._models.lifecycle import Lifecycle from databricks.bundles.pipelines._models.notifications import Notifications from databricks.bundles.pipelines._models.pipeline import Pipeline from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission -from databricks.bundles.pipelines._models.pipeline_permission_level import ( - PipelinePermissionLevel, -) -from databricks.bundles.pipelines._models.pipelines_environment import ( - PipelinesEnvironment, -) +from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel +from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment from databricks.bundles.pipelines._models.run_as import RunAs -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_pipeline, - dict_example={ - "clusters": [{}], - "configuration": {"key": "value"}, - "environment": {}, - "event_log": {}, - "filters": {}, - "ingestion_definition": {}, - "libraries": [{}], - "lifecycle": {}, - "notifications": [{}], - "permissions": [{"level": "CAN_MANAGE"}], - "run_as": {}, - "tags": {"key": "value"}, - }, - dataclass_example=Pipeline( - clusters=[PipelineCluster()], - configuration={"key": "value"}, - environment=PipelinesEnvironment(), - event_log=EventLogSpec(), - filters=Filters(), - ingestion_definition=IngestionPipelineDefinition(), - libraries=[PipelineLibrary()], - lifecycle=Lifecycle(), - notifications=[Notifications()], - permissions=[ - PipelinePermission(level=PipelinePermissionLevel.CAN_MANAGE) - ], - run_as=RunAs(), - tags={"key": "value"}, - ), + dict_example={"clusters": [{}], "configuration": {"key": "value"}, "environment": {}, "event_log": {}, "filters": {}, "ingestion_definition": {}, "libraries": [{}], "lifecycle": {}, "notifications": [{}], "permissions": [{"level": "CAN_MANAGE"}], "run_as": {}, "tags": {"key": "value"}}, + dataclass_example=Pipeline(clusters=[PipelineCluster()], configuration={"key": "value"}, environment=PipelinesEnvironment(), event_log=EventLogSpec(), filters=Filters(), ingestion_definition=IngestionPipelineDefinition(), libraries=[PipelineLibrary()], lifecycle=Lifecycle(), notifications=[Notifications()], permissions=[PipelinePermission(level=PipelinePermissionLevel.CAN_MANAGE)], run_as=RunAs(), tags={"key": "value"}), mutator=pipeline_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/quality_monitors.py b/python/databricks_tests/core/_generated/quality_monitors.py index 4e244d7cdd5..f3b0040d937 100644 --- a/python/databricks_tests/core/_generated/quality_monitors.py +++ b/python/databricks_tests/core/_generated/quality_monitors.py @@ -2,100 +2,25 @@ from databricks.bundles.core import Resources, quality_monitor_mutator from databricks.bundles.core._generated.quality_monitors import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import ( - MonitorCronSchedule, -) -from databricks.bundles.quality_monitors._models.monitor_inference_log import ( - MonitorInferenceLog, -) -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import ( - MonitorInferenceLogProblemType, -) +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule +from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric -from databricks.bundles.quality_monitors._models.monitor_metric_type import ( - MonitorMetricType, -) -from databricks.bundles.quality_monitors._models.monitor_notifications import ( - MonitorNotifications, -) +from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType +from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot -from databricks.bundles.quality_monitors._models.monitor_time_series import ( - MonitorTimeSeries, -) +from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_quality_monitor, - dict_example={ - "assets_dir": "assets_dir", - "custom_metrics": [ - { - "definition": "definition", - "input_columns": ["input_columns"], - "name": "name", - "output_data_type": "output_data_type", - "type": "CUSTOM_METRIC_TYPE_AGGREGATE", - } - ], - "inference_log": { - "granularities": ["granularities"], - "model_id_col": "model_id_col", - "prediction_col": "prediction_col", - "problem_type": "PROBLEM_TYPE_CLASSIFICATION", - "timestamp_col": "timestamp_col", - }, - "lifecycle": {}, - "notifications": {}, - "output_schema_name": "output_schema_name", - "schedule": { - "quartz_cron_expression": "quartz_cron_expression", - "timezone_id": "timezone_id", - }, - "slicing_exprs": ["slicing_exprs"], - "snapshot": {}, - "table_name": "table_name", - "time_series": { - "granularities": ["granularities"], - "timestamp_col": "timestamp_col", - }, - }, - dataclass_example=QualityMonitor( - assets_dir="assets_dir", - custom_metrics=[ - MonitorMetric( - definition="definition", - input_columns=["input_columns"], - name="name", - output_data_type="output_data_type", - type=MonitorMetricType.CUSTOM_METRIC_TYPE_AGGREGATE, - ) - ], - inference_log=MonitorInferenceLog( - granularities=["granularities"], - model_id_col="model_id_col", - prediction_col="prediction_col", - problem_type=MonitorInferenceLogProblemType.PROBLEM_TYPE_CLASSIFICATION, - timestamp_col="timestamp_col", - ), - lifecycle=Lifecycle(), - notifications=MonitorNotifications(), - output_schema_name="output_schema_name", - schedule=MonitorCronSchedule( - quartz_cron_expression="quartz_cron_expression", - timezone_id="timezone_id", - ), - slicing_exprs=["slicing_exprs"], - snapshot=MonitorSnapshot(), - table_name="table_name", - time_series=MonitorTimeSeries( - granularities=["granularities"], timestamp_col="timestamp_col" - ), - ), + dict_example={"assets_dir": "assets_dir", "custom_metrics": [{"definition": "definition", "input_columns": ["input_columns"], "name": "name", "output_data_type": "output_data_type", "type": "CUSTOM_METRIC_TYPE_AGGREGATE"}], "inference_log": {"granularities": ["granularities"], "model_id_col": "model_id_col", "prediction_col": "prediction_col", "problem_type": "PROBLEM_TYPE_CLASSIFICATION", "timestamp_col": "timestamp_col"}, "lifecycle": {}, "notifications": {}, "output_schema_name": "output_schema_name", "schedule": {"quartz_cron_expression": "quartz_cron_expression", "timezone_id": "timezone_id"}, "slicing_exprs": ["slicing_exprs"], "snapshot": {}, "table_name": "table_name", "time_series": {"granularities": ["granularities"], "timestamp_col": "timestamp_col"}}, + dataclass_example=QualityMonitor(assets_dir="assets_dir", custom_metrics=[MonitorMetric(definition="definition", input_columns=["input_columns"], name="name", output_data_type="output_data_type", type=MonitorMetricType.CUSTOM_METRIC_TYPE_AGGREGATE)], inference_log=MonitorInferenceLog(granularities=["granularities"], model_id_col="model_id_col", prediction_col="prediction_col", problem_type=MonitorInferenceLogProblemType.PROBLEM_TYPE_CLASSIFICATION, timestamp_col="timestamp_col"), lifecycle=Lifecycle(), notifications=MonitorNotifications(), output_schema_name="output_schema_name", schedule=MonitorCronSchedule(quartz_cron_expression="quartz_cron_expression", timezone_id="timezone_id"), slicing_exprs=["slicing_exprs"], snapshot=MonitorSnapshot(), table_name="table_name", time_series=MonitorTimeSeries(granularities=["granularities"], timestamp_col="timestamp_col")), mutator=quality_monitor_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/registered_models.py b/python/databricks_tests/core/_generated/registered_models.py index 81d5f2d7ed4..b3689337a5f 100644 --- a/python/databricks_tests/core/_generated/registered_models.py +++ b/python/databricks_tests/core/_generated/registered_models.py @@ -2,17 +2,11 @@ from databricks.bundles.core import Resources, registered_model_mutator from databricks.bundles.core._generated.registered_models import _resource_type -from databricks.bundles.registered_models._models.lifecycle import Lifecycle -from databricks.bundles.registered_models._models.privilege_assignment import ( - PrivilegeAssignment, -) -from databricks.bundles.registered_models._models.registered_model import ( - RegisteredModel, -) -from databricks.bundles.registered_models._models.registered_model_alias import ( - RegisteredModelAlias, -) from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.registered_models._models.lifecycle import Lifecycle +from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment +from databricks.bundles.registered_models._models.registered_model import RegisteredModel +from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias def _test_case(): @@ -20,11 +14,7 @@ def _test_case(): ResourceTestCase( add_resource=Resources.add_registered_model, dict_example={"aliases": [{}], "grants": [{}], "lifecycle": {}}, - dataclass_example=RegisteredModel( - aliases=[RegisteredModelAlias()], - grants=[PrivilegeAssignment()], - lifecycle=Lifecycle(), - ), + dataclass_example=RegisteredModel(aliases=[RegisteredModelAlias()], grants=[PrivilegeAssignment()], lifecycle=Lifecycle()), mutator=registered_model_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/schemas.py b/python/databricks_tests/core/_generated/schemas.py index 576b0c7d3cd..a68dcedc8ed 100644 --- a/python/databricks_tests/core/_generated/schemas.py +++ b/python/databricks_tests/core/_generated/schemas.py @@ -2,30 +2,18 @@ from databricks.bundles.core import Resources, schema_mutator from databricks.bundles.core._generated.schemas import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.schemas._models.lifecycle import Lifecycle from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment from databricks.bundles.schemas._models.schema import Schema -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_schema, - dict_example={ - "catalog_name": "catalog_name", - "grants": [{}], - "lifecycle": {}, - "name": "name", - "properties": {"key": "value"}, - }, - dataclass_example=Schema( - catalog_name="catalog_name", - grants=[PrivilegeAssignment()], - lifecycle=Lifecycle(), - name="name", - properties={"key": "value"}, - ), + dict_example={"catalog_name": "catalog_name", "grants": [{}], "lifecycle": {}, "name": "name", "properties": {"key": "value"}}, + dataclass_example=Schema(catalog_name="catalog_name", grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), name="name", properties={"key": "value"}), mutator=schema_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/secret_scopes.py b/python/databricks_tests/core/_generated/secret_scopes.py index 8edc6ce6e1b..601c0194360 100644 --- a/python/databricks_tests/core/_generated/secret_scopes.py +++ b/python/databricks_tests/core/_generated/secret_scopes.py @@ -2,46 +2,21 @@ from databricks.bundles.core import Resources, secret_scope_mutator from databricks.bundles.core._generated.secret_scopes import _resource_type -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import ( - AzureKeyVaultSecretScopeMetadata, -) +from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType from databricks.bundles.secret_scopes._models.secret_scope import SecretScope -from databricks.bundles.secret_scopes._models.secret_scope_permission import ( - SecretScopePermission, -) -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import ( - SecretScopePermissionLevel, -) -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_secret_scope, - dict_example={ - "backend_type": "DATABRICKS", - "keyvault_metadata": { - "dns_name": "dns_name", - "resource_id": "resource_id", - }, - "lifecycle": {}, - "name": "name", - "permissions": [{"level": "READ"}], - }, - dataclass_example=SecretScope( - backend_type=ScopeBackendType.DATABRICKS, - keyvault_metadata=AzureKeyVaultSecretScopeMetadata( - dns_name="dns_name", resource_id="resource_id" - ), - lifecycle=Lifecycle(), - name="name", - permissions=[ - SecretScopePermission(level=SecretScopePermissionLevel.READ) - ], - ), + dict_example={"backend_type": "DATABRICKS", "keyvault_metadata": {"dns_name": "dns_name", "resource_id": "resource_id"}, "lifecycle": {}, "name": "name", "permissions": [{"level": "READ"}]}, + dataclass_example=SecretScope(backend_type=ScopeBackendType.DATABRICKS, keyvault_metadata=AzureKeyVaultSecretScopeMetadata(dns_name="dns_name", resource_id="resource_id"), lifecycle=Lifecycle(), name="name", permissions=[SecretScopePermission(level=SecretScopePermissionLevel.READ)]), mutator=secret_scope_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/secrets.py b/python/databricks_tests/core/_generated/secrets.py index 65e782459d3..65cf7791b63 100644 --- a/python/databricks_tests/core/_generated/secrets.py +++ b/python/databricks_tests/core/_generated/secrets.py @@ -2,32 +2,18 @@ from databricks.bundles.core import Resources, secret_mutator from databricks.bundles.core._generated.secrets import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.secrets._models.lifecycle import Lifecycle from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment from databricks.bundles.secrets._models.secret import Secret -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_secret, - dict_example={ - "catalog_name": "catalog_name", - "grants": [{}], - "lifecycle": {}, - "name": "name", - "schema_name": "schema_name", - "value": "value", - }, - dataclass_example=Secret( - catalog_name="catalog_name", - grants=[PrivilegeAssignment()], - lifecycle=Lifecycle(), - name="name", - schema_name="schema_name", - value="value", - ), + dict_example={"catalog_name": "catalog_name", "grants": [{}], "lifecycle": {}, "name": "name", "schema_name": "schema_name", "value": "value"}, + dataclass_example=Secret(catalog_name="catalog_name", grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), name="name", schema_name="schema_name", value="value"), mutator=secret_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/sql_warehouses.py b/python/databricks_tests/core/_generated/sql_warehouses.py index e6810ecd57c..2f3754f9063 100644 --- a/python/databricks_tests/core/_generated/sql_warehouses.py +++ b/python/databricks_tests/core/_generated/sql_warehouses.py @@ -2,49 +2,23 @@ from databricks.bundles.core import Resources, sql_warehouse_mutator from databricks.bundles.core._generated.sql_warehouses import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.sql_warehouses._models.channel import Channel -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import ( - CreateWarehouseRequestWarehouseType, -) +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import ( - LifecycleWithStarted, -) -from databricks.bundles.sql_warehouses._models.spot_instance_policy import ( - SpotInstancePolicy, -) +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted +from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import ( - SqlWarehousePermission, -) -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import ( - WarehousePermissionLevel, -) -from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_sql_warehouse, - dict_example={ - "channel": {}, - "lifecycle": {}, - "permissions": [{"level": "CAN_MANAGE"}], - "spot_instance_policy": "POLICY_UNSPECIFIED", - "tags": {}, - "warehouse_type": "TYPE_UNSPECIFIED", - }, - dataclass_example=SqlWarehouse( - channel=Channel(), - lifecycle=LifecycleWithStarted(), - permissions=[ - SqlWarehousePermission(level=WarehousePermissionLevel.CAN_MANAGE) - ], - spot_instance_policy=SpotInstancePolicy.POLICY_UNSPECIFIED, - tags=EndpointTags(), - warehouse_type=CreateWarehouseRequestWarehouseType.TYPE_UNSPECIFIED, - ), + dict_example={"channel": {}, "lifecycle": {}, "permissions": [{"level": "CAN_MANAGE"}], "spot_instance_policy": "POLICY_UNSPECIFIED", "tags": {}, "warehouse_type": "TYPE_UNSPECIFIED"}, + dataclass_example=SqlWarehouse(channel=Channel(), lifecycle=LifecycleWithStarted(), permissions=[SqlWarehousePermission(level=WarehousePermissionLevel.CAN_MANAGE)], spot_instance_policy=SpotInstancePolicy.POLICY_UNSPECIFIED, tags=EndpointTags(), warehouse_type=CreateWarehouseRequestWarehouseType.TYPE_UNSPECIFIED), mutator=sql_warehouse_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/synced_database_tables.py b/python/databricks_tests/core/_generated/synced_database_tables.py index dea7172089a..2e20187b252 100644 --- a/python/databricks_tests/core/_generated/synced_database_tables.py +++ b/python/databricks_tests/core/_generated/synced_database_tables.py @@ -2,14 +2,10 @@ from databricks.bundles.core import Resources, synced_database_table_mutator from databricks.bundles.core._generated.synced_database_tables import _resource_type -from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle -from databricks.bundles.synced_database_tables._models.synced_database_table import ( - SyncedDatabaseTable, -) -from databricks.bundles.synced_database_tables._models.synced_table_spec import ( - SyncedTableSpec, -) from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle +from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable +from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec def _test_case(): @@ -17,9 +13,7 @@ def _test_case(): ResourceTestCase( add_resource=Resources.add_synced_database_table, dict_example={"lifecycle": {}, "name": "name", "spec": {}}, - dataclass_example=SyncedDatabaseTable( - lifecycle=Lifecycle(), name="name", spec=SyncedTableSpec() - ), + dataclass_example=SyncedDatabaseTable(lifecycle=Lifecycle(), name="name", spec=SyncedTableSpec()), mutator=synced_database_table_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/vector_search_endpoints.py b/python/databricks_tests/core/_generated/vector_search_endpoints.py index 30bdb70e91e..fe2a4f5f844 100644 --- a/python/databricks_tests/core/_generated/vector_search_endpoints.py +++ b/python/databricks_tests/core/_generated/vector_search_endpoints.py @@ -2,42 +2,20 @@ from databricks.bundles.core import Resources, vector_search_endpoint_mutator from databricks.bundles.core._generated.vector_search_endpoints import _resource_type -from databricks.bundles.vector_search_endpoints._models.endpoint_type import ( - EndpointType, -) -from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( - VectorSearchEndpoint, -) -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import ( - VectorSearchEndpointPermission, -) -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import ( - VectorSearchEndpointPermissionLevel, -) from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType +from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_vector_search_endpoint, - dict_example={ - "endpoint_type": "STORAGE_OPTIMIZED", - "lifecycle": {}, - "name": "name", - "permissions": [{"level": "CAN_CREATE"}], - }, - dataclass_example=VectorSearchEndpoint( - endpoint_type=EndpointType.STORAGE_OPTIMIZED, - lifecycle=Lifecycle(), - name="name", - permissions=[ - VectorSearchEndpointPermission( - level=VectorSearchEndpointPermissionLevel.CAN_CREATE - ) - ], - ), + dict_example={"endpoint_type": "STORAGE_OPTIMIZED", "lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_CREATE"}]}, + dataclass_example=VectorSearchEndpoint(endpoint_type=EndpointType.STORAGE_OPTIMIZED, lifecycle=Lifecycle(), name="name", permissions=[VectorSearchEndpointPermission(level=VectorSearchEndpointPermissionLevel.CAN_CREATE)]), mutator=vector_search_endpoint_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/vector_search_indexes.py b/python/databricks_tests/core/_generated/vector_search_indexes.py index d72ed337f55..ba7205f6fe1 100644 --- a/python/databricks_tests/core/_generated/vector_search_indexes.py +++ b/python/databricks_tests/core/_generated/vector_search_indexes.py @@ -2,49 +2,21 @@ from databricks.bundles.core import Resources, vector_search_index_mutator from databricks.bundles.core._generated.vector_search_indexes import _resource_type -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import ( - DeltaSyncVectorIndexSpecRequest, -) -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import ( - DirectAccessVectorIndexSpec, -) -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle -from databricks.bundles.vector_search_indexes._models.privilege_assignment import ( - PrivilegeAssignment, -) -from databricks.bundles.vector_search_indexes._models.vector_index_type import ( - VectorIndexType, -) -from databricks.bundles.vector_search_indexes._models.vector_search_index import ( - VectorSearchIndex, -) from databricks_tests.core._resource_test_case import ResourceTestCase +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec +from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle +from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment +from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType +from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_vector_search_index, - dict_example={ - "delta_sync_index_spec": {}, - "direct_access_index_spec": {}, - "endpoint_name": "endpoint_name", - "grants": [{}], - "index_type": "DELTA_SYNC", - "lifecycle": {}, - "name": "name", - "primary_key": "primary_key", - }, - dataclass_example=VectorSearchIndex( - delta_sync_index_spec=DeltaSyncVectorIndexSpecRequest(), - direct_access_index_spec=DirectAccessVectorIndexSpec(), - endpoint_name="endpoint_name", - grants=[PrivilegeAssignment()], - index_type=VectorIndexType.DELTA_SYNC, - lifecycle=Lifecycle(), - name="name", - primary_key="primary_key", - ), + dict_example={"delta_sync_index_spec": {}, "direct_access_index_spec": {}, "endpoint_name": "endpoint_name", "grants": [{}], "index_type": "DELTA_SYNC", "lifecycle": {}, "name": "name", "primary_key": "primary_key"}, + dataclass_example=VectorSearchIndex(delta_sync_index_spec=DeltaSyncVectorIndexSpecRequest(), direct_access_index_spec=DirectAccessVectorIndexSpec(), endpoint_name="endpoint_name", grants=[PrivilegeAssignment()], index_type=VectorIndexType.DELTA_SYNC, lifecycle=Lifecycle(), name="name", primary_key="primary_key"), mutator=vector_search_index_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/volumes.py b/python/databricks_tests/core/_generated/volumes.py index 58d1b9a4459..eb4fbb4b732 100644 --- a/python/databricks_tests/core/_generated/volumes.py +++ b/python/databricks_tests/core/_generated/volumes.py @@ -2,33 +2,19 @@ from databricks.bundles.core import Resources, volume_mutator from databricks.bundles.core._generated.volumes import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.volumes._models.lifecycle import Lifecycle from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment from databricks.bundles.volumes._models.volume import Volume from databricks.bundles.volumes._models.volume_type import VolumeType -from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_volume, - dict_example={ - "catalog_name": "catalog_name", - "grants": [{}], - "lifecycle": {}, - "name": "name", - "schema_name": "schema_name", - "volume_type": "MANAGED", - }, - dataclass_example=Volume( - catalog_name="catalog_name", - grants=[PrivilegeAssignment()], - lifecycle=Lifecycle(), - name="name", - schema_name="schema_name", - volume_type=VolumeType.MANAGED, - ), + dict_example={"catalog_name": "catalog_name", "grants": [{}], "lifecycle": {}, "name": "name", "schema_name": "schema_name", "volume_type": "MANAGED"}, + dataclass_example=Volume(catalog_name="catalog_name", grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), name="name", schema_name="schema_name", volume_type=VolumeType.MANAGED), mutator=volume_mutator, ), _resource_type(), From b47d8abbf4e64f0c571c8260b790db191cd67fec Mon Sep 17 00:00:00 2001 From: Jan Rose Date: Thu, 22 Oct 2026 06:01:17 +0000 Subject: [PATCH 2/4] Regenerate pydabs and restore codegen uv.lock What the bumper would have committed with #6987 and #7156: in its environment pydabs-codegen stopped at `uvx ruff ... format` (no uvx in the image), leaving raw, unformatted codegen output with circular imports and proxy URLs in python/codegen/uv.lock. - Rerun generate-check: only the pydabs models and generated tests change (ruff fix + format), down to the real v0.190.0 additions. - Restore python/codegen/uv.lock, which --frozen (#6987) leaves alone. - test-update: no golden changes. Co-authored-by: Isaac --- python/codegen/uv.lock | 36 +- python/databricks/bundles/alerts/__init__.py | 195 ++- .../bundles/alerts/_models/aggregation.py | 36 +- .../bundles/alerts/_models/alert.py | 67 +- .../alerts/_models/alert_evaluation_state.py | 36 +- .../_models/alert_statement_parameter.py | 43 +- .../alerts/_models/alert_v2_evaluation.py | 62 +- .../alerts/_models/alert_v2_notification.py | 44 +- .../alerts/_models/alert_v2_operand.py | 50 +- .../alerts/_models/alert_v2_operand_column.py | 41 +- .../alerts/_models/alert_v2_operand_value.py | 40 +- .../bundles/alerts/_models/alert_v2_run_as.py | 40 +- .../alerts/_models/alert_v2_subscription.py | 40 +- .../alerts/_models/comparison_operator.py | 45 +- .../bundles/alerts/_models/cron_schedule.py | 44 +- .../alerts/_models/custom_template_format.py | 34 +- .../bundles/alerts/_models/lifecycle.py | 40 +- .../bundles/alerts/_models/permission.py | 44 +- .../alerts/_models/permission_level.py | 58 +- .../alerts/_models/schedule_pause_status.py | 31 +- python/databricks/bundles/apps/__init__.py | 359 +++-- python/databricks/bundles/apps/_models/app.py | 94 +- .../bundles/apps/_models/app_config.py | 63 +- .../bundles/apps/_models/app_env_var.py | 62 +- .../bundles/apps/_models/app_permission.py | 66 +- .../apps/_models/app_permission_level.py | 54 +- .../bundles/apps/_models/app_resource.py | 104 +- .../bundles/apps/_models/app_resource_app.py | 66 +- .../app_resource_app_app_permission.py | 53 +- .../apps/_models/app_resource_database.py | 66 +- ...p_resource_database_database_permission.py | 57 +- .../apps/_models/app_resource_experiment.py | 66 +- ...source_experiment_experiment_permission.py | 58 +- .../apps/_models/app_resource_genie_space.py | 66 +- ...urce_genie_space_genie_space_permission.py | 58 +- .../bundles/apps/_models/app_resource_job.py | 66 +- .../app_resource_job_job_permission.py | 58 +- .../apps/_models/app_resource_postgres.py | 66 +- ...p_resource_postgres_postgres_permission.py | 57 +- .../apps/_models/app_resource_secret.py | 66 +- .../app_resource_secret_secret_permission.py | 58 +- .../_models/app_resource_serving_endpoint.py | 70 +- ...ng_endpoint_serving_endpoint_permission.py | 58 +- .../_models/app_resource_sql_warehouse.py | 66 +- ..._sql_warehouse_sql_warehouse_permission.py | 58 +- .../apps/_models/app_resource_uc_securable.py | 72 +- ...ce_uc_securable_uc_securable_permission.py | 60 +- ...resource_uc_securable_uc_securable_type.py | 58 +- .../bundles/apps/_models/compute_size.py | 53 +- .../bundles/apps/_models/git_repository.py | 63 +- .../bundles/apps/_models/git_source.py | 63 +- .../apps/_models/lifecycle_with_started.py | 62 +- .../_models/telemetry_export_destination.py | 71 +- .../bundles/apps/_models/unity_catalog.py | 63 +- .../databricks/bundles/catalogs/__init__.py | 75 +- .../_models/azure_encryption_settings.py | 23 +- .../bundles/catalogs/_models/catalog.py | 38 +- .../catalogs/_models/encryption_settings.py | 28 +- .../bundles/catalogs/_models/lifecycle.py | 23 +- .../bundles/catalogs/_models/privilege.py | 72 +- .../catalogs/_models/privilege_assignment.py | 24 +- .../bundles/cluster_policies/__init__.py | 94 +- .../_models/cluster_policy.py | 35 +- .../_models/cluster_policy_permission.py | 29 +- .../cluster_policy_permission_level.py | 17 +- .../cluster_policies/_models/library.py | 39 +- .../cluster_policies/_models/lifecycle.py | 25 +- .../cluster_policies/_models/maven_library.py | 29 +- .../_models/python_py_pi_library.py | 25 +- .../_models/r_cran_library.py | 25 +- .../databricks/bundles/clusters/__init__.py | 407 +++-- .../bundles/clusters/_models/adlsgen2_info.py | 63 +- .../bundles/clusters/_models/auto_scale.py | 62 +- .../clusters/_models/aws_attributes.py | 73 +- .../clusters/_models/aws_availability.py | 60 +- .../clusters/_models/azure_attributes.py | 73 +- .../clusters/_models/azure_availability.py | 59 +- .../bundles/clusters/_models/clients_types.py | 62 +- .../bundles/clusters/_models/cluster.py | 126 +- .../clusters/_models/cluster_log_conf.py | 77 +- .../clusters/_models/cluster_permission.py | 66 +- .../_models/cluster_permission_level.py | 58 +- .../_models/confidential_compute_type.py | 60 +- .../clusters/_models/data_security_mode.py | 80 +- .../clusters/_models/dbfs_storage_info.py | 63 +- .../clusters/_models/dependency_mode.py | 65 +- .../clusters/_models/docker_basic_auth.py | 62 +- .../bundles/clusters/_models/docker_image.py | 66 +- .../clusters/_models/ebs_volume_type.py | 58 +- .../clusters/_models/gcp_attributes.py | 73 +- .../clusters/_models/gcp_availability.py | 59 +- .../clusters/_models/gcs_storage_info.py | 63 +- .../clusters/_models/init_script_info.py | 93 +- .../bundles/clusters/_models/kind.py | 60 +- .../bundles/clusters/_models/library.py | 76 +- .../_models/lifecycle_with_started.py | 62 +- .../clusters/_models/local_file_info.py | 62 +- .../clusters/_models/log_analytics_info.py | 62 +- .../bundles/clusters/_models/maven_library.py | 66 +- .../clusters/_models/node_type_flexibility.py | 63 +- .../clusters/_models/python_py_pi_library.py | 62 +- .../clusters/_models/r_cran_library.py | 62 +- .../clusters/_models/runtime_engine.py | 53 +- .../clusters/_models/s3_storage_info.py | 63 +- .../clusters/_models/volumes_storage_info.py | 63 +- .../bundles/clusters/_models/workload_type.py | 67 +- .../_models/workspace_storage_info.py | 63 +- python/databricks/bundles/core/__init__.py | 36 +- .../bundles/core/_generated/__init__.py | 125 +- .../bundles/core/_generated/alerts.py | 4 +- .../bundles/core/_generated/apps.py | 4 +- .../bundles/core/_generated/catalogs.py | 4 +- .../core/_generated/cluster_policies.py | 11 +- .../bundles/core/_generated/clusters.py | 4 +- .../bundles/core/_generated/dashboards.py | 5 +- .../core/_generated/database_catalogs.py | 17 +- .../core/_generated/database_instances.py | 21 +- .../bundles/core/_generated/experiments.py | 21 +- .../core/_generated/external_locations.py | 21 +- .../bundles/core/_generated/genie_spaces.py | 5 +- .../bundles/core/_generated/instance_pools.py | 5 +- .../bundles/core/_generated/mcp_services.py | 5 +- .../_generated/model_provider_services.py | 29 +- .../bundles/core/_generated/model_services.py | 5 +- .../_generated/model_serving_endpoints.py | 29 +- .../bundles/core/_generated/models.py | 5 +- .../core/_generated/quality_monitors.py | 17 +- .../core/_generated/registered_models.py | 17 +- .../bundles/core/_generated/secret_scopes.py | 5 +- .../bundles/core/_generated/sql_warehouses.py | 5 +- .../core/_generated/synced_database_tables.py | 25 +- .../_generated/vector_search_endpoints.py | 29 +- .../core/_generated/vector_search_indexes.py | 25 +- .../databricks/bundles/dashboards/__init__.py | 56 +- .../bundles/dashboards/_models/dashboard.py | 28 +- .../bundles/dashboards/_models/lifecycle.py | 23 +- .../dashboards/_models/lifecycle_state.py | 14 +- .../bundles/dashboards/_models/permission.py | 27 +- .../dashboards/_models/permission_level.py | 41 +- .../bundles/database_catalogs/__init__.py | 24 +- .../_models/database_catalog.py | 21 +- .../database_catalogs/_models/lifecycle.py | 18 +- .../bundles/database_instances/__init__.py | 70 +- .../database_instances/_models/custom_tag.py | 23 +- .../_models/database_instance.py | 43 +- .../_models/database_instance_ref.py | 24 +- .../database_instances/_models/lifecycle.py | 23 +- .../database_instances/_models/permission.py | 27 +- .../_models/permission_level.py | 41 +- .../bundles/experiments/__init__.py | 82 +- .../_models/experiment_permission_level.py | 20 +- .../experiments/_models/experiment_tag.py | 25 +- .../_models/experiment_trace_location.py | 31 +- .../bundles/experiments/_models/lifecycle.py | 24 +- .../experiments/_models/mlflow_experiment.py | 47 +- .../_models/mlflow_experiment_permission.py | 32 +- .../experiments/_models/uc_trace_location.py | 27 +- .../bundles/external_locations/__init__.py | 128 +- .../_models/aws_sqs_queue.py | 29 +- .../_models/azure_queue_storage.py | 29 +- .../_models/encryption_details.py | 34 +- .../_models/external_location.py | 48 +- .../_models/file_event_queue.py | 41 +- .../external_locations/_models/gcp_pubsub.py | 29 +- .../external_locations/_models/lifecycle.py | 29 +- .../external_locations/_models/privilege.py | 78 +- .../_models/privilege_assignment.py | 33 +- .../_models/sse_encryption_details.py | 34 +- .../sse_encryption_details_algorithm.py | 25 +- .../databricks/bundles/features/__init__.py | 538 ++++--- .../features/_models/aggregation_function.py | 132 +- .../_models/approx_count_distinct_function.py | 67 +- .../_models/approx_percentile_function.py | 63 +- .../bundles/features/_models/avg_function.py | 63 +- .../features/_models/column_selection.py | 63 +- .../features/_models/count_function.py | 63 +- .../bundles/features/_models/custom_udf.py | 69 +- .../bundles/features/_models/data_source.py | 88 +- .../features/_models/delta_table_source.py | 62 +- .../bundles/features/_models/entity_column.py | 62 +- .../bundles/features/_models/feature.py | 89 +- .../features/_models/feature_reference.py | 65 +- .../features/_models/feature_view_source.py | 69 +- .../features/_models/field_definition.py | 67 +- .../_models/first_distinct_function.py | 65 +- .../features/_models/first_function.py | 63 +- .../features/_models/first_n_function.py | 65 +- .../bundles/features/_models/flat_schema.py | 67 +- .../bundles/features/_models/function.py | 74 +- .../bundles/features/_models/input_binding.py | 65 +- .../bundles/features/_models/job_context.py | 63 +- .../bundles/features/_models/kafka_source.py | 62 +- .../_models/last_distinct_function.py | 65 +- .../bundles/features/_models/last_function.py | 63 +- .../features/_models/last_n_function.py | 65 +- .../bundles/features/_models/lifecycle.py | 62 +- .../features/_models/lineage_context.py | 69 +- .../bundles/features/_models/max_function.py | 63 +- .../bundles/features/_models/min_function.py | 63 +- .../features/_models/request_source.py | 64 +- .../features/_models/rolling_window.py | 63 +- .../features/_models/sawtooth_window.py | 65 +- .../features/_models/scalar_data_type.py | 71 +- .../bundles/features/_models/sdp_source.py | 65 +- .../features/_models/sliding_window.py | 62 +- .../features/_models/source_lateness.py | 65 +- .../features/_models/stddev_pop_function.py | 63 +- .../features/_models/stddev_samp_function.py | 63 +- .../bundles/features/_models/stream_source.py | 63 +- .../bundles/features/_models/sum_function.py | 63 +- .../bundles/features/_models/time_window.py | 78 +- .../features/_models/timeseries_column.py | 62 +- .../features/_models/tumbling_window.py | 62 +- .../features/_models/var_pop_function.py | 63 +- .../features/_models/var_samp_function.py | 63 +- .../bundles/genie_spaces/__init__.py | 46 +- .../genie_spaces/_models/genie_space.py | 25 +- .../bundles/genie_spaces/_models/lifecycle.py | 21 +- .../genie_spaces/_models/permission.py | 25 +- .../genie_spaces/_models/permission_level.py | 39 +- .../bundles/instance_pools/__init__.py | 192 ++- .../instance_pools/_models/disk_spec.py | 41 +- .../instance_pools/_models/disk_type.py | 48 +- .../disk_type_azure_disk_volume_type.py | 35 +- .../_models/disk_type_ebs_volume_type.py | 35 +- .../_models/docker_basic_auth.py | 39 +- .../instance_pools/_models/docker_image.py | 43 +- .../_models/gcp_availability.py | 36 +- .../instance_pools/_models/instance_pool.py | 72 +- .../_models/instance_pool_aws_attributes.py | 48 +- ...stance_pool_aws_attributes_availability.py | 35 +- .../_models/instance_pool_azure_attributes.py | 48 +- ...ance_pool_azure_attributes_availability.py | 35 +- .../_models/instance_pool_gcp_attributes.py | 48 +- .../_models/instance_pool_permission.py | 43 +- .../_models/instance_pool_permission_level.py | 35 +- .../instance_pools/_models/lifecycle.py | 39 +- .../_models/node_type_flexibility.py | 40 +- .../databricks/bundles/job_runs/__init__.py | 64 +- .../bundles/job_runs/_models/job_run.py | 43 +- .../job_runs/_models/job_run_lifecycle.py | 27 +- .../job_runs/_models/job_run_trigger.py | 23 +- .../job_runs/_models/performance_target.py | 19 +- .../job_runs/_models/pipeline_params.py | 23 +- .../job_runs/_models/queue_settings.py | 23 +- python/databricks/bundles/jobs/__init__.py | 1418 +++++++++++------ .../bundles/jobs/_models/adlsgen2_info.py | 177 +- .../bundles/jobs/_models/ai_runtime_task.py | 191 +-- .../_models/ai_runtime_task_priority_class.py | 174 +- .../bundles/jobs/_models/alert_task.py | 184 +-- .../jobs/_models/alert_task_subscriber.py | 177 +- .../jobs/_models/authentication_method.py | 167 +- .../bundles/jobs/_models/auto_scale.py | 176 +- .../bundles/jobs/_models/aws_attributes.py | 185 +-- .../bundles/jobs/_models/aws_availability.py | 174 +- .../bundles/jobs/_models/azure_attributes.py | 185 +-- .../jobs/_models/azure_availability.py | 173 +- .../jobs/_models/clean_rooms_notebook_task.py | 181 +-- .../bundles/jobs/_models/clients_types.py | 176 +- .../bundles/jobs/_models/cluster_log_conf.py | 189 +-- .../bundles/jobs/_models/cluster_spec.py | 225 +-- .../bundles/jobs/_models/compute.py | 180 +-- .../bundles/jobs/_models/compute_config.py | 177 +- .../bundles/jobs/_models/compute_spec.py | 181 +-- .../_models/compute_spec_accelerator_type.py | 173 +- .../bundles/jobs/_models/condition.py | 167 +- .../bundles/jobs/_models/condition_task.py | 180 +-- .../bundles/jobs/_models/condition_task_op.py | 182 +-- .../jobs/_models/confidential_compute_type.py | 174 +- .../bundles/jobs/_models/continuous.py | 183 +-- .../continuous_trigger_configuration.py | 189 +-- .../bundles/jobs/_models/cron_schedule.py | 179 +-- .../_models/cron_trigger_configuration.py | 177 +- .../bundles/jobs/_models/dashboard_task.py | 181 +-- .../jobs/_models/data_security_mode.py | 194 +-- .../bundles/jobs/_models/day_of_week.py | 175 +- .../bundles/jobs/_models/dbfs_storage_info.py | 177 +- .../bundles/jobs/_models/dbt_platform_task.py | 177 +- .../bundles/jobs/_models/dbt_task.py | 175 +- .../bundles/jobs/_models/dependency_mode.py | 179 +-- .../bundles/jobs/_models/deployment_spec.py | 181 +-- .../bundles/jobs/_models/docker_basic_auth.py | 176 +- .../bundles/jobs/_models/docker_image.py | 180 +-- .../bundles/jobs/_models/ebs_volume_type.py | 172 +- .../bundles/jobs/_models/environment.py | 177 +- .../file_arrival_trigger_configuration.py | 180 +-- .../bundles/jobs/_models/for_each_task.py | 178 +-- .../bundles/jobs/_models/gcp_attributes.py | 185 +-- .../bundles/jobs/_models/gcp_availability.py | 173 +- .../bundles/jobs/_models/gcs_storage_info.py | 177 +- .../jobs/_models/gen_ai_compute_task.py | 182 +-- .../bundles/jobs/_models/git_provider.py | 181 +-- .../bundles/jobs/_models/git_source.py | 184 +-- .../jobs/_models/hardware_accelerator_type.py | 173 +- .../bundles/jobs/_models/init_script_info.py | 205 +-- python/databricks/bundles/jobs/_models/job.py | 244 +-- .../bundles/jobs/_models/job_cluster.py | 177 +- .../jobs/_models/job_email_notifications.py | 180 +-- .../bundles/jobs/_models/job_environment.py | 177 +- .../jobs/_models/job_environment_variables.py | 181 +-- .../_models/job_environment_variables_spec.py | 181 +-- .../jobs/_models/job_notification_settings.py | 176 +- .../jobs/_models/job_parameter_definition.py | 176 +- .../bundles/jobs/_models/job_permission.py | 180 +-- .../jobs/_models/job_permission_level.py | 172 +- .../bundles/jobs/_models/job_run_as.py | 179 +-- .../jobs/_models/jobs_health_metric.py | 181 +-- .../jobs/_models/jobs_health_operator.py | 168 +- .../bundles/jobs/_models/jobs_health_rule.py | 184 +-- .../bundles/jobs/_models/jobs_health_rules.py | 181 +-- .../databricks/bundles/jobs/_models/kind.py | 174 +- .../bundles/jobs/_models/library.py | 188 +-- .../bundles/jobs/_models/lifecycle.py | 176 +- .../bundles/jobs/_models/local_file_info.py | 176 +- .../jobs/_models/log_analytics_info.py | 176 +- .../jobs/_models/maintenance_window.py | 176 +- .../bundles/jobs/_models/maven_library.py | 180 +-- .../_models/model_trigger_configuration.py | 189 +-- .../model_trigger_configuration_condition.py | 173 +- .../jobs/_models/node_type_flexibility.py | 177 +- .../bundles/jobs/_models/notebook_task.py | 179 +-- .../bundles/jobs/_models/pause_status.py | 167 +- .../jobs/_models/performance_target.py | 172 +- .../_models/periodic_trigger_configuration.py | 184 +-- ...eriodic_trigger_configuration_time_unit.py | 171 +- .../bundles/jobs/_models/pipeline_params.py | 176 +- .../bundles/jobs/_models/pipeline_task.py | 181 +-- .../bundles/jobs/_models/power_bi_model.py | 179 +-- .../bundles/jobs/_models/power_bi_table.py | 175 +- .../bundles/jobs/_models/power_bi_task.py | 184 +-- .../jobs/_models/python_operator_task.py | 185 +-- .../_models/python_operator_task_parameter.py | 181 +-- .../jobs/_models/python_py_pi_library.py | 176 +- .../bundles/jobs/_models/python_wheel_task.py | 176 +- .../bundles/jobs/_models/queue_settings.py | 176 +- .../bundles/jobs/_models/r_cran_library.py | 176 +- .../databricks/bundles/jobs/_models/run_if.py | 182 +-- .../bundles/jobs/_models/run_job_task.py | 184 +-- .../bundles/jobs/_models/runtime_engine.py | 167 +- .../bundles/jobs/_models/s3_storage_info.py | 177 +- .../databricks/bundles/jobs/_models/source.py | 168 +- .../bundles/jobs/_models/spark_jar_task.py | 180 +-- .../bundles/jobs/_models/spark_python_task.py | 179 +-- .../bundles/jobs/_models/spark_submit_task.py | 179 +-- .../bundles/jobs/_models/sparse_checkout.py | 176 +- .../_models/sql_condition_configuration.py | 185 +-- .../_models/sql_condition_trigger_mode.py | 176 +- .../bundles/jobs/_models/sql_task.py | 193 +-- .../bundles/jobs/_models/sql_task_alert.py | 184 +-- .../jobs/_models/sql_task_dashboard.py | 184 +-- .../bundles/jobs/_models/sql_task_file.py | 175 +- .../bundles/jobs/_models/sql_task_query.py | 176 +- .../jobs/_models/sql_task_subscription.py | 176 +- .../bundles/jobs/_models/storage_mode.py | 167 +- .../bundles/jobs/_models/subscription.py | 184 +-- .../jobs/_models/subscription_attachment.py | 185 +-- ...subscription_attachment_data_attachment.py | 197 +-- ..._attachment_data_attachment_file_format.py | 173 +- ...ption_attachment_data_attachment_widget.py | 184 +-- ...subscription_attachment_page_attachment.py | 193 +-- ...tachment_page_attachment_page_selection.py | 173 +- .../jobs/_models/subscription_subscriber.py | 176 +- .../table_update_trigger_configuration.py | 179 +-- .../databricks/bundles/jobs/_models/task.py | 263 ++- .../bundles/jobs/_models/task_dependency.py | 176 +- .../jobs/_models/task_email_notifications.py | 180 +-- .../_models/task_notification_settings.py | 176 +- .../bundles/jobs/_models/task_retry_mode.py | 168 +- .../jobs/_models/trigger_configuration.py | 204 +-- .../bundles/jobs/_models/trigger_settings.py | 195 +-- .../jobs/_models/volumes_storage_info.py | 177 +- .../bundles/jobs/_models/webhook.py | 176 +- .../jobs/_models/webhook_notifications.py | 181 +-- .../bundles/jobs/_models/workload_type.py | 181 +-- .../jobs/_models/workspace_storage_info.py | 177 +- .../bundles/mcp_services/__init__.py | 97 +- .../bundles/mcp_services/_models/lifecycle.py | 28 +- .../mcp_services/_models/mcp_service.py | 43 +- .../_models/mcp_service_config.py | 37 +- .../mcp_service_config_source_connection.py | 33 +- .../bundles/mcp_services/_models/privilege.py | 77 +- .../_models/privilege_assignment.py | 27 +- .../mcp_services/_models/rate_limit.py | 37 +- .../_models/rate_limit_rate_limit_key.py | 31 +- .../rate_limit_rate_limit_renewal_period.py | 25 +- .../model_provider_services/__init__.py | 388 +++-- .../_models/inference_table_config.py | 54 +- .../_models/lifecycle.py | 53 +- .../_models/model_provider_service.py | 68 +- .../_models/model_provider_service_config.py | 166 +- ...e_config_amazon_bedrock_provider_config.py | 73 +- ...g_amazon_bedrock_provider_direct_config.py | 85 +- ...ervice_config_anthropic_provider_config.py | 85 +- ...config_anthropic_provider_direct_config.py | 69 +- ...onfig_anthropic_provider_relayed_config.py | 64 +- ..._provider_service_config_aws_access_key.py | 66 +- ...ce_config_azure_open_ai_provider_config.py | 73 +- ...ig_azure_open_ai_provider_direct_config.py | 95 +- ...fig_custom_provider_api_key_header_auth.py | 69 +- ...r_service_config_custom_provider_config.py | 73 +- ...ce_config_custom_provider_direct_config.py | 77 +- ..._service_config_entra_service_principal.py | 65 +- ...ice_config_external_model_provider_type.py | 81 +- ...onfig_gemini_enterprise_provider_config.py | 77 +- ...emini_enterprise_provider_direct_config.py | 83 +- ...onfig_microsoft_foundry_provider_config.py | 77 +- ...icrosoft_foundry_provider_direct_config.py | 97 +- ...ider_service_config_model_target_config.py | 65 +- ..._service_config_open_ai_provider_config.py | 73 +- ...e_config_open_ai_provider_direct_config.py | 65 +- ..._service_config_provider_pricing_config.py | 61 +- ...provider_service_config_provider_secret.py | 67 +- ...rovider_service_config_secret_reference.py | 61 +- ...vider_service_config_service_credential.py | 61 +- .../_models/privilege.py | 102 +- .../_models/privilege_assignment.py | 57 +- .../_models/rate_limit.py | 62 +- .../_models/rate_limit_rate_limit_key.py | 56 +- .../rate_limit_rate_limit_renewal_period.py | 50 +- .../bundles/model_services/__init__.py | 196 ++- .../_models/inference_table_config.py | 38 +- .../model_services/_models/lifecycle.py | 37 +- ...ider_service_config_model_target_config.py | 49 +- .../model_services/_models/model_service.py | 52 +- .../_models/model_service_config.py | 50 +- ...model_service_config_destination_config.py | 78 +- ...fig_destination_config_destination_type.py | 50 +- ...el_service_config_external_model_config.py | 46 +- .../model_service_config_fallback_config.py | 50 +- ...del_service_config_pay_per_token_config.py | 42 +- ...ce_config_provisioned_throughput_config.py | 45 +- .../model_service_config_routing_config.py | 54 +- .../model_services/_models/privilege.py | 86 +- .../_models/privilege_assignment.py | 41 +- .../model_services/_models/rate_limit.py | 46 +- .../_models/rate_limit_rate_limit_key.py | 40 +- .../rate_limit_rate_limit_renewal_period.py | 34 +- .../model_serving_endpoints/__init__.py | 530 +++--- .../_models/ai21_labs_config.py | 73 +- .../_models/ai_gateway_config.py | 93 +- .../ai_gateway_guardrail_parameters.py | 81 +- .../ai_gateway_guardrail_pii_behavior.py | 81 +- ...gateway_guardrail_pii_behavior_behavior.py | 68 +- .../_models/ai_gateway_guardrails.py | 77 +- .../ai_gateway_inference_table_config.py | 77 +- .../_models/ai_gateway_rate_limit.py | 81 +- .../_models/ai_gateway_rate_limit_key.py | 69 +- .../ai_gateway_rate_limit_renewal_period.py | 68 +- .../ai_gateway_usage_tracking_config.py | 77 +- .../_models/amazon_bedrock_config.py | 77 +- .../amazon_bedrock_config_bedrock_provider.py | 69 +- .../_models/anthropic_config.py | 73 +- .../_models/api_key_auth.py | 73 +- .../_models/auto_capture_config_input.py | 74 +- .../_models/bearer_token_auth.py | 73 +- .../_models/cohere_config.py | 73 +- .../_models/custom_provider_config.py | 82 +- .../databricks_model_serving_config.py | 77 +- .../_models/email_notifications.py | 73 +- .../_models/endpoint_core_config_input.py | 89 +- .../_models/endpoint_tag.py | 73 +- .../_models/external_model.py | 121 +- .../_models/external_model_provider.py | 79 +- .../_models/fallback_config.py | 73 +- .../_models/google_cloud_vertex_ai_config.py | 77 +- .../_models/lifecycle.py | 73 +- .../_models/model_serving_endpoint.py | 112 +- .../model_serving_endpoint_permission.py | 81 +- .../_models/open_ai_config.py | 74 +- .../_models/pa_lm_config.py | 73 +- .../_models/rate_limit.py | 84 +- .../_models/rate_limit_key.py | 67 +- .../_models/rate_limit_renewal_period.py | 67 +- .../model_serving_endpoints/_models/route.py | 73 +- .../_models/served_entity_input.py | 81 +- .../_models/served_model_input.py | 81 +- .../served_model_input_workload_type.py | 80 +- .../serving_endpoint_permission_level.py | 69 +- .../_models/serving_model_workload_type.py | 80 +- .../_models/telemetry_config.py | 89 +- .../_models/telemetry_feature.py | 75 +- .../telemetry_inference_table_config.py | 78 +- .../_models/traffic_config.py | 77 +- .../_models/unity_catalog_table_names.py | 73 +- python/databricks/bundles/models/__init__.py | 58 +- .../bundles/models/_models/lifecycle.py | 22 +- .../bundles/models/_models/mlflow_model.py | 31 +- .../models/_models/mlflow_model_permission.py | 26 +- .../bundles/models/_models/model_tag.py | 23 +- .../registered_model_permission_level.py | 25 +- .../databricks/bundles/pipelines/__init__.py | 1164 +++++++++----- .../pipelines/_models/adlsgen2_info.py | 142 +- .../_models/api_source_connector_config.py | 144 +- .../_models/api_source_connector_options.py | 148 +- .../_models/auto_full_refresh_policy.py | 142 +- .../_models/avro_transformer_options.py | 144 +- .../pipelines/_models/aws_attributes.py | 150 +- .../pipelines/_models/aws_availability.py | 139 +- .../pipelines/_models/azure_attributes.py | 150 +- .../pipelines/_models/azure_availability.py | 138 +- .../pipelines/_models/cluster_log_conf.py | 154 +- .../_models/confidential_compute_type.py | 139 +- .../_models/confluence_connector_options.py | 146 +- .../confluent_schema_registry_options.py | 145 +- .../_models/connection_parameters.py | 142 +- .../pipelines/_models/connector_options.py | 206 +-- .../pipelines/_models/connector_type.py | 133 +- .../pipelines/_models/data_staging_options.py | 142 +- .../bundles/pipelines/_models/day_of_week.py | 140 +- .../pipelines/_models/dbfs_storage_info.py | 142 +- .../pipelines/_models/ebs_volume_type.py | 137 +- .../pipelines/_models/event_log_spec.py | 142 +- .../bundles/pipelines/_models/file_filter.py | 142 +- .../_models/file_ingestion_options.py | 163 +- .../file_ingestion_options_file_format.py | 138 +- ...ingestion_options_schema_evolution_mode.py | 144 +- .../bundles/pipelines/_models/file_library.py | 141 +- .../bundles/pipelines/_models/filters.py | 141 +- .../pipelines/_models/gcp_attributes.py | 150 +- .../pipelines/_models/gcp_availability.py | 138 +- .../pipelines/_models/gcs_storage_info.py | 142 +- .../pipelines/_models/google_ads_config.py | 142 +- .../google_ads_custom_report_options.py | 148 +- .../pipelines/_models/google_ads_options.py | 148 +- .../pipelines/_models/google_drive_options.py | 150 +- ..._drive_options_google_drive_entity_type.py | 138 +- .../pipelines/_models/ingestion_config.py | 147 +- .../ingestion_gateway_pipeline_definition.py | 150 +- .../_models/ingestion_pipeline_definition.py | 169 +- ...tion_pipeline_definition_fanout_options.py | 151 +- ...fic_config_query_based_connector_config.py | 156 +- ...ne_definition_workday_report_parameters.py | 149 +- .../pipelines/_models/init_script_info.py | 170 +- .../_models/jira_connector_options.py | 142 +- .../_models/json_transformer_options.py | 153 +- .../pipelines/_models/kafka_options.py | 149 +- .../bundles/pipelines/_models/lifecycle.py | 141 +- .../_models/linked_in_ads_options.py | 156 +- ...ons_linked_in_ads_custom_report_options.py | 191 +-- ...ptions_linked_in_ads_entity_granularity.py | 144 +- ...tom_report_options_linked_in_ads_finder.py | 140 +- ..._options_linked_in_ads_time_granularity.py | 140 +- .../pipelines/_models/local_file_info.py | 141 +- .../pipelines/_models/log_analytics_info.py | 141 +- .../pipelines/_models/marketo_options.py | 144 +- .../pipelines/_models/maven_library.py | 145 +- .../_models/meta_marketing_options.py | 154 +- ...ns_meta_marketing_custom_report_options.py | 151 +- .../pipelines/_models/notebook_library.py | 141 +- .../pipelines/_models/notifications.py | 141 +- .../_models/operation_time_window.py | 145 +- .../_models/outlook_attachment_mode.py | 139 +- .../pipelines/_models/outlook_body_format.py | 135 +- .../pipelines/_models/outlook_options.py | 152 +- .../bundles/pipelines/_models/parse_mode.py | 133 +- .../bundles/pipelines/_models/path_pattern.py | 141 +- .../bundles/pipelines/_models/pipeline.py | 186 +-- .../pipelines/_models/pipeline_cluster.py | 169 +- .../_models/pipeline_cluster_autoscale.py | 145 +- .../pipeline_cluster_autoscale_mode.py | 137 +- .../pipelines/_models/pipeline_library.py | 157 +- .../pipelines/_models/pipeline_permission.py | 145 +- .../_models/pipeline_permission_level.py | 137 +- .../_models/pipelines_environment.py | 142 +- .../_models/postgres_catalog_config.py | 146 +- .../pipelines/_models/postgres_slot_config.py | 142 +- .../_models/protobuf_transformer_options.py | 148 +- .../pipelines/_models/rabbitmq_options.py | 142 +- .../pipelines/_models/reddit_ads_options.py | 156 +- ...ptions_reddit_ads_custom_report_options.py | 151 +- .../bundles/pipelines/_models/report_spec.py | 145 +- .../pipelines/_models/restart_window.py | 145 +- .../bundles/pipelines/_models/run_as.py | 144 +- .../pipelines/_models/s3_storage_info.py | 142 +- .../_models/schema_registry_config.py | 145 +- .../bundles/pipelines/_models/schema_spec.py | 153 +- .../pipelines/_models/sharepoint_options.py | 150 +- ...arepoint_options_sharepoint_entity_type.py | 138 +- .../pipelines/_models/smartsheet_options.py | 142 +- .../_models/source_catalog_config.py | 146 +- .../pipelines/_models/source_config.py | 153 +- .../bundles/pipelines/_models/table_spec.py | 149 +- .../_models/table_specific_config.py | 177 +- .../_models/table_specific_config_scd_type.py | 137 +- .../pipelines/_models/tik_tok_ads_options.py | 162 +- ...tions_tik_tok_ads_custom_report_options.py | 159 +- .../tik_tok_ads_options_tik_tok_data_level.py | 138 +- ...tik_tok_ads_options_tik_tok_report_type.py | 138 +- .../bundles/pipelines/_models/transformer.py | 158 +- .../pipelines/_models/transformer_format.py | 136 +- .../pipelines/_models/volumes_storage_info.py | 142 +- .../_models/workspace_storage_info.py | 142 +- .../_models/zendesk_support_options.py | 142 +- .../bundles/quality_monitors/__init__.py | 150 +- .../quality_monitors/_models/lifecycle.py | 32 +- .../_models/monitor_cron_schedule.py | 36 +- .../monitor_cron_schedule_pause_status.py | 28 +- .../monitor_data_classification_config.py | 39 +- .../_models/monitor_destination.py | 32 +- .../_models/monitor_inference_log.py | 40 +- .../monitor_inference_log_problem_type.py | 28 +- .../_models/monitor_metric.py | 37 +- .../_models/monitor_metric_type.py | 33 +- .../_models/monitor_notifications.py | 36 +- .../_models/monitor_snapshot.py | 32 +- .../_models/monitor_time_series.py | 33 +- .../_models/quality_monitor.py | 71 +- .../bundles/registered_models/__init__.py | 58 +- .../registered_models/_models/lifecycle.py | 22 +- .../registered_models/_models/privilege.py | 71 +- .../_models/privilege_assignment.py | 26 +- .../_models/registered_model.py | 33 +- .../_models/registered_model_alias.py | 22 +- python/databricks/bundles/schemas/__init__.py | 47 +- .../bundles/schemas/_models/lifecycle.py | 21 +- .../bundles/schemas/_models/privilege.py | 70 +- .../schemas/_models/privilege_assignment.py | 20 +- .../bundles/schemas/_models/schema.py | 30 +- .../bundles/secret_scopes/__init__.py | 68 +- .../azure_key_vault_secret_scope_metadata.py | 29 +- .../secret_scopes/_models/lifecycle.py | 24 +- .../_models/scope_backend_type.py | 16 +- .../secret_scopes/_models/secret_scope.py | 40 +- .../_models/secret_scope_permission.py | 28 +- .../_models/secret_scope_permission_level.py | 19 +- python/databricks/bundles/secrets/__init__.py | 37 +- .../bundles/secrets/_models/lifecycle.py | 21 +- .../bundles/secrets/_models/privilege.py | 70 +- .../secrets/_models/privilege_assignment.py | 20 +- .../bundles/secrets/_models/secret.py | 30 +- .../bundles/sql_warehouses/__init__.py | 112 +- .../bundles/sql_warehouses/_models/channel.py | 35 +- .../sql_warehouses/_models/channel_name.py | 31 +- ...create_warehouse_request_warehouse_type.py | 26 +- .../_models/endpoint_tag_pair.py | 30 +- .../sql_warehouses/_models/endpoint_tags.py | 34 +- .../_models/lifecycle_with_started.py | 30 +- .../_models/spot_instance_policy.py | 31 +- .../sql_warehouses/_models/sql_warehouse.py | 51 +- .../_models/sql_warehouse_permission.py | 34 +- .../_models/warehouse_permission_level.py | 27 +- .../synced_database_tables/__init__.py | 80 +- .../_models/lifecycle.py | 25 +- .../_models/new_pipeline_spec.py | 26 +- .../_models/synced_database_table.py | 32 +- .../_models/synced_table_scheduling_policy.py | 20 +- .../_models/synced_table_spec.py | 42 +- .../synced_table_spec_pg_specific_type.py | 28 +- .../synced_table_spec_type_override.py | 36 +- .../vector_search_endpoints/__init__.py | 56 +- .../_models/endpoint_type.py | 15 +- .../_models/lifecycle.py | 23 +- .../_models/vector_search_endpoint.py | 42 +- .../vector_search_endpoint_permission.py | 31 +- ...vector_search_endpoint_permission_level.py | 19 +- .../bundles/vector_search_indexes/__init__.py | 124 +- .../delta_sync_vector_index_spec_request.py | 55 +- .../direct_access_vector_index_spec.py | 51 +- .../_models/embedding_source_column.py | 31 +- .../_models/embedding_vector_column.py | 31 +- .../_models/index_subtype.py | 23 +- .../_models/lifecycle.py | 31 +- .../_models/pipeline_type.py | 23 +- .../_models/privilege.py | 80 +- .../_models/privilege_assignment.py | 35 +- .../_models/vector_index_type.py | 23 +- .../_models/vector_search_index.py | 58 +- python/databricks/bundles/volumes/__init__.py | 52 +- .../bundles/volumes/_models/lifecycle.py | 23 +- .../bundles/volumes/_models/privilege.py | 72 +- .../volumes/_models/privilege_assignment.py | 22 +- .../bundles/volumes/_models/volume.py | 30 +- .../bundles/volumes/_models/volume_type.py | 14 +- .../core/_generated/alerts.py | 44 +- .../databricks_tests/core/_generated/apps.py | 36 +- .../core/_generated/catalogs.py | 24 +- .../core/_generated/cluster_policies.py | 28 +- .../core/_generated/clusters.py | 66 +- .../core/_generated/dashboards.py | 14 +- .../core/_generated/database_catalogs.py | 20 +- .../core/_generated/database_instances.py | 28 +- .../core/_generated/experiments.py | 28 +- .../core/_generated/external_locations.py | 38 +- .../core/_generated/features.py | 20 +- .../core/_generated/genie_spaces.py | 7 +- .../core/_generated/instance_pools.py | 58 +- .../core/_generated/job_runs.py | 22 +- .../databricks_tests/core/_generated/jobs.py | 64 +- .../core/_generated/mcp_services.py | 22 +- .../_generated/model_provider_services.py | 30 +- .../core/_generated/model_services.py | 26 +- .../_generated/model_serving_endpoints.py | 56 +- .../core/_generated/models.py | 28 +- .../core/_generated/pipelines.py | 46 +- .../core/_generated/quality_monitors.py | 93 +- .../core/_generated/registered_models.py | 20 +- .../core/_generated/schemas.py | 18 +- .../core/_generated/secret_scopes.py | 37 +- .../core/_generated/secrets.py | 20 +- .../core/_generated/sql_warehouses.py | 42 +- .../core/_generated/synced_database_tables.py | 14 +- .../_generated/vector_search_endpoints.py | 36 +- .../core/_generated/vector_search_indexes.py | 44 +- .../core/_generated/volumes.py | 20 +- 704 files changed, 14194 insertions(+), 53086 deletions(-) diff --git a/python/codegen/uv.lock b/python/codegen/uv.lock index 4835268ec96..681ead3c4f0 100644 --- a/python/codegen/uv.lock +++ b/python/codegen/uv.lock @@ -20,52 +20,52 @@ dev = [{ name = "pytest" }] [[package]] name = "colorama" version = "0.4.6" -source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } -sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/d8/53/6f443c9a4a8358a93a6792e2acffb9d9d5cb0a5cfd8802644b7b1c9a02e4/colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44", size = 27697, upload-time = "2022-10-25T02:36:22.414Z" } +source = { registry = "https://pypi.org/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/d8/53/6f443c9a4a8358a93a6792e2acffb9d9d5cb0a5cfd8802644b7b1c9a02e4/colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44", size = 27697, upload-time = "2022-10-25T02:36:22.414Z" } wheels = [ - { url = "https://pypi-proxy.cloud.databricks.com/packages/d1/d6/3965ed04c63042e047cb6a3e6ed1a63a35087b6a609aa3a15ed8ac56c221/colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6", size = 25335, upload-time = "2022-10-25T02:36:20.889Z" }, + { url = "https://files.pythonhosted.org/packages/d1/d6/3965ed04c63042e047cb6a3e6ed1a63a35087b6a609aa3a15ed8ac56c221/colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6", size = 25335, upload-time = "2022-10-25T02:36:20.889Z" }, ] [[package]] name = "iniconfig" version = "2.1.0" -source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } -sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/f2/97/ebf4da567aa6827c909642694d71c9fcf53e5b504f2d96afea02718862f3/iniconfig-2.1.0.tar.gz", hash = "sha256:3abbd2e30b36733fee78f9c7f7308f2d0050e88f0087fd25c2645f63c773e1c7", size = 4793, upload-time = "2025-03-19T20:09:59.721Z" } +source = { registry = "https://pypi.org/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/f2/97/ebf4da567aa6827c909642694d71c9fcf53e5b504f2d96afea02718862f3/iniconfig-2.1.0.tar.gz", hash = "sha256:3abbd2e30b36733fee78f9c7f7308f2d0050e88f0087fd25c2645f63c773e1c7", size = 4793, upload-time = "2025-03-19T20:09:59.721Z" } wheels = [ - { url = "https://pypi-proxy.cloud.databricks.com/packages/2c/e1/e6716421ea10d38022b952c159d5161ca1193197fb744506875fbb87ea7b/iniconfig-2.1.0-py3-none-any.whl", hash = "sha256:9deba5723312380e77435581c6bf4935c94cbfab9b1ed33ef8d238ea168eb760", size = 6050, upload-time = "2025-03-19T20:10:01.071Z" }, + { url = "https://files.pythonhosted.org/packages/2c/e1/e6716421ea10d38022b952c159d5161ca1193197fb744506875fbb87ea7b/iniconfig-2.1.0-py3-none-any.whl", hash = "sha256:9deba5723312380e77435581c6bf4935c94cbfab9b1ed33ef8d238ea168eb760", size = 6050, upload-time = "2025-03-19T20:10:01.071Z" }, ] [[package]] name = "packaging" version = "24.2" -source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } -sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/d0/63/68dbb6eb2de9cb10ee4c9c14a0148804425e13c4fb20d61cce69f53106da/packaging-24.2.tar.gz", hash = "sha256:c228a6dc5e932d346bc5739379109d49e8853dd8223571c7c5b55260edc0b97f", size = 163950, upload-time = "2024-11-08T09:47:47.202Z" } +source = { registry = "https://pypi.org/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/d0/63/68dbb6eb2de9cb10ee4c9c14a0148804425e13c4fb20d61cce69f53106da/packaging-24.2.tar.gz", hash = "sha256:c228a6dc5e932d346bc5739379109d49e8853dd8223571c7c5b55260edc0b97f", size = 163950, upload-time = "2024-11-08T09:47:47.202Z" } wheels = [ - { url = "https://pypi-proxy.cloud.databricks.com/packages/88/ef/eb23f262cca3c0c4eb7ab1933c3b1f03d021f2c48f54763065b6f0e321be/packaging-24.2-py3-none-any.whl", hash = "sha256:09abb1bccd265c01f4a3aa3f7a7db064b36514d2cba19a2f694fe6150451a759", size = 65451, upload-time = "2024-11-08T09:47:44.722Z" }, + { url = "https://files.pythonhosted.org/packages/88/ef/eb23f262cca3c0c4eb7ab1933c3b1f03d021f2c48f54763065b6f0e321be/packaging-24.2-py3-none-any.whl", hash = "sha256:09abb1bccd265c01f4a3aa3f7a7db064b36514d2cba19a2f694fe6150451a759", size = 65451, upload-time = "2024-11-08T09:47:44.722Z" }, ] [[package]] name = "pluggy" version = "1.5.0" -source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } -sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/96/2d/02d4312c973c6050a18b314a5ad0b3210edb65a906f868e31c111dede4a6/pluggy-1.5.0.tar.gz", hash = "sha256:2cffa88e94fdc978c4c574f15f9e59b7f4201d439195c3715ca9e2486f1d0cf1", size = 67955, upload-time = "2024-04-20T21:34:42.531Z" } +source = { registry = "https://pypi.org/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/96/2d/02d4312c973c6050a18b314a5ad0b3210edb65a906f868e31c111dede4a6/pluggy-1.5.0.tar.gz", hash = "sha256:2cffa88e94fdc978c4c574f15f9e59b7f4201d439195c3715ca9e2486f1d0cf1", size = 67955, upload-time = "2024-04-20T21:34:42.531Z" } wheels = [ - { url = "https://pypi-proxy.cloud.databricks.com/packages/88/5f/e351af9a41f866ac3f1fac4ca0613908d9a41741cfcf2228f4ad853b697d/pluggy-1.5.0-py3-none-any.whl", hash = "sha256:44e1ad92c8ca002de6377e165f3e0f1be63266ab4d554740532335b9d75ea669", size = 20556, upload-time = "2024-04-20T21:34:40.434Z" }, + { url = "https://files.pythonhosted.org/packages/88/5f/e351af9a41f866ac3f1fac4ca0613908d9a41741cfcf2228f4ad853b697d/pluggy-1.5.0-py3-none-any.whl", hash = "sha256:44e1ad92c8ca002de6377e165f3e0f1be63266ab4d554740532335b9d75ea669", size = 20556, upload-time = "2024-04-20T21:34:40.434Z" }, ] [[package]] name = "pygments" version = "2.20.0" -source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } -sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/c3/b2/bc9c9196916376152d655522fdcebac55e66de6603a76a02bca1b6414f6c/pygments-2.20.0.tar.gz", hash = "sha256:6757cd03768053ff99f3039c1a36d6c0aa0b263438fcab17520b30a303a82b5f", size = 4955991, upload-time = "2026-03-29T13:29:33.898Z" } +source = { registry = "https://pypi.org/simple" } +sdist = { url = "https://files.pythonhosted.org/packages/c3/b2/bc9c9196916376152d655522fdcebac55e66de6603a76a02bca1b6414f6c/pygments-2.20.0.tar.gz", hash = "sha256:6757cd03768053ff99f3039c1a36d6c0aa0b263438fcab17520b30a303a82b5f", size = 4955991, upload-time = "2026-03-29T13:29:33.898Z" } wheels = [ - { url = "https://pypi-proxy.cloud.databricks.com/packages/f4/7e/a72dd26f3b0f4f2bf1dd8923c85f7ceb43172af56d63c7383eb62b332364/pygments-2.20.0-py3-none-any.whl", hash = "sha256:81a9e26dd42fd28a23a2d169d86d7ac03b46e2f8b59ed4698fb4785f946d0176", size = 1231151, upload-time = "2026-03-29T13:29:30.038Z" }, + { url = "https://files.pythonhosted.org/packages/f4/7e/a72dd26f3b0f4f2bf1dd8923c85f7ceb43172af56d63c7383eb62b332364/pygments-2.20.0-py3-none-any.whl", hash = "sha256:81a9e26dd42fd28a23a2d169d86d7ac03b46e2f8b59ed4698fb4785f946d0176", size = 1231151, upload-time = "2026-03-29T13:29:30.038Z" }, ] [[package]] name = "pytest" version = "9.0.3" -source = { registry = "https://pypi-proxy.cloud.databricks.com/simple" } +source = { registry = "https://pypi.org/simple" } dependencies = [ { name = "colorama", marker = "sys_platform == 'win32'" }, { name = "iniconfig" }, @@ -73,7 +73,7 @@ dependencies = [ { name = "pluggy" }, { name = "pygments" }, ] -sdist = { url = "https://pypi-proxy.cloud.databricks.com/packages/7d/0d/549bd94f1a0a402dc8cf64563a117c0f3765662e2e668477624baeec44d5/pytest-9.0.3.tar.gz", hash = "sha256:b86ada508af81d19edeb213c681b1d48246c1a91d304c6c81a427674c17eb91c", size = 1572165, upload-time = "2026-04-07T17:16:18.027Z" } +sdist = { url = "https://files.pythonhosted.org/packages/7d/0d/549bd94f1a0a402dc8cf64563a117c0f3765662e2e668477624baeec44d5/pytest-9.0.3.tar.gz", hash = "sha256:b86ada508af81d19edeb213c681b1d48246c1a91d304c6c81a427674c17eb91c", size = 1572165, upload-time = "2026-04-07T17:16:18.027Z" } wheels = [ - { url = "https://pypi-proxy.cloud.databricks.com/packages/d4/24/a372aaf5c9b7208e7112038812994107bc65a84cd00e0354a88c2c77a617/pytest-9.0.3-py3-none-any.whl", hash = "sha256:2c5efc453d45394fdd706ade797c0a81091eccd1d6e4bccfcd476e2b8e0ab5d9", size = 375249, upload-time = "2026-04-07T17:16:16.13Z" }, + { url = "https://files.pythonhosted.org/packages/d4/24/a372aaf5c9b7208e7112038812994107bc65a84cd00e0354a88c2c77a617/pytest-9.0.3-py3-none-any.whl", hash = "sha256:2c5efc453d45394fdd706ade797c0a81091eccd1d6e4bccfcd476e2b8e0ab5d9", size = 375249, upload-time = "2026-04-07T17:16:16.13Z" }, ] diff --git a/python/databricks/bundles/alerts/__init__.py b/python/databricks/bundles/alerts/__init__.py index a64c273c35e..3d8987ba271 100644 --- a/python/databricks/bundles/alerts/__init__.py +++ b/python/databricks/bundles/alerts/__init__.py @@ -1,78 +1,131 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Aggregation', - 'AggregationParam', - 'Alert', - 'AlertDict', - 'AlertEvaluationState', - 'AlertEvaluationStateParam', - 'AlertParam', - 'AlertStatementParameter', - 'AlertStatementParameterDict', - 'AlertStatementParameterParam', - 'AlertV2Evaluation', - 'AlertV2EvaluationDict', - 'AlertV2EvaluationParam', - 'AlertV2Notification', - 'AlertV2NotificationDict', - 'AlertV2NotificationParam', - 'AlertV2Operand', - 'AlertV2OperandColumn', - 'AlertV2OperandColumnDict', - 'AlertV2OperandColumnParam', - 'AlertV2OperandDict', - 'AlertV2OperandParam', - 'AlertV2OperandValue', - 'AlertV2OperandValueDict', - 'AlertV2OperandValueParam', - 'AlertV2RunAs', - 'AlertV2RunAsDict', - 'AlertV2RunAsParam', - 'AlertV2Subscription', - 'AlertV2SubscriptionDict', - 'AlertV2SubscriptionParam', - 'ComparisonOperator', - 'ComparisonOperatorParam', - 'CronSchedule', - 'CronScheduleDict', - 'CronScheduleParam', - 'CustomTemplateFormat', - 'CustomTemplateFormatParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'Permission', - 'PermissionDict', - 'PermissionLevel', - 'PermissionLevelParam', - 'PermissionParam', - 'SchedulePauseStatus', - 'SchedulePauseStatusParam', + "Aggregation", + "AggregationParam", + "Alert", + "AlertDict", + "AlertEvaluationState", + "AlertEvaluationStateParam", + "AlertParam", + "AlertStatementParameter", + "AlertStatementParameterDict", + "AlertStatementParameterParam", + "AlertV2Evaluation", + "AlertV2EvaluationDict", + "AlertV2EvaluationParam", + "AlertV2Notification", + "AlertV2NotificationDict", + "AlertV2NotificationParam", + "AlertV2Operand", + "AlertV2OperandColumn", + "AlertV2OperandColumnDict", + "AlertV2OperandColumnParam", + "AlertV2OperandDict", + "AlertV2OperandParam", + "AlertV2OperandValue", + "AlertV2OperandValueDict", + "AlertV2OperandValueParam", + "AlertV2RunAs", + "AlertV2RunAsDict", + "AlertV2RunAsParam", + "AlertV2Subscription", + "AlertV2SubscriptionDict", + "AlertV2SubscriptionParam", + "ComparisonOperator", + "ComparisonOperatorParam", + "CronSchedule", + "CronScheduleDict", + "CronScheduleParam", + "CustomTemplateFormat", + "CustomTemplateFormatParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "Permission", + "PermissionDict", + "PermissionLevel", + "PermissionLevelParam", + "PermissionParam", + "SchedulePauseStatus", + "SchedulePauseStatusParam", ] -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam +from databricks.bundles.alerts._models.alert_evaluation_state import ( + AlertEvaluationState, + AlertEvaluationStateParam, +) +from databricks.bundles.alerts._models.alert_statement_parameter import ( + AlertStatementParameter, + AlertStatementParameterDict, + AlertStatementParameterParam, +) +from databricks.bundles.alerts._models.alert_v2_evaluation import ( + AlertV2Evaluation, + AlertV2EvaluationDict, + AlertV2EvaluationParam, +) +from databricks.bundles.alerts._models.alert_v2_notification import ( + AlertV2Notification, + AlertV2NotificationDict, + AlertV2NotificationParam, +) +from databricks.bundles.alerts._models.alert_v2_operand import ( + AlertV2Operand, + AlertV2OperandDict, + AlertV2OperandParam, +) +from databricks.bundles.alerts._models.alert_v2_operand_column import ( + AlertV2OperandColumn, + AlertV2OperandColumnDict, + AlertV2OperandColumnParam, +) +from databricks.bundles.alerts._models.alert_v2_operand_value import ( + AlertV2OperandValue, + AlertV2OperandValueDict, + AlertV2OperandValueParam, +) +from databricks.bundles.alerts._models.alert_v2_run_as import ( + AlertV2RunAs, + AlertV2RunAsDict, + AlertV2RunAsParam, +) +from databricks.bundles.alerts._models.alert_v2_subscription import ( + AlertV2Subscription, + AlertV2SubscriptionDict, + AlertV2SubscriptionParam, +) +from databricks.bundles.alerts._models.comparison_operator import ( + ComparisonOperator, + ComparisonOperatorParam, +) +from databricks.bundles.alerts._models.cron_schedule import ( + CronSchedule, + CronScheduleDict, + CronScheduleParam, +) +from databricks.bundles.alerts._models.custom_template_format import ( + CustomTemplateFormat, + CustomTemplateFormatParam, +) +from databricks.bundles.alerts._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.alerts._models.permission import ( + Permission, + PermissionDict, + PermissionParam, +) +from databricks.bundles.alerts._models.permission_level import ( + PermissionLevel, + PermissionLevelParam, +) +from databricks.bundles.alerts._models.schedule_pause_status import ( + SchedulePauseStatus, + SchedulePauseStatusParam, +) diff --git a/python/databricks/bundles/alerts/_models/aggregation.py b/python/databricks/bundles/alerts/_models/aggregation.py index 809aae0b3c1..ff9efddf6e9 100644 --- a/python/databricks/bundles/alerts/_models/aggregation.py +++ b/python/databricks/bundles/alerts/_models/aggregation.py @@ -1,35 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam +from typing import Literal class Aggregation(Enum): @@ -42,4 +14,8 @@ class Aggregation(Enum): MAX = "MAX" STDDEV = "STDDEV" -AggregationParam = Literal["SUM", "COUNT", "COUNT_DISTINCT", "AVG", "MEDIAN", "MIN", "MAX", "STDDEV"] | Aggregation + +AggregationParam = ( + Literal["SUM", "COUNT", "COUNT_DISTINCT", "AVG", "MEDIAN", "MIN", "MAX", "STDDEV"] + | Aggregation +) diff --git a/python/databricks/bundles/alerts/_models/alert.py b/python/databricks/bundles/alerts/_models/alert.py index 6232280649e..2a66b1d169c 100644 --- a/python/databricks/bundles/alerts/_models/alert.py +++ b/python/databricks/bundles/alerts/_models/alert.py @@ -1,41 +1,43 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.alerts._models.alert_statement_parameter import ( + AlertStatementParameter, + AlertStatementParameterParam, +) +from databricks.bundles.alerts._models.alert_v2_evaluation import ( + AlertV2Evaluation, + AlertV2EvaluationParam, +) +from databricks.bundles.alerts._models.alert_v2_run_as import ( + AlertV2RunAs, + AlertV2RunAsParam, +) +from databricks.bundles.alerts._models.cron_schedule import ( + CronSchedule, + CronScheduleParam, +) +from databricks.bundles.alerts._models.custom_template_format import ( + CustomTemplateFormat, + CustomTemplateFormatParam, +) +from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.alerts._models.permission import Permission, PermissionParam from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Alert(Resource): """""" @@ -135,12 +137,11 @@ class Alert(Resource): """ @classmethod - def from_dict(cls, value: 'AlertDict') -> 'Self': + def from_dict(cls, value: "AlertDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertDict": + return _transform_to_json_value(self) # type:ignore class AlertDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_evaluation_state.py b/python/databricks/bundles/alerts/_models/alert_evaluation_state.py index 4cf7627c088..abee35a5a4a 100644 --- a/python/databricks/bundles/alerts/_models/alert_evaluation_state.py +++ b/python/databricks/bundles/alerts/_models/alert_evaluation_state.py @@ -1,35 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam +from typing import Literal class AlertEvaluationState(Enum): @@ -40,9 +12,13 @@ class AlertEvaluationState(Enum): OK - alert is not triggered ERROR - alert evaluation failed """ + UNKNOWN = "UNKNOWN" TRIGGERED = "TRIGGERED" OK = "OK" ERROR = "ERROR" -AlertEvaluationStateParam = Literal["UNKNOWN", "TRIGGERED", "OK", "ERROR"] | AlertEvaluationState + +AlertEvaluationStateParam = ( + Literal["UNKNOWN", "TRIGGERED", "OK", "ERROR"] | AlertEvaluationState +) diff --git a/python/databricks/bundles/alerts/_models/alert_statement_parameter.py b/python/databricks/bundles/alerts/_models/alert_statement_parameter.py index b2d33f0eb0e..def206d148d 100644 --- a/python/databricks/bundles/alerts/_models/alert_statement_parameter.py +++ b/python/databricks/bundles/alerts/_models/alert_statement_parameter.py @@ -1,49 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertStatementParameter: """ :meta private: [EXPERIMENTAL] - + A named parameter bound to the alert query. Only flat, named scalar parameters are supported; complex types such as ARRAY, MAP, and STRUCT are not. """ + name: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -68,12 +44,11 @@ class AlertStatementParameter: """ @classmethod - def from_dict(cls, value: 'AlertStatementParameterDict') -> 'Self': + def from_dict(cls, value: "AlertStatementParameterDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertStatementParameterDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertStatementParameterDict": + return _transform_to_json_value(self) # type:ignore class AlertStatementParameterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_evaluation.py b/python/databricks/bundles/alerts/_models/alert_v2_evaluation.py index 88c6f05a946..4d599872ccb 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_evaluation.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_evaluation.py @@ -1,41 +1,36 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.alerts._models.alert_evaluation_state import ( + AlertEvaluationState, + AlertEvaluationStateParam, +) +from databricks.bundles.alerts._models.alert_v2_notification import ( + AlertV2Notification, + AlertV2NotificationParam, +) +from databricks.bundles.alerts._models.alert_v2_operand import ( + AlertV2Operand, + AlertV2OperandParam, +) +from databricks.bundles.alerts._models.alert_v2_operand_column import ( + AlertV2OperandColumn, + AlertV2OperandColumnParam, +) +from databricks.bundles.alerts._models.comparison_operator import ( + ComparisonOperator, + ComparisonOperatorParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertV2Evaluation: """""" @@ -66,12 +61,11 @@ class AlertV2Evaluation: """ @classmethod - def from_dict(cls, value: 'AlertV2EvaluationDict') -> 'Self': + def from_dict(cls, value: "AlertV2EvaluationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertV2EvaluationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertV2EvaluationDict": + return _transform_to_json_value(self) # type:ignore class AlertV2EvaluationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_notification.py b/python/databricks/bundles/alerts/_models/alert_v2_notification.py index 33507c80c81..81c892f1d90 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_notification.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_notification.py @@ -1,41 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.alerts._models.alert_v2_subscription import ( + AlertV2Subscription, + AlertV2SubscriptionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertV2Notification: """""" @@ -55,12 +34,11 @@ class AlertV2Notification: subscriptions: VariableOrList[AlertV2Subscription] = field(default_factory=list) @classmethod - def from_dict(cls, value: 'AlertV2NotificationDict') -> 'Self': + def from_dict(cls, value: "AlertV2NotificationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertV2NotificationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertV2NotificationDict": + return _transform_to_json_value(self) # type:ignore class AlertV2NotificationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_operand.py b/python/databricks/bundles/alerts/_models/alert_v2_operand.py index fa391b35c2d..5e39c938860 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_operand.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_operand.py @@ -1,41 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.alerts._models.alert_v2_operand_column import ( + AlertV2OperandColumn, + AlertV2OperandColumnParam, +) +from databricks.bundles.alerts._models.alert_v2_operand_value import ( + AlertV2OperandValue, + AlertV2OperandValueParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertV2Operand: """""" @@ -45,12 +28,11 @@ class AlertV2Operand: value: VariableOrOptional[AlertV2OperandValue] = None @classmethod - def from_dict(cls, value: 'AlertV2OperandDict') -> 'Self': + def from_dict(cls, value: "AlertV2OperandDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertV2OperandDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertV2OperandDict": + return _transform_to_json_value(self) # type:ignore class AlertV2OperandDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_operand_column.py b/python/databricks/bundles/alerts/_models/alert_v2_operand_column.py index 3a6f83d2614..da2f90107e8 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_operand_column.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_operand_column.py @@ -1,41 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertV2OperandColumn: """""" @@ -50,12 +26,11 @@ class AlertV2OperandColumn: display: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: 'AlertV2OperandColumnDict') -> 'Self': + def from_dict(cls, value: "AlertV2OperandColumnDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertV2OperandColumnDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertV2OperandColumnDict": + return _transform_to_json_value(self) # type:ignore class AlertV2OperandColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_operand_value.py b/python/databricks/bundles/alerts/_models/alert_v2_operand_value.py index 8fda36d762e..5abc7f386ee 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_operand_value.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_operand_value.py @@ -1,41 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertV2OperandValue: """""" @@ -47,12 +22,11 @@ class AlertV2OperandValue: string_value: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: 'AlertV2OperandValueDict') -> 'Self': + def from_dict(cls, value: "AlertV2OperandValueDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertV2OperandValueDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertV2OperandValueDict": + return _transform_to_json_value(self) # type:ignore class AlertV2OperandValueDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_run_as.py b/python/databricks/bundles/alerts/_models/alert_v2_run_as.py index 07fafdc17d3..164318184ab 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_run_as.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_run_as.py @@ -1,41 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertV2RunAs: """""" @@ -51,12 +26,11 @@ class AlertV2RunAs: """ @classmethod - def from_dict(cls, value: 'AlertV2RunAsDict') -> 'Self': + def from_dict(cls, value: "AlertV2RunAsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertV2RunAsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertV2RunAsDict": + return _transform_to_json_value(self) # type:ignore class AlertV2RunAsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/alert_v2_subscription.py b/python/databricks/bundles/alerts/_models/alert_v2_subscription.py index d87b69d0198..8d379a1ced7 100644 --- a/python/databricks/bundles/alerts/_models/alert_v2_subscription.py +++ b/python/databricks/bundles/alerts/_models/alert_v2_subscription.py @@ -1,41 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertV2Subscription: """""" @@ -45,12 +20,11 @@ class AlertV2Subscription: user_email: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: 'AlertV2SubscriptionDict') -> 'Self': + def from_dict(cls, value: "AlertV2SubscriptionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertV2SubscriptionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertV2SubscriptionDict": + return _transform_to_json_value(self) # type:ignore class AlertV2SubscriptionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/comparison_operator.py b/python/databricks/bundles/alerts/_models/comparison_operator.py index 268914118ca..95dc77260a0 100644 --- a/python/databricks/bundles/alerts/_models/comparison_operator.py +++ b/python/databricks/bundles/alerts/_models/comparison_operator.py @@ -1,35 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam +from typing import Literal class ComparisonOperator(Enum): @@ -42,4 +14,17 @@ class ComparisonOperator(Enum): IS_NULL = "IS_NULL" IS_NOT_NULL = "IS_NOT_NULL" -ComparisonOperatorParam = Literal["LESS_THAN", "GREATER_THAN", "EQUAL", "NOT_EQUAL", "GREATER_THAN_OR_EQUAL", "LESS_THAN_OR_EQUAL", "IS_NULL", "IS_NOT_NULL"] | ComparisonOperator + +ComparisonOperatorParam = ( + Literal[ + "LESS_THAN", + "GREATER_THAN", + "EQUAL", + "NOT_EQUAL", + "GREATER_THAN_OR_EQUAL", + "LESS_THAN_OR_EQUAL", + "IS_NULL", + "IS_NOT_NULL", + ] + | ComparisonOperator +) diff --git a/python/databricks/bundles/alerts/_models/cron_schedule.py b/python/databricks/bundles/alerts/_models/cron_schedule.py index 474e5e841f1..987a9a1056b 100644 --- a/python/databricks/bundles/alerts/_models/cron_schedule.py +++ b/python/databricks/bundles/alerts/_models/cron_schedule.py @@ -1,41 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.alerts._models.schedule_pause_status import ( + SchedulePauseStatus, + SchedulePauseStatusParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class CronSchedule: """""" @@ -59,12 +38,11 @@ class CronSchedule: """ @classmethod - def from_dict(cls, value: 'CronScheduleDict') -> 'Self': + def from_dict(cls, value: "CronScheduleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CronScheduleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CronScheduleDict": + return _transform_to_json_value(self) # type:ignore class CronScheduleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/custom_template_format.py b/python/databricks/bundles/alerts/_models/custom_template_format.py index d22bbab2303..fbddaecfc0e 100644 --- a/python/databricks/bundles/alerts/_models/custom_template_format.py +++ b/python/databricks/bundles/alerts/_models/custom_template_format.py @@ -1,44 +1,18 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam +from typing import Literal class CustomTemplateFormat(Enum): """ :meta private: [EXPERIMENTAL] - + Rendering format for the `custom_summary` and `custom_description` templates. """ + HTML = "HTML" MARKDOWN = "MARKDOWN" + CustomTemplateFormatParam = Literal["HTML", "MARKDOWN"] | CustomTemplateFormat diff --git a/python/databricks/bundles/alerts/_models/lifecycle.py b/python/databricks/bundles/alerts/_models/lifecycle.py index ef6a9fe2352..697776a198e 100644 --- a/python/databricks/bundles/alerts/_models/lifecycle.py +++ b/python/databricks/bundles/alerts/_models/lifecycle.py @@ -1,41 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -46,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/permission.py b/python/databricks/bundles/alerts/_models/permission.py index 9d4b5e813d5..e4091c6d061 100644 --- a/python/databricks/bundles/alerts/_models/permission.py +++ b/python/databricks/bundles/alerts/_models/permission.py @@ -1,41 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.alerts._models.permission_level import ( + PermissionLevel, + PermissionLevelParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Permission: """""" @@ -61,12 +40,11 @@ class Permission: """ @classmethod - def from_dict(cls, value: 'PermissionDict') -> 'Self': + def from_dict(cls, value: "PermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PermissionDict": + return _transform_to_json_value(self) # type:ignore class PermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/alerts/_models/permission_level.py b/python/databricks/bundles/alerts/_models/permission_level.py index c709c9ac644..c6111911b29 100644 --- a/python/databricks/bundles/alerts/_models/permission_level.py +++ b/python/databricks/bundles/alerts/_models/permission_level.py @@ -1,41 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam - -from databricks.bundles.alerts._models.schedule_pause_status import SchedulePauseStatus, SchedulePauseStatusParam +from typing import Literal class PermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" @@ -57,4 +30,29 @@ class PermissionLevel(Enum): CAN_MONITOR_ONLY = "CAN_MONITOR_ONLY" CAN_CREATE_APP = "CAN_CREATE_APP" -PermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW", "CAN_READ", "CAN_RUN", "CAN_EDIT", "CAN_USE", "CAN_MANAGE_STAGING_VERSIONS", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_EDIT_METADATA", "CAN_VIEW_METADATA", "CAN_BIND", "CAN_QUERY", "CAN_MONITOR", "CAN_CREATE", "CAN_MONITOR_ONLY", "CAN_CREATE_APP"] | PermissionLevel + +PermissionLevelParam = ( + Literal[ + "CAN_MANAGE", + "CAN_RESTART", + "CAN_ATTACH_TO", + "IS_OWNER", + "CAN_MANAGE_RUN", + "CAN_VIEW", + "CAN_READ", + "CAN_RUN", + "CAN_EDIT", + "CAN_USE", + "CAN_MANAGE_STAGING_VERSIONS", + "CAN_MANAGE_PRODUCTION_VERSIONS", + "CAN_EDIT_METADATA", + "CAN_VIEW_METADATA", + "CAN_BIND", + "CAN_QUERY", + "CAN_MONITOR", + "CAN_CREATE", + "CAN_MONITOR_ONLY", + "CAN_CREATE_APP", + ] + | PermissionLevel +) diff --git a/python/databricks/bundles/alerts/_models/schedule_pause_status.py b/python/databricks/bundles/alerts/_models/schedule_pause_status.py index a99d86c8174..ab05367f8c1 100644 --- a/python/databricks/bundles/alerts/_models/schedule_pause_status.py +++ b/python/databricks/bundles/alerts/_models/schedule_pause_status.py @@ -1,39 +1,12 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.alerts._models.alert_statement_parameter import AlertStatementParameter, AlertStatementParameterDict, AlertStatementParameterParam -from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation, AlertV2EvaluationDict, AlertV2EvaluationParam -from databricks.bundles.alerts._models.alert_v2_notification import AlertV2Notification, AlertV2NotificationDict, AlertV2NotificationParam -from databricks.bundles.alerts._models.alert_v2_operand import AlertV2Operand, AlertV2OperandDict, AlertV2OperandParam -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn, AlertV2OperandColumnDict, AlertV2OperandColumnParam -from databricks.bundles.alerts._models.alert_v2_operand_value import AlertV2OperandValue, AlertV2OperandValueDict, AlertV2OperandValueParam -from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs, AlertV2RunAsDict, AlertV2RunAsParam -from databricks.bundles.alerts._models.alert_v2_subscription import AlertV2Subscription, AlertV2SubscriptionDict, AlertV2SubscriptionParam -from databricks.bundles.alerts._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.alerts._models.alert import Alert, AlertDict, AlertParam -from databricks.bundles.alerts._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.alerts._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.alerts._models.permission_level import PermissionLevel, PermissionLevelParam - -from databricks.bundles.alerts._models.aggregation import Aggregation, AggregationParam - -from databricks.bundles.alerts._models.alert_evaluation_state import AlertEvaluationState, AlertEvaluationStateParam - -from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator, ComparisonOperatorParam - -from databricks.bundles.alerts._models.custom_template_format import CustomTemplateFormat, CustomTemplateFormatParam +from typing import Literal class SchedulePauseStatus(Enum): UNPAUSED = "UNPAUSED" PAUSED = "PAUSED" + SchedulePauseStatusParam = Literal["UNPAUSED", "PAUSED"] | SchedulePauseStatus diff --git a/python/databricks/bundles/apps/__init__.py b/python/databricks/bundles/apps/__init__.py index 442928f9265..7b92e6d61bb 100644 --- a/python/databricks/bundles/apps/__init__.py +++ b/python/databricks/bundles/apps/__init__.py @@ -1,138 +1,237 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'App', - 'AppConfig', - 'AppConfigDict', - 'AppConfigParam', - 'AppDict', - 'AppEnvVar', - 'AppEnvVarDict', - 'AppEnvVarParam', - 'AppParam', - 'AppPermission', - 'AppPermissionDict', - 'AppPermissionLevel', - 'AppPermissionLevelParam', - 'AppPermissionParam', - 'AppResource', - 'AppResourceApp', - 'AppResourceAppAppPermission', - 'AppResourceAppAppPermissionParam', - 'AppResourceAppDict', - 'AppResourceAppParam', - 'AppResourceDatabase', - 'AppResourceDatabaseDatabasePermission', - 'AppResourceDatabaseDatabasePermissionParam', - 'AppResourceDatabaseDict', - 'AppResourceDatabaseParam', - 'AppResourceDict', - 'AppResourceExperiment', - 'AppResourceExperimentDict', - 'AppResourceExperimentExperimentPermission', - 'AppResourceExperimentExperimentPermissionParam', - 'AppResourceExperimentParam', - 'AppResourceGenieSpace', - 'AppResourceGenieSpaceDict', - 'AppResourceGenieSpaceGenieSpacePermission', - 'AppResourceGenieSpaceGenieSpacePermissionParam', - 'AppResourceGenieSpaceParam', - 'AppResourceJob', - 'AppResourceJobDict', - 'AppResourceJobJobPermission', - 'AppResourceJobJobPermissionParam', - 'AppResourceJobParam', - 'AppResourceParam', - 'AppResourcePostgres', - 'AppResourcePostgresDict', - 'AppResourcePostgresParam', - 'AppResourcePostgresPostgresPermission', - 'AppResourcePostgresPostgresPermissionParam', - 'AppResourceSecret', - 'AppResourceSecretDict', - 'AppResourceSecretParam', - 'AppResourceSecretSecretPermission', - 'AppResourceSecretSecretPermissionParam', - 'AppResourceServingEndpoint', - 'AppResourceServingEndpointDict', - 'AppResourceServingEndpointParam', - 'AppResourceServingEndpointServingEndpointPermission', - 'AppResourceServingEndpointServingEndpointPermissionParam', - 'AppResourceSqlWarehouse', - 'AppResourceSqlWarehouseDict', - 'AppResourceSqlWarehouseParam', - 'AppResourceSqlWarehouseSqlWarehousePermission', - 'AppResourceSqlWarehouseSqlWarehousePermissionParam', - 'AppResourceUcSecurable', - 'AppResourceUcSecurableDict', - 'AppResourceUcSecurableParam', - 'AppResourceUcSecurableUcSecurablePermission', - 'AppResourceUcSecurableUcSecurablePermissionParam', - 'AppResourceUcSecurableUcSecurableType', - 'AppResourceUcSecurableUcSecurableTypeParam', - 'ComputeSize', - 'ComputeSizeParam', - 'GitRepository', - 'GitRepositoryDict', - 'GitRepositoryParam', - 'GitSource', - 'GitSourceDict', - 'GitSourceParam', - 'LifecycleWithStarted', - 'LifecycleWithStartedDict', - 'LifecycleWithStartedParam', - 'TelemetryExportDestination', - 'TelemetryExportDestinationDict', - 'TelemetryExportDestinationParam', - 'UnityCatalog', - 'UnityCatalogDict', - 'UnityCatalogParam', + "App", + "AppConfig", + "AppConfigDict", + "AppConfigParam", + "AppDict", + "AppEnvVar", + "AppEnvVarDict", + "AppEnvVarParam", + "AppParam", + "AppPermission", + "AppPermissionDict", + "AppPermissionLevel", + "AppPermissionLevelParam", + "AppPermissionParam", + "AppResource", + "AppResourceApp", + "AppResourceAppAppPermission", + "AppResourceAppAppPermissionParam", + "AppResourceAppDict", + "AppResourceAppParam", + "AppResourceDatabase", + "AppResourceDatabaseDatabasePermission", + "AppResourceDatabaseDatabasePermissionParam", + "AppResourceDatabaseDict", + "AppResourceDatabaseParam", + "AppResourceDict", + "AppResourceExperiment", + "AppResourceExperimentDict", + "AppResourceExperimentExperimentPermission", + "AppResourceExperimentExperimentPermissionParam", + "AppResourceExperimentParam", + "AppResourceGenieSpace", + "AppResourceGenieSpaceDict", + "AppResourceGenieSpaceGenieSpacePermission", + "AppResourceGenieSpaceGenieSpacePermissionParam", + "AppResourceGenieSpaceParam", + "AppResourceJob", + "AppResourceJobDict", + "AppResourceJobJobPermission", + "AppResourceJobJobPermissionParam", + "AppResourceJobParam", + "AppResourceParam", + "AppResourcePostgres", + "AppResourcePostgresDict", + "AppResourcePostgresParam", + "AppResourcePostgresPostgresPermission", + "AppResourcePostgresPostgresPermissionParam", + "AppResourceSecret", + "AppResourceSecretDict", + "AppResourceSecretParam", + "AppResourceSecretSecretPermission", + "AppResourceSecretSecretPermissionParam", + "AppResourceServingEndpoint", + "AppResourceServingEndpointDict", + "AppResourceServingEndpointParam", + "AppResourceServingEndpointServingEndpointPermission", + "AppResourceServingEndpointServingEndpointPermissionParam", + "AppResourceSqlWarehouse", + "AppResourceSqlWarehouseDict", + "AppResourceSqlWarehouseParam", + "AppResourceSqlWarehouseSqlWarehousePermission", + "AppResourceSqlWarehouseSqlWarehousePermissionParam", + "AppResourceUcSecurable", + "AppResourceUcSecurableDict", + "AppResourceUcSecurableParam", + "AppResourceUcSecurableUcSecurablePermission", + "AppResourceUcSecurableUcSecurablePermissionParam", + "AppResourceUcSecurableUcSecurableType", + "AppResourceUcSecurableUcSecurableTypeParam", + "ComputeSize", + "ComputeSizeParam", + "GitRepository", + "GitRepositoryDict", + "GitRepositoryParam", + "GitSource", + "GitSourceDict", + "GitSourceParam", + "LifecycleWithStarted", + "LifecycleWithStartedDict", + "LifecycleWithStartedParam", + "TelemetryExportDestination", + "TelemetryExportDestinationDict", + "TelemetryExportDestinationParam", + "UnityCatalog", + "UnityCatalogDict", + "UnityCatalogParam", ] -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - +from databricks.bundles.apps._models.app_config import ( + AppConfig, + AppConfigDict, + AppConfigParam, +) +from databricks.bundles.apps._models.app_env_var import ( + AppEnvVar, + AppEnvVarDict, + AppEnvVarParam, +) +from databricks.bundles.apps._models.app_permission import ( + AppPermission, + AppPermissionDict, + AppPermissionParam, +) +from databricks.bundles.apps._models.app_permission_level import ( + AppPermissionLevel, + AppPermissionLevelParam, +) +from databricks.bundles.apps._models.app_resource import ( + AppResource, + AppResourceDict, + AppResourceParam, +) +from databricks.bundles.apps._models.app_resource_app import ( + AppResourceApp, + AppResourceAppDict, + AppResourceAppParam, +) +from databricks.bundles.apps._models.app_resource_app_app_permission import ( + AppResourceAppAppPermission, + AppResourceAppAppPermissionParam, +) +from databricks.bundles.apps._models.app_resource_database import ( + AppResourceDatabase, + AppResourceDatabaseDict, + AppResourceDatabaseParam, +) +from databricks.bundles.apps._models.app_resource_database_database_permission import ( + AppResourceDatabaseDatabasePermission, + AppResourceDatabaseDatabasePermissionParam, +) +from databricks.bundles.apps._models.app_resource_experiment import ( + AppResourceExperiment, + AppResourceExperimentDict, + AppResourceExperimentParam, +) +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import ( + AppResourceExperimentExperimentPermission, + AppResourceExperimentExperimentPermissionParam, +) +from databricks.bundles.apps._models.app_resource_genie_space import ( + AppResourceGenieSpace, + AppResourceGenieSpaceDict, + AppResourceGenieSpaceParam, +) +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import ( + AppResourceGenieSpaceGenieSpacePermission, + AppResourceGenieSpaceGenieSpacePermissionParam, +) +from databricks.bundles.apps._models.app_resource_job import ( + AppResourceJob, + AppResourceJobDict, + AppResourceJobParam, +) +from databricks.bundles.apps._models.app_resource_job_job_permission import ( + AppResourceJobJobPermission, + AppResourceJobJobPermissionParam, +) +from databricks.bundles.apps._models.app_resource_postgres import ( + AppResourcePostgres, + AppResourcePostgresDict, + AppResourcePostgresParam, +) +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import ( + AppResourcePostgresPostgresPermission, + AppResourcePostgresPostgresPermissionParam, +) +from databricks.bundles.apps._models.app_resource_secret import ( + AppResourceSecret, + AppResourceSecretDict, + AppResourceSecretParam, +) +from databricks.bundles.apps._models.app_resource_secret_secret_permission import ( + AppResourceSecretSecretPermission, + AppResourceSecretSecretPermissionParam, +) +from databricks.bundles.apps._models.app_resource_serving_endpoint import ( + AppResourceServingEndpoint, + AppResourceServingEndpointDict, + AppResourceServingEndpointParam, +) +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import ( + AppResourceServingEndpointServingEndpointPermission, + AppResourceServingEndpointServingEndpointPermissionParam, +) +from databricks.bundles.apps._models.app_resource_sql_warehouse import ( + AppResourceSqlWarehouse, + AppResourceSqlWarehouseDict, + AppResourceSqlWarehouseParam, +) +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import ( + AppResourceSqlWarehouseSqlWarehousePermission, + AppResourceSqlWarehouseSqlWarehousePermissionParam, +) +from databricks.bundles.apps._models.app_resource_uc_securable import ( + AppResourceUcSecurable, + AppResourceUcSecurableDict, + AppResourceUcSecurableParam, +) +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import ( + AppResourceUcSecurableUcSecurablePermission, + AppResourceUcSecurableUcSecurablePermissionParam, +) +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import ( + AppResourceUcSecurableUcSecurableType, + AppResourceUcSecurableUcSecurableTypeParam, +) from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.apps._models.git_repository import ( + GitRepository, + GitRepositoryDict, + GitRepositoryParam, +) +from databricks.bundles.apps._models.git_source import ( + GitSource, + GitSourceDict, + GitSourceParam, +) +from databricks.bundles.apps._models.lifecycle_with_started import ( + LifecycleWithStarted, + LifecycleWithStartedDict, + LifecycleWithStartedParam, +) +from databricks.bundles.apps._models.telemetry_export_destination import ( + TelemetryExportDestination, + TelemetryExportDestinationDict, + TelemetryExportDestinationParam, +) +from databricks.bundles.apps._models.unity_catalog import ( + UnityCatalog, + UnityCatalogDict, + UnityCatalogParam, +) diff --git a/python/databricks/bundles/apps/_models/app.py b/python/databricks/bundles/apps/_models/app.py index 3826caa1379..533172ae6b2 100644 --- a/python/databricks/bundles/apps/_models/app.py +++ b/python/databricks/bundles/apps/_models/app.py @@ -1,63 +1,44 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.apps._models.app_config import ( + AppConfig, + AppConfigParam, +) +from databricks.bundles.apps._models.app_permission import ( + AppPermission, + AppPermissionParam, +) +from databricks.bundles.apps._models.app_resource import AppResource, AppResourceParam +from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from databricks.bundles.apps._models.git_repository import ( + GitRepository, + GitRepositoryParam, +) +from databricks.bundles.apps._models.git_source import GitSource, GitSourceParam +from databricks.bundles.apps._models.lifecycle_with_started import ( + LifecycleWithStarted, + LifecycleWithStartedParam, +) +from databricks.bundles.apps._models.telemetry_export_destination import ( + TelemetryExportDestination, + TelemetryExportDestinationParam, +) from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class App(Resource): """""" @@ -136,7 +117,9 @@ class App(Resource): [Private Preview] Name of the space this app belongs to. """ - telemetry_export_destinations: VariableOrList[TelemetryExportDestination] = field(default_factory=list) + telemetry_export_destinations: VariableOrList[TelemetryExportDestination] = field( + default_factory=list + ) usage_policy_id: VariableOrOptional[str] = None """ @@ -149,12 +132,11 @@ class App(Resource): """ @classmethod - def from_dict(cls, value: 'AppDict') -> 'Self': + def from_dict(cls, value: "AppDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppDict": + return _transform_to_json_value(self) # type:ignore class AppDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_config.py b/python/databricks/bundles/apps/_models/app_config.py index e689c834e26..e8a8f5fd70e 100644 --- a/python/databricks/bundles/apps/_models/app_config.py +++ b/python/databricks/bundles/apps/_models/app_config.py @@ -1,63 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarParam from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppConfig: """""" @@ -67,12 +21,11 @@ class AppConfig: env: VariableOrList[AppEnvVar] = field(default_factory=list) @classmethod - def from_dict(cls, value: 'AppConfigDict') -> 'Self': + def from_dict(cls, value: "AppConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppConfigDict": + return _transform_to_json_value(self) # type:ignore class AppConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_env_var.py b/python/databricks/bundles/apps/_models/app_env_var.py index f02879f73b9..a0afec4223a 100644 --- a/python/databricks/bundles/apps/_models/app_env_var.py +++ b/python/databricks/bundles/apps/_models/app_env_var.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppEnvVar: """""" @@ -69,12 +22,11 @@ class AppEnvVar: value_from: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: 'AppEnvVarDict') -> 'Self': + def from_dict(cls, value: "AppEnvVarDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppEnvVarDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppEnvVarDict": + return _transform_to_json_value(self) # type:ignore class AppEnvVarDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_permission.py b/python/databricks/bundles/apps/_models/app_permission.py index 7c00cb07c69..f04638d0df2 100644 --- a/python/databricks/bundles/apps/_models/app_permission.py +++ b/python/databricks/bundles/apps/_models/app_permission.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_permission_level import ( + AppPermissionLevel, + AppPermissionLevelParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppPermission: """""" @@ -83,12 +40,11 @@ class AppPermission: """ @classmethod - def from_dict(cls, value: 'AppPermissionDict') -> 'Self': + def from_dict(cls, value: "AppPermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppPermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppPermissionDict": + return _transform_to_json_value(self) # type:ignore class AppPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_permission_level.py b/python/databricks/bundles/apps/_models/app_permission_level.py index 00b62909dec..a951c137ea7 100644 --- a/python/databricks/bundles/apps/_models/app_permission_level.py +++ b/python/databricks/bundles/apps/_models/app_permission_level.py @@ -1,64 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppPermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_USE = "CAN_USE" + AppPermissionLevelParam = Literal["CAN_MANAGE", "CAN_USE"] | AppPermissionLevel diff --git a/python/databricks/bundles/apps/_models/app_resource.py b/python/databricks/bundles/apps/_models/app_resource.py index bbb77c9604b..cc96b083c2c 100644 --- a/python/databricks/bundles/apps/_models/app_resource.py +++ b/python/databricks/bundles/apps/_models/app_resource.py @@ -1,63 +1,56 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.apps._models.app_resource_app import ( + AppResourceApp, + AppResourceAppParam, +) +from databricks.bundles.apps._models.app_resource_database import ( + AppResourceDatabase, + AppResourceDatabaseParam, +) +from databricks.bundles.apps._models.app_resource_experiment import ( + AppResourceExperiment, + AppResourceExperimentParam, +) +from databricks.bundles.apps._models.app_resource_genie_space import ( + AppResourceGenieSpace, + AppResourceGenieSpaceParam, +) +from databricks.bundles.apps._models.app_resource_job import ( + AppResourceJob, + AppResourceJobParam, +) +from databricks.bundles.apps._models.app_resource_postgres import ( + AppResourcePostgres, + AppResourcePostgresParam, +) +from databricks.bundles.apps._models.app_resource_secret import ( + AppResourceSecret, + AppResourceSecretParam, +) +from databricks.bundles.apps._models.app_resource_serving_endpoint import ( + AppResourceServingEndpoint, + AppResourceServingEndpointParam, +) +from databricks.bundles.apps._models.app_resource_sql_warehouse import ( + AppResourceSqlWarehouse, + AppResourceSqlWarehouseParam, +) +from databricks.bundles.apps._models.app_resource_uc_securable import ( + AppResourceUcSecurable, + AppResourceUcSecurableParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResource: """""" @@ -93,12 +86,11 @@ class AppResource: uc_securable: VariableOrOptional[AppResourceUcSecurable] = None @classmethod - def from_dict(cls, value: 'AppResourceDict') -> 'Self': + def from_dict(cls, value: "AppResourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceDict": + return _transform_to_json_value(self) # type:ignore class AppResourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_app.py b/python/databricks/bundles/apps/_models/app_resource_app.py index ef048ba8f0b..a7a05689c2e 100644 --- a/python/databricks/bundles/apps/_models/app_resource_app.py +++ b/python/databricks/bundles/apps/_models/app_resource_app.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_resource_app_app_permission import ( + AppResourceAppAppPermission, + AppResourceAppAppPermissionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourceApp: """""" @@ -67,12 +24,11 @@ class AppResourceApp: permission: VariableOrOptional[AppResourceAppAppPermission] = None @classmethod - def from_dict(cls, value: 'AppResourceAppDict') -> 'Self': + def from_dict(cls, value: "AppResourceAppDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceAppDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceAppDict": + return _transform_to_json_value(self) # type:ignore class AppResourceAppDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_app_app_permission.py b/python/databricks/bundles/apps/_models/app_resource_app_app_permission.py index 1c91a920547..a58dca414e4 100644 --- a/python/databricks/bundles/apps/_models/app_resource_app_app_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_app_app_permission.py @@ -1,60 +1,11 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceAppAppPermission(Enum): CAN_USE = "CAN_USE" + AppResourceAppAppPermissionParam = Literal["CAN_USE"] | AppResourceAppAppPermission diff --git a/python/databricks/bundles/apps/_models/app_resource_database.py b/python/databricks/bundles/apps/_models/app_resource_database.py index c7f963650d0..b29ce973105 100644 --- a/python/databricks/bundles/apps/_models/app_resource_database.py +++ b/python/databricks/bundles/apps/_models/app_resource_database.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_resource_database_database_permission import ( + AppResourceDatabaseDatabasePermission, + AppResourceDatabaseDatabasePermissionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourceDatabase: """""" @@ -69,12 +26,11 @@ class AppResourceDatabase: permission: VariableOr[AppResourceDatabaseDatabasePermission] @classmethod - def from_dict(cls, value: 'AppResourceDatabaseDict') -> 'Self': + def from_dict(cls, value: "AppResourceDatabaseDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceDatabaseDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceDatabaseDict": + return _transform_to_json_value(self) # type:ignore class AppResourceDatabaseDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_database_database_permission.py b/python/databricks/bundles/apps/_models/app_resource_database_database_permission.py index ceb3547b2b6..447d015a1c2 100644 --- a/python/databricks/bundles/apps/_models/app_resource_database_database_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_database_database_permission.py @@ -1,60 +1,13 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceDatabaseDatabasePermission(Enum): CAN_CONNECT_AND_CREATE = "CAN_CONNECT_AND_CREATE" -AppResourceDatabaseDatabasePermissionParam = Literal["CAN_CONNECT_AND_CREATE"] | AppResourceDatabaseDatabasePermission + +AppResourceDatabaseDatabasePermissionParam = ( + Literal["CAN_CONNECT_AND_CREATE"] | AppResourceDatabaseDatabasePermission +) diff --git a/python/databricks/bundles/apps/_models/app_resource_experiment.py b/python/databricks/bundles/apps/_models/app_resource_experiment.py index 5c3d8c63446..e69d4be1a17 100644 --- a/python/databricks/bundles/apps/_models/app_resource_experiment.py +++ b/python/databricks/bundles/apps/_models/app_resource_experiment.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import ( + AppResourceExperimentExperimentPermission, + AppResourceExperimentExperimentPermissionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourceExperiment: """""" @@ -67,12 +24,11 @@ class AppResourceExperiment: permission: VariableOr[AppResourceExperimentExperimentPermission] @classmethod - def from_dict(cls, value: 'AppResourceExperimentDict') -> 'Self': + def from_dict(cls, value: "AppResourceExperimentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceExperimentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceExperimentDict": + return _transform_to_json_value(self) # type:ignore class AppResourceExperimentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_experiment_experiment_permission.py b/python/databricks/bundles/apps/_models/app_resource_experiment_experiment_permission.py index f8e4ab24765..230b8d5c06e 100644 --- a/python/databricks/bundles/apps/_models/app_resource_experiment_experiment_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_experiment_experiment_permission.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceExperimentExperimentPermission(Enum): @@ -59,4 +9,8 @@ class AppResourceExperimentExperimentPermission(Enum): CAN_EDIT = "CAN_EDIT" CAN_READ = "CAN_READ" -AppResourceExperimentExperimentPermissionParam = Literal["CAN_MANAGE", "CAN_EDIT", "CAN_READ"] | AppResourceExperimentExperimentPermission + +AppResourceExperimentExperimentPermissionParam = ( + Literal["CAN_MANAGE", "CAN_EDIT", "CAN_READ"] + | AppResourceExperimentExperimentPermission +) diff --git a/python/databricks/bundles/apps/_models/app_resource_genie_space.py b/python/databricks/bundles/apps/_models/app_resource_genie_space.py index cb3f30a767c..0cfdaf316f4 100644 --- a/python/databricks/bundles/apps/_models/app_resource_genie_space.py +++ b/python/databricks/bundles/apps/_models/app_resource_genie_space.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import ( + AppResourceGenieSpaceGenieSpacePermission, + AppResourceGenieSpaceGenieSpacePermissionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourceGenieSpace: """""" @@ -69,12 +26,11 @@ class AppResourceGenieSpace: space_id: VariableOr[str] @classmethod - def from_dict(cls, value: 'AppResourceGenieSpaceDict') -> 'Self': + def from_dict(cls, value: "AppResourceGenieSpaceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceGenieSpaceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceGenieSpaceDict": + return _transform_to_json_value(self) # type:ignore class AppResourceGenieSpaceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_genie_space_genie_space_permission.py b/python/databricks/bundles/apps/_models/app_resource_genie_space_genie_space_permission.py index 591978b1637..312897a8cce 100644 --- a/python/databricks/bundles/apps/_models/app_resource_genie_space_genie_space_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_genie_space_genie_space_permission.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceGenieSpaceGenieSpacePermission(Enum): @@ -60,4 +10,8 @@ class AppResourceGenieSpaceGenieSpacePermission(Enum): CAN_RUN = "CAN_RUN" CAN_VIEW = "CAN_VIEW" -AppResourceGenieSpaceGenieSpacePermissionParam = Literal["CAN_MANAGE", "CAN_EDIT", "CAN_RUN", "CAN_VIEW"] | AppResourceGenieSpaceGenieSpacePermission + +AppResourceGenieSpaceGenieSpacePermissionParam = ( + Literal["CAN_MANAGE", "CAN_EDIT", "CAN_RUN", "CAN_VIEW"] + | AppResourceGenieSpaceGenieSpacePermission +) diff --git a/python/databricks/bundles/apps/_models/app_resource_job.py b/python/databricks/bundles/apps/_models/app_resource_job.py index 83a3c397197..08ca1be6cfc 100644 --- a/python/databricks/bundles/apps/_models/app_resource_job.py +++ b/python/databricks/bundles/apps/_models/app_resource_job.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_resource_job_job_permission import ( + AppResourceJobJobPermission, + AppResourceJobJobPermissionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourceJob: """""" @@ -73,12 +30,11 @@ class AppResourceJob: """ @classmethod - def from_dict(cls, value: 'AppResourceJobDict') -> 'Self': + def from_dict(cls, value: "AppResourceJobDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceJobDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceJobDict": + return _transform_to_json_value(self) # type:ignore class AppResourceJobDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_job_job_permission.py b/python/databricks/bundles/apps/_models/app_resource_job_job_permission.py index 14fdea47b31..29f08d0aab1 100644 --- a/python/databricks/bundles/apps/_models/app_resource_job_job_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_job_job_permission.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceJobJobPermission(Enum): @@ -60,4 +10,8 @@ class AppResourceJobJobPermission(Enum): CAN_MANAGE_RUN = "CAN_MANAGE_RUN" CAN_VIEW = "CAN_VIEW" -AppResourceJobJobPermissionParam = Literal["CAN_MANAGE", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW"] | AppResourceJobJobPermission + +AppResourceJobJobPermissionParam = ( + Literal["CAN_MANAGE", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW"] + | AppResourceJobJobPermission +) diff --git a/python/databricks/bundles/apps/_models/app_resource_postgres.py b/python/databricks/bundles/apps/_models/app_resource_postgres.py index 5c6bf80ed85..0d3b6e39b17 100644 --- a/python/databricks/bundles/apps/_models/app_resource_postgres.py +++ b/python/databricks/bundles/apps/_models/app_resource_postgres.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import ( + AppResourcePostgresPostgresPermission, + AppResourcePostgresPostgresPermissionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourcePostgres: """""" @@ -69,12 +26,11 @@ class AppResourcePostgres: permission: VariableOrOptional[AppResourcePostgresPostgresPermission] = None @classmethod - def from_dict(cls, value: 'AppResourcePostgresDict') -> 'Self': + def from_dict(cls, value: "AppResourcePostgresDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourcePostgresDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourcePostgresDict": + return _transform_to_json_value(self) # type:ignore class AppResourcePostgresDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_postgres_postgres_permission.py b/python/databricks/bundles/apps/_models/app_resource_postgres_postgres_permission.py index 19687d44b81..85657edfc35 100644 --- a/python/databricks/bundles/apps/_models/app_resource_postgres_postgres_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_postgres_postgres_permission.py @@ -1,60 +1,13 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourcePostgresPostgresPermission(Enum): CAN_CONNECT_AND_CREATE = "CAN_CONNECT_AND_CREATE" -AppResourcePostgresPostgresPermissionParam = Literal["CAN_CONNECT_AND_CREATE"] | AppResourcePostgresPostgresPermission + +AppResourcePostgresPostgresPermissionParam = ( + Literal["CAN_CONNECT_AND_CREATE"] | AppResourcePostgresPostgresPermission +) diff --git a/python/databricks/bundles/apps/_models/app_resource_secret.py b/python/databricks/bundles/apps/_models/app_resource_secret.py index c09c69d7fa5..3ecca5e2199 100644 --- a/python/databricks/bundles/apps/_models/app_resource_secret.py +++ b/python/databricks/bundles/apps/_models/app_resource_secret.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_resource_secret_secret_permission import ( + AppResourceSecretSecretPermission, + AppResourceSecretSecretPermissionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourceSecret: """""" @@ -78,12 +35,11 @@ class AppResourceSecret: """ @classmethod - def from_dict(cls, value: 'AppResourceSecretDict') -> 'Self': + def from_dict(cls, value: "AppResourceSecretDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceSecretDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceSecretDict": + return _transform_to_json_value(self) # type:ignore class AppResourceSecretDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_secret_secret_permission.py b/python/databricks/bundles/apps/_models/app_resource_secret_secret_permission.py index 76a03686e7f..c30b6d977ff 100644 --- a/python/databricks/bundles/apps/_models/app_resource_secret_secret_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_secret_secret_permission.py @@ -1,65 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceSecretSecretPermission(Enum): """ Permission to grant on the secret scope. Supported permissions are: "READ", "WRITE", "MANAGE". """ + READ = "READ" WRITE = "WRITE" MANAGE = "MANAGE" -AppResourceSecretSecretPermissionParam = Literal["READ", "WRITE", "MANAGE"] | AppResourceSecretSecretPermission + +AppResourceSecretSecretPermissionParam = ( + Literal["READ", "WRITE", "MANAGE"] | AppResourceSecretSecretPermission +) diff --git a/python/databricks/bundles/apps/_models/app_resource_serving_endpoint.py b/python/databricks/bundles/apps/_models/app_resource_serving_endpoint.py index a8c00664f9c..ee444cd7692 100644 --- a/python/databricks/bundles/apps/_models/app_resource_serving_endpoint.py +++ b/python/databricks/bundles/apps/_models/app_resource_serving_endpoint.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import ( + AppResourceServingEndpointServingEndpointPermission, + AppResourceServingEndpointServingEndpointPermissionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourceServingEndpoint: """""" @@ -73,12 +30,11 @@ class AppResourceServingEndpoint: """ @classmethod - def from_dict(cls, value: 'AppResourceServingEndpointDict') -> 'Self': + def from_dict(cls, value: "AppResourceServingEndpointDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceServingEndpointDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceServingEndpointDict": + return _transform_to_json_value(self) # type:ignore class AppResourceServingEndpointDict(TypedDict, total=False): @@ -95,4 +51,6 @@ class AppResourceServingEndpointDict(TypedDict, total=False): """ -AppResourceServingEndpointParam = AppResourceServingEndpointDict | AppResourceServingEndpoint +AppResourceServingEndpointParam = ( + AppResourceServingEndpointDict | AppResourceServingEndpoint +) diff --git a/python/databricks/bundles/apps/_models/app_resource_serving_endpoint_serving_endpoint_permission.py b/python/databricks/bundles/apps/_models/app_resource_serving_endpoint_serving_endpoint_permission.py index 10320643658..abfe5e63be0 100644 --- a/python/databricks/bundles/apps/_models/app_resource_serving_endpoint_serving_endpoint_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_serving_endpoint_serving_endpoint_permission.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceServingEndpointServingEndpointPermission(Enum): @@ -59,4 +9,8 @@ class AppResourceServingEndpointServingEndpointPermission(Enum): CAN_QUERY = "CAN_QUERY" CAN_VIEW = "CAN_VIEW" -AppResourceServingEndpointServingEndpointPermissionParam = Literal["CAN_MANAGE", "CAN_QUERY", "CAN_VIEW"] | AppResourceServingEndpointServingEndpointPermission + +AppResourceServingEndpointServingEndpointPermissionParam = ( + Literal["CAN_MANAGE", "CAN_QUERY", "CAN_VIEW"] + | AppResourceServingEndpointServingEndpointPermission +) diff --git a/python/databricks/bundles/apps/_models/app_resource_sql_warehouse.py b/python/databricks/bundles/apps/_models/app_resource_sql_warehouse.py index 00d50b41d8a..691528a0450 100644 --- a/python/databricks/bundles/apps/_models/app_resource_sql_warehouse.py +++ b/python/databricks/bundles/apps/_models/app_resource_sql_warehouse.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import ( + AppResourceSqlWarehouseSqlWarehousePermission, + AppResourceSqlWarehouseSqlWarehousePermissionParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourceSqlWarehouse: """""" @@ -73,12 +30,11 @@ class AppResourceSqlWarehouse: """ @classmethod - def from_dict(cls, value: 'AppResourceSqlWarehouseDict') -> 'Self': + def from_dict(cls, value: "AppResourceSqlWarehouseDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceSqlWarehouseDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceSqlWarehouseDict": + return _transform_to_json_value(self) # type:ignore class AppResourceSqlWarehouseDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_sql_warehouse_sql_warehouse_permission.py b/python/databricks/bundles/apps/_models/app_resource_sql_warehouse_sql_warehouse_permission.py index 7911bec3f99..ceb417a6edb 100644 --- a/python/databricks/bundles/apps/_models/app_resource_sql_warehouse_sql_warehouse_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_sql_warehouse_sql_warehouse_permission.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceSqlWarehouseSqlWarehousePermission(Enum): @@ -59,4 +9,8 @@ class AppResourceSqlWarehouseSqlWarehousePermission(Enum): CAN_USE = "CAN_USE" IS_OWNER = "IS_OWNER" -AppResourceSqlWarehouseSqlWarehousePermissionParam = Literal["CAN_MANAGE", "CAN_USE", "IS_OWNER"] | AppResourceSqlWarehouseSqlWarehousePermission + +AppResourceSqlWarehouseSqlWarehousePermissionParam = ( + Literal["CAN_MANAGE", "CAN_USE", "IS_OWNER"] + | AppResourceSqlWarehouseSqlWarehousePermission +) diff --git a/python/databricks/bundles/apps/_models/app_resource_uc_securable.py b/python/databricks/bundles/apps/_models/app_resource_uc_securable.py index ef587a9c3c2..196b2396592 100644 --- a/python/databricks/bundles/apps/_models/app_resource_uc_securable.py +++ b/python/databricks/bundles/apps/_models/app_resource_uc_securable.py @@ -1,63 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import ( + AppResourceUcSecurableUcSecurablePermission, + AppResourceUcSecurableUcSecurablePermissionParam, +) +from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import ( + AppResourceUcSecurableUcSecurableType, + AppResourceUcSecurableUcSecurableTypeParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AppResourceUcSecurable: """""" @@ -69,12 +30,11 @@ class AppResourceUcSecurable: securable_type: VariableOr[AppResourceUcSecurableUcSecurableType] @classmethod - def from_dict(cls, value: 'AppResourceUcSecurableDict') -> 'Self': + def from_dict(cls, value: "AppResourceUcSecurableDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AppResourceUcSecurableDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AppResourceUcSecurableDict": + return _transform_to_json_value(self) # type:ignore class AppResourceUcSecurableDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_permission.py b/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_permission.py index b163818ff50..bd4ed5005db 100644 --- a/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_permission.py +++ b/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_permission.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceUcSecurableUcSecurablePermission(Enum): @@ -62,4 +12,10 @@ class AppResourceUcSecurableUcSecurablePermission(Enum): USE_CONNECTION = "USE_CONNECTION" MODIFY = "MODIFY" -AppResourceUcSecurableUcSecurablePermissionParam = Literal["READ_VOLUME", "WRITE_VOLUME", "SELECT", "EXECUTE", "USE_CONNECTION", "MODIFY"] | AppResourceUcSecurableUcSecurablePermission + +AppResourceUcSecurableUcSecurablePermissionParam = ( + Literal[ + "READ_VOLUME", "WRITE_VOLUME", "SELECT", "EXECUTE", "USE_CONNECTION", "MODIFY" + ] + | AppResourceUcSecurableUcSecurablePermission +) diff --git a/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_type.py b/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_type.py index ed3bd8009fd..46fc5390e24 100644 --- a/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_type.py +++ b/python/databricks/bundles/apps/_models/app_resource_uc_securable_uc_securable_type.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam +from typing import Literal class AppResourceUcSecurableUcSecurableType(Enum): @@ -61,4 +11,8 @@ class AppResourceUcSecurableUcSecurableType(Enum): CONNECTION = "CONNECTION" MODEL_SERVICE = "MODEL_SERVICE" -AppResourceUcSecurableUcSecurableTypeParam = Literal["VOLUME", "TABLE", "FUNCTION", "CONNECTION", "MODEL_SERVICE"] | AppResourceUcSecurableUcSecurableType + +AppResourceUcSecurableUcSecurableTypeParam = ( + Literal["VOLUME", "TABLE", "FUNCTION", "CONNECTION", "MODEL_SERVICE"] + | AppResourceUcSecurableUcSecurableType +) diff --git a/python/databricks/bundles/apps/_models/compute_size.py b/python/databricks/bundles/apps/_models/compute_size.py index a8e07dcde5f..4881d034935 100644 --- a/python/databricks/bundles/apps/_models/compute_size.py +++ b/python/databricks/bundles/apps/_models/compute_size.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam +from typing import Literal class ComputeSize(Enum): @@ -59,4 +9,5 @@ class ComputeSize(Enum): LARGE = "LARGE" XLARGE = "XLARGE" + ComputeSizeParam = Literal["MEDIUM", "LARGE", "XLARGE"] | ComputeSize diff --git a/python/databricks/bundles/apps/_models/git_repository.py b/python/databricks/bundles/apps/_models/git_repository.py index b8877195bfa..d13ae5b2a56 100644 --- a/python/databricks/bundles/apps/_models/git_repository.py +++ b/python/databricks/bundles/apps/_models/git_repository.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GitRepository: """ Git repository configuration specifying the location of the repository. """ + provider: VariableOr[str] """ Git provider. Case insensitive. Supported values: gitHub, gitHubEnterprise, bitbucketCloud, @@ -87,12 +41,11 @@ class GitRepository: """ @classmethod - def from_dict(cls, value: 'GitRepositoryDict') -> 'Self': + def from_dict(cls, value: "GitRepositoryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GitRepositoryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GitRepositoryDict": + return _transform_to_json_value(self) # type:ignore class GitRepositoryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/git_source.py b/python/databricks/bundles/apps/_models/git_source.py index cecee53d65a..93bd61f5aa4 100644 --- a/python/databricks/bundles/apps/_models/git_source.py +++ b/python/databricks/bundles/apps/_models/git_source.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GitSource: """ Complete git source specification including repository location and reference. """ + branch: VariableOrOptional[str] = None """ Git branch to checkout. @@ -85,12 +39,11 @@ class GitSource: """ @classmethod - def from_dict(cls, value: 'GitSourceDict') -> 'Self': + def from_dict(cls, value: "GitSourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GitSourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GitSourceDict": + return _transform_to_json_value(self) # type:ignore class GitSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/lifecycle_with_started.py b/python/databricks/bundles/apps/_models/lifecycle_with_started.py index 9b04e6ddfbd..f9cb9b8d49c 100644 --- a/python/databricks/bundles/apps/_models/lifecycle_with_started.py +++ b/python/databricks/bundles/apps/_models/lifecycle_with_started.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LifecycleWithStarted: """""" @@ -73,12 +26,11 @@ class LifecycleWithStarted: """ @classmethod - def from_dict(cls, value: 'LifecycleWithStartedDict') -> 'Self': + def from_dict(cls, value: "LifecycleWithStartedDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleWithStartedDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleWithStartedDict": + return _transform_to_json_value(self) # type:ignore class LifecycleWithStartedDict(TypedDict, total=False): diff --git a/python/databricks/bundles/apps/_models/telemetry_export_destination.py b/python/databricks/bundles/apps/_models/telemetry_export_destination.py index fc346ec938d..edf9108c8a1 100644 --- a/python/databricks/bundles/apps/_models/telemetry_export_destination.py +++ b/python/databricks/bundles/apps/_models/telemetry_export_destination.py @@ -1,80 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.apps._models.unity_catalog import ( + UnityCatalog, + UnityCatalogParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.unity_catalog import UnityCatalog, UnityCatalogDict, UnityCatalogParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TelemetryExportDestination: """ A single telemetry export destination with its configuration and status. """ + unity_catalog: VariableOrOptional[UnityCatalog] = None """ Unity Catalog Destinations for OTEL telemetry export. """ @classmethod - def from_dict(cls, value: 'TelemetryExportDestinationDict') -> 'Self': + def from_dict(cls, value: "TelemetryExportDestinationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TelemetryExportDestinationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TelemetryExportDestinationDict": + return _transform_to_json_value(self) # type:ignore class TelemetryExportDestinationDict(TypedDict, total=False): @@ -86,4 +43,6 @@ class TelemetryExportDestinationDict(TypedDict, total=False): """ -TelemetryExportDestinationParam = TelemetryExportDestinationDict | TelemetryExportDestination +TelemetryExportDestinationParam = ( + TelemetryExportDestinationDict | TelemetryExportDestination +) diff --git a/python/databricks/bundles/apps/_models/unity_catalog.py b/python/databricks/bundles/apps/_models/unity_catalog.py index 3dafe0f017d..f20d1c373f3 100644 --- a/python/databricks/bundles/apps/_models/unity_catalog.py +++ b/python/databricks/bundles/apps/_models/unity_catalog.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.apps._models.app_resource import AppResource, AppResourceDict, AppResourceParam -from databricks.bundles.apps._models.app_resource_app import AppResourceApp, AppResourceAppDict, AppResourceAppParam -from databricks.bundles.apps._models.app_resource_database import AppResourceDatabase, AppResourceDatabaseDict, AppResourceDatabaseParam -from databricks.bundles.apps._models.app_resource_experiment import AppResourceExperiment, AppResourceExperimentDict, AppResourceExperimentParam -from databricks.bundles.apps._models.app_resource_genie_space import AppResourceGenieSpace, AppResourceGenieSpaceDict, AppResourceGenieSpaceParam -from databricks.bundles.apps._models.app_resource_job import AppResourceJob, AppResourceJobDict, AppResourceJobParam -from databricks.bundles.apps._models.app_resource_postgres import AppResourcePostgres, AppResourcePostgresDict, AppResourcePostgresParam -from databricks.bundles.apps._models.app_resource_secret import AppResourceSecret, AppResourceSecretDict, AppResourceSecretParam -from databricks.bundles.apps._models.app_resource_serving_endpoint import AppResourceServingEndpoint, AppResourceServingEndpointDict, AppResourceServingEndpointParam -from databricks.bundles.apps._models.app_resource_sql_warehouse import AppResourceSqlWarehouse, AppResourceSqlWarehouseDict, AppResourceSqlWarehouseParam -from databricks.bundles.apps._models.app_resource_uc_securable import AppResourceUcSecurable, AppResourceUcSecurableDict, AppResourceUcSecurableParam -from databricks.bundles.apps._models.git_repository import GitRepository, GitRepositoryDict, GitRepositoryParam -from databricks.bundles.apps._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination, TelemetryExportDestinationDict, TelemetryExportDestinationParam -from databricks.bundles.apps._models.app import App, AppDict, AppParam -from databricks.bundles.apps._models.app_config import AppConfig, AppConfigDict, AppConfigParam -from databricks.bundles.apps._models.app_env_var import AppEnvVar, AppEnvVarDict, AppEnvVarParam -from databricks.bundles.apps._models.app_permission import AppPermission, AppPermissionDict, AppPermissionParam -from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.apps._models.app_permission_level import AppPermissionLevel, AppPermissionLevelParam - -from databricks.bundles.apps._models.app_resource_app_app_permission import AppResourceAppAppPermission, AppResourceAppAppPermissionParam - -from databricks.bundles.apps._models.app_resource_database_database_permission import AppResourceDatabaseDatabasePermission, AppResourceDatabaseDatabasePermissionParam - -from databricks.bundles.apps._models.app_resource_experiment_experiment_permission import AppResourceExperimentExperimentPermission, AppResourceExperimentExperimentPermissionParam - -from databricks.bundles.apps._models.app_resource_genie_space_genie_space_permission import AppResourceGenieSpaceGenieSpacePermission, AppResourceGenieSpaceGenieSpacePermissionParam - -from databricks.bundles.apps._models.app_resource_job_job_permission import AppResourceJobJobPermission, AppResourceJobJobPermissionParam - -from databricks.bundles.apps._models.app_resource_postgres_postgres_permission import AppResourcePostgresPostgresPermission, AppResourcePostgresPostgresPermissionParam - -from databricks.bundles.apps._models.app_resource_secret_secret_permission import AppResourceSecretSecretPermission, AppResourceSecretSecretPermissionParam - -from databricks.bundles.apps._models.app_resource_serving_endpoint_serving_endpoint_permission import AppResourceServingEndpointServingEndpointPermission, AppResourceServingEndpointServingEndpointPermissionParam - -from databricks.bundles.apps._models.app_resource_sql_warehouse_sql_warehouse_permission import AppResourceSqlWarehouseSqlWarehousePermission, AppResourceSqlWarehouseSqlWarehousePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_permission import AppResourceUcSecurableUcSecurablePermission, AppResourceUcSecurableUcSecurablePermissionParam - -from databricks.bundles.apps._models.app_resource_uc_securable_uc_securable_type import AppResourceUcSecurableUcSecurableType, AppResourceUcSecurableUcSecurableTypeParam - -from databricks.bundles.apps._models.compute_size import ComputeSize, ComputeSizeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class UnityCatalog: """ Unity Catalog Destinations for OTEL telemetry export. """ + logs_table: VariableOr[str] """ Unity Catalog table for OTEL logs. @@ -79,12 +33,11 @@ class UnityCatalog: """ @classmethod - def from_dict(cls, value: 'UnityCatalogDict') -> 'Self': + def from_dict(cls, value: "UnityCatalogDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'UnityCatalogDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "UnityCatalogDict": + return _transform_to_json_value(self) # type:ignore class UnityCatalogDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/__init__.py b/python/databricks/bundles/catalogs/__init__.py index cdd8527365f..f36750b4816 100644 --- a/python/databricks/bundles/catalogs/__init__.py +++ b/python/databricks/bundles/catalogs/__init__.py @@ -1,38 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'AzureEncryptionSettings', - 'AzureEncryptionSettingsDict', - 'AzureEncryptionSettingsParam', - 'Catalog', - 'CatalogDict', - 'CatalogParam', - 'EncryptionSettings', - 'EncryptionSettingsDict', - 'EncryptionSettingsParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'CatalogGrant', - 'CatalogGrantDict', - 'CatalogGrantParam', - 'CatalogGrantPrivilege', - 'CatalogGrantPrivilegeParam', + "AzureEncryptionSettings", + "AzureEncryptionSettingsDict", + "AzureEncryptionSettingsParam", + "Catalog", + "CatalogDict", + "CatalogGrant", + "CatalogGrantDict", + "CatalogGrantParam", + "CatalogGrantPrivilege", + "CatalogGrantPrivilegeParam", + "CatalogParam", + "EncryptionSettings", + "EncryptionSettingsDict", + "EncryptionSettingsParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", ] -from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam -from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam -from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam -from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.catalogs._models.azure_encryption_settings import ( + AzureEncryptionSettings, + AzureEncryptionSettingsDict, + AzureEncryptionSettingsParam, +) +from databricks.bundles.catalogs._models.catalog import ( + Catalog, + CatalogDict, + CatalogParam, +) +from databricks.bundles.catalogs._models.encryption_settings import ( + EncryptionSettings, + EncryptionSettingsDict, + EncryptionSettingsParam, +) +from databricks.bundles.catalogs._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.catalogs._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) CatalogGrant = PrivilegeAssignment CatalogGrantDict = PrivilegeAssignmentDict diff --git a/python/databricks/bundles/catalogs/_models/azure_encryption_settings.py b/python/databricks/bundles/catalogs/_models/azure_encryption_settings.py index dcab98af41b..d68ba739a24 100644 --- a/python/databricks/bundles/catalogs/_models/azure_encryption_settings.py +++ b/python/databricks/bundles/catalogs/_models/azure_encryption_settings.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam -from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam -from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AzureEncryptionSettings: """""" @@ -30,12 +22,11 @@ class AzureEncryptionSettings: azure_cmk_managed_identity_id: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: 'AzureEncryptionSettingsDict') -> 'Self': + def from_dict(cls, value: "AzureEncryptionSettingsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AzureEncryptionSettingsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AzureEncryptionSettingsDict": + return _transform_to_json_value(self) # type:ignore class AzureEncryptionSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/_models/catalog.py b/python/databricks/bundles/catalogs/_models/catalog.py index 43e3874b93f..bdb397b0cb0 100644 --- a/python/databricks/bundles/catalogs/_models/catalog.py +++ b/python/databricks/bundles/catalogs/_models/catalog.py @@ -1,24 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.catalogs._models.encryption_settings import ( + EncryptionSettings, + EncryptionSettingsParam, +) +from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.catalogs._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam -from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam -from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Catalog(Resource): """""" @@ -86,12 +93,11 @@ class Catalog(Resource): """ @classmethod - def from_dict(cls, value: 'CatalogDict') -> 'Self': + def from_dict(cls, value: "CatalogDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CatalogDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CatalogDict": + return _transform_to_json_value(self) # type:ignore class CatalogDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/_models/encryption_settings.py b/python/databricks/bundles/catalogs/_models/encryption_settings.py index 8f197ef3367..dfb53029dbe 100644 --- a/python/databricks/bundles/catalogs/_models/encryption_settings.py +++ b/python/databricks/bundles/catalogs/_models/encryption_settings.py @@ -1,30 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.catalogs._models.azure_encryption_settings import ( + AzureEncryptionSettings, + AzureEncryptionSettingsParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam -from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam -from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EncryptionSettings: """ Encryption Settings are used to carry metadata for securable encryption at rest. Currently used for catalogs, we can use the information supplied here to interact with a CMK. """ + azure_encryption_settings: VariableOrOptional[AzureEncryptionSettings] = None """ optional Azure settings - only required if an Azure CMK is used. @@ -41,12 +38,11 @@ class EncryptionSettings: """ @classmethod - def from_dict(cls, value: 'EncryptionSettingsDict') -> 'Self': + def from_dict(cls, value: "EncryptionSettingsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EncryptionSettingsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EncryptionSettingsDict": + return _transform_to_json_value(self) # type:ignore class EncryptionSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/_models/lifecycle.py b/python/databricks/bundles/catalogs/_models/lifecycle.py index f247f3c9202..697776a198e 100644 --- a/python/databricks/bundles/catalogs/_models/lifecycle.py +++ b/python/databricks/bundles/catalogs/_models/lifecycle.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam -from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam -from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam -from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -29,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/catalogs/_models/privilege.py b/python/databricks/bundles/catalogs/_models/privilege.py index 13d8ac1ea54..df45211a76a 100644 --- a/python/databricks/bundles/catalogs/_models/privilege.py +++ b/python/databricks/bundles/catalogs/_models/privilege.py @@ -1,19 +1,8 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam -from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam -from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam -from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam class Privilege(Enum): SELECT = "SELECT" @@ -69,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/catalogs/_models/privilege_assignment.py b/python/databricks/bundles/catalogs/_models/privilege_assignment.py index fbd6dc782cd..a85471c5931 100644 --- a/python/databricks/bundles/catalogs/_models/privilege_assignment.py +++ b/python/databricks/bundles/catalogs/_models/privilege_assignment.py @@ -1,24 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.catalogs._models.azure_encryption_settings import AzureEncryptionSettings, AzureEncryptionSettingsDict, AzureEncryptionSettingsParam -from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings, EncryptionSettingsDict, EncryptionSettingsParam -from databricks.bundles.catalogs._models.catalog import Catalog, CatalogDict, CatalogParam -from databricks.bundles.catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.catalogs._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -35,12 +28,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/__init__.py b/python/databricks/bundles/cluster_policies/__init__.py index 78e8a0ea7aa..531fa17821a 100644 --- a/python/databricks/bundles/cluster_policies/__init__.py +++ b/python/databricks/bundles/cluster_policies/__init__.py @@ -1,38 +1,68 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'ClusterPolicy', - 'ClusterPolicyDict', - 'ClusterPolicyParam', - 'ClusterPolicyPermission', - 'ClusterPolicyPermissionDict', - 'ClusterPolicyPermissionLevel', - 'ClusterPolicyPermissionLevelParam', - 'ClusterPolicyPermissionParam', - 'Library', - 'LibraryDict', - 'LibraryParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'MavenLibrary', - 'MavenLibraryDict', - 'MavenLibraryParam', - 'PythonPyPiLibrary', - 'PythonPyPiLibraryDict', - 'PythonPyPiLibraryParam', - 'RCranLibrary', - 'RCranLibraryDict', - 'RCranLibraryParam', + "ClusterPolicy", + "ClusterPolicyDict", + "ClusterPolicyParam", + "ClusterPolicyPermission", + "ClusterPolicyPermissionDict", + "ClusterPolicyPermissionLevel", + "ClusterPolicyPermissionLevelParam", + "ClusterPolicyPermissionParam", + "Library", + "LibraryDict", + "LibraryParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "MavenLibrary", + "MavenLibraryDict", + "MavenLibraryParam", + "PythonPyPiLibrary", + "PythonPyPiLibraryDict", + "PythonPyPiLibraryParam", + "RCranLibrary", + "RCranLibraryDict", + "RCranLibraryParam", ] -from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam -from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam - +from databricks.bundles.cluster_policies._models.cluster_policy import ( + ClusterPolicy, + ClusterPolicyDict, + ClusterPolicyParam, +) +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ( + ClusterPolicyPermission, + ClusterPolicyPermissionDict, + ClusterPolicyPermissionParam, +) +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ( + ClusterPolicyPermissionLevel, + ClusterPolicyPermissionLevelParam, +) +from databricks.bundles.cluster_policies._models.library import ( + Library, + LibraryDict, + LibraryParam, +) +from databricks.bundles.cluster_policies._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.cluster_policies._models.maven_library import ( + MavenLibrary, + MavenLibraryDict, + MavenLibraryParam, +) +from databricks.bundles.cluster_policies._models.python_py_pi_library import ( + PythonPyPiLibrary, + PythonPyPiLibraryDict, + PythonPyPiLibraryParam, +) +from databricks.bundles.cluster_policies._models.r_cran_library import ( + RCranLibrary, + RCranLibraryDict, + RCranLibraryParam, +) diff --git a/python/databricks/bundles/cluster_policies/_models/cluster_policy.py b/python/databricks/bundles/cluster_policies/_models/cluster_policy.py index e0214a61b78..d8033d95728 100644 --- a/python/databricks/bundles/cluster_policies/_models/cluster_policy.py +++ b/python/databricks/bundles/cluster_policies/_models/cluster_policy.py @@ -1,26 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, Any, TypedDict + +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ( + ClusterPolicyPermission, + ClusterPolicyPermissionParam, +) +from databricks.bundles.cluster_policies._models.library import Library, LibraryParam +from databricks.bundles.cluster_policies._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam -from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ClusterPolicy(Resource): """""" @@ -77,12 +77,11 @@ class ClusterPolicy(Resource): """ @classmethod - def from_dict(cls, value: 'ClusterPolicyDict') -> 'Self': + def from_dict(cls, value: "ClusterPolicyDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClusterPolicyDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClusterPolicyDict": + return _transform_to_json_value(self) # type:ignore class ClusterPolicyDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission.py b/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission.py index e1bad34183c..de9e59b93a4 100644 --- a/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission.py +++ b/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission.py @@ -1,26 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ( + ClusterPolicyPermissionLevel, + ClusterPolicyPermissionLevelParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam -from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ClusterPolicyPermission: """""" @@ -46,12 +40,11 @@ class ClusterPolicyPermission: """ @classmethod - def from_dict(cls, value: 'ClusterPolicyPermissionDict') -> 'Self': + def from_dict(cls, value: "ClusterPolicyPermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClusterPolicyPermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClusterPolicyPermissionDict": + return _transform_to_json_value(self) # type:ignore class ClusterPolicyPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission_level.py b/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission_level.py index 4dc021a4252..2773971dce1 100644 --- a/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission_level.py +++ b/python/databricks/bundles/cluster_policies/_models/cluster_policy_permission_level.py @@ -1,26 +1,15 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam -from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam class ClusterPolicyPermissionLevel(Enum): """ Permission level """ + CAN_USE = "CAN_USE" + ClusterPolicyPermissionLevelParam = Literal["CAN_USE"] | ClusterPolicyPermissionLevel diff --git a/python/databricks/bundles/cluster_policies/_models/library.py b/python/databricks/bundles/cluster_policies/_models/library.py index 21e9a466338..42781f4ccb5 100644 --- a/python/databricks/bundles/cluster_policies/_models/library.py +++ b/python/databricks/bundles/cluster_policies/_models/library.py @@ -1,26 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.cluster_policies._models.maven_library import ( + MavenLibrary, + MavenLibraryParam, +) +from databricks.bundles.cluster_policies._models.python_py_pi_library import ( + PythonPyPiLibrary, + PythonPyPiLibraryParam, +) +from databricks.bundles.cluster_policies._models.r_cran_library import ( + RCranLibrary, + RCranLibraryParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam -from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Library: """""" @@ -72,12 +74,11 @@ class Library: """ @classmethod - def from_dict(cls, value: 'LibraryDict') -> 'Self': + def from_dict(cls, value: "LibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LibraryDict": + return _transform_to_json_value(self) # type:ignore class LibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/lifecycle.py b/python/databricks/bundles/cluster_policies/_models/lifecycle.py index e01071d8106..697776a198e 100644 --- a/python/databricks/bundles/cluster_policies/_models/lifecycle.py +++ b/python/databricks/bundles/cluster_policies/_models/lifecycle.py @@ -1,26 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -31,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/maven_library.py b/python/databricks/bundles/cluster_policies/_models/maven_library.py index 7569afb3fdd..25be1599c28 100644 --- a/python/databricks/bundles/cluster_policies/_models/maven_library.py +++ b/python/databricks/bundles/cluster_policies/_models/maven_library.py @@ -1,26 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam -from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MavenLibrary: """""" @@ -45,12 +39,11 @@ class MavenLibrary: """ @classmethod - def from_dict(cls, value: 'MavenLibraryDict') -> 'Self': + def from_dict(cls, value: "MavenLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MavenLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MavenLibraryDict": + return _transform_to_json_value(self) # type:ignore class MavenLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/python_py_pi_library.py b/python/databricks/bundles/cluster_policies/_models/python_py_pi_library.py index b4d26b3fa5b..037c2029f50 100644 --- a/python/databricks/bundles/cluster_policies/_models/python_py_pi_library.py +++ b/python/databricks/bundles/cluster_policies/_models/python_py_pi_library.py @@ -1,26 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.cluster_policies._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam -from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PythonPyPiLibrary: """""" @@ -38,12 +28,11 @@ class PythonPyPiLibrary: """ @classmethod - def from_dict(cls, value: 'PythonPyPiLibraryDict') -> 'Self': + def from_dict(cls, value: "PythonPyPiLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PythonPyPiLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PythonPyPiLibraryDict": + return _transform_to_json_value(self) # type:ignore class PythonPyPiLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/cluster_policies/_models/r_cran_library.py b/python/databricks/bundles/cluster_policies/_models/r_cran_library.py index 9d91505595b..0bd2c4d5296 100644 --- a/python/databricks/bundles/cluster_policies/_models/r_cran_library.py +++ b/python/databricks/bundles/cluster_policies/_models/r_cran_library.py @@ -1,26 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.cluster_policies._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.cluster_policies._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.cluster_policies._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyDict, ClusterPolicyParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission, ClusterPolicyPermissionDict, ClusterPolicyPermissionParam -from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel, ClusterPolicyPermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RCranLibrary: """""" @@ -36,12 +26,11 @@ class RCranLibrary: """ @classmethod - def from_dict(cls, value: 'RCranLibraryDict') -> 'Self': + def from_dict(cls, value: "RCranLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RCranLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RCranLibraryDict": + return _transform_to_json_value(self) # type:ignore class RCranLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/__init__.py b/python/databricks/bundles/clusters/__init__.py index 4f19f68cb15..0ccecb208b0 100644 --- a/python/databricks/bundles/clusters/__init__.py +++ b/python/databricks/bundles/clusters/__init__.py @@ -1,150 +1,271 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Adlsgen2Info', - 'Adlsgen2InfoDict', - 'Adlsgen2InfoParam', - 'AutoScale', - 'AutoScaleDict', - 'AutoScaleParam', - 'AwsAttributes', - 'AwsAttributesDict', - 'AwsAttributesParam', - 'AwsAvailability', - 'AwsAvailabilityParam', - 'AzureAttributes', - 'AzureAttributesDict', - 'AzureAttributesParam', - 'AzureAvailability', - 'AzureAvailabilityParam', - 'ClientsTypes', - 'ClientsTypesDict', - 'ClientsTypesParam', - 'Cluster', - 'ClusterDict', - 'ClusterLogConf', - 'ClusterLogConfDict', - 'ClusterLogConfParam', - 'ClusterParam', - 'ClusterPermission', - 'ClusterPermissionDict', - 'ClusterPermissionLevel', - 'ClusterPermissionLevelParam', - 'ClusterPermissionParam', - 'ConfidentialComputeType', - 'ConfidentialComputeTypeParam', - 'DataSecurityMode', - 'DataSecurityModeParam', - 'DbfsStorageInfo', - 'DbfsStorageInfoDict', - 'DbfsStorageInfoParam', - 'DependencyMode', - 'DependencyModeParam', - 'DockerBasicAuth', - 'DockerBasicAuthDict', - 'DockerBasicAuthParam', - 'DockerImage', - 'DockerImageDict', - 'DockerImageParam', - 'EbsVolumeType', - 'EbsVolumeTypeParam', - 'GcpAttributes', - 'GcpAttributesDict', - 'GcpAttributesParam', - 'GcpAvailability', - 'GcpAvailabilityParam', - 'GcsStorageInfo', - 'GcsStorageInfoDict', - 'GcsStorageInfoParam', - 'InitScriptInfo', - 'InitScriptInfoDict', - 'InitScriptInfoParam', - 'Kind', - 'KindParam', - 'Library', - 'LibraryDict', - 'LibraryParam', - 'LifecycleWithStarted', - 'LifecycleWithStartedDict', - 'LifecycleWithStartedParam', - 'LocalFileInfo', - 'LocalFileInfoDict', - 'LocalFileInfoParam', - 'LogAnalyticsInfo', - 'LogAnalyticsInfoDict', - 'LogAnalyticsInfoParam', - 'MavenLibrary', - 'MavenLibraryDict', - 'MavenLibraryParam', - 'NodeTypeFlexibility', - 'NodeTypeFlexibilityDict', - 'NodeTypeFlexibilityParam', - 'PythonPyPiLibrary', - 'PythonPyPiLibraryDict', - 'PythonPyPiLibraryParam', - 'RCranLibrary', - 'RCranLibraryDict', - 'RCranLibraryParam', - 'RuntimeEngine', - 'RuntimeEngineParam', - 'S3StorageInfo', - 'S3StorageInfoDict', - 'S3StorageInfoParam', - 'VolumesStorageInfo', - 'VolumesStorageInfoDict', - 'VolumesStorageInfoParam', - 'WorkloadType', - 'WorkloadTypeDict', - 'WorkloadTypeParam', - 'WorkspaceStorageInfo', - 'WorkspaceStorageInfoDict', - 'WorkspaceStorageInfoParam', + "Adlsgen2Info", + "Adlsgen2InfoDict", + "Adlsgen2InfoParam", + "AutoScale", + "AutoScaleDict", + "AutoScaleParam", + "AwsAttributes", + "AwsAttributesDict", + "AwsAttributesParam", + "AwsAvailability", + "AwsAvailabilityParam", + "AzureAttributes", + "AzureAttributesDict", + "AzureAttributesParam", + "AzureAvailability", + "AzureAvailabilityParam", + "ClientsTypes", + "ClientsTypesDict", + "ClientsTypesParam", + "Cluster", + "ClusterDict", + "ClusterLogConf", + "ClusterLogConfDict", + "ClusterLogConfParam", + "ClusterParam", + "ClusterPermission", + "ClusterPermissionDict", + "ClusterPermissionLevel", + "ClusterPermissionLevelParam", + "ClusterPermissionParam", + "ConfidentialComputeType", + "ConfidentialComputeTypeParam", + "DataSecurityMode", + "DataSecurityModeParam", + "DbfsStorageInfo", + "DbfsStorageInfoDict", + "DbfsStorageInfoParam", + "DependencyMode", + "DependencyModeParam", + "DockerBasicAuth", + "DockerBasicAuthDict", + "DockerBasicAuthParam", + "DockerImage", + "DockerImageDict", + "DockerImageParam", + "EbsVolumeType", + "EbsVolumeTypeParam", + "GcpAttributes", + "GcpAttributesDict", + "GcpAttributesParam", + "GcpAvailability", + "GcpAvailabilityParam", + "GcsStorageInfo", + "GcsStorageInfoDict", + "GcsStorageInfoParam", + "InitScriptInfo", + "InitScriptInfoDict", + "InitScriptInfoParam", + "Kind", + "KindParam", + "Library", + "LibraryDict", + "LibraryParam", + "LifecycleWithStarted", + "LifecycleWithStartedDict", + "LifecycleWithStartedParam", + "LocalFileInfo", + "LocalFileInfoDict", + "LocalFileInfoParam", + "LogAnalyticsInfo", + "LogAnalyticsInfoDict", + "LogAnalyticsInfoParam", + "MavenLibrary", + "MavenLibraryDict", + "MavenLibraryParam", + "NodeTypeFlexibility", + "NodeTypeFlexibilityDict", + "NodeTypeFlexibilityParam", + "PythonPyPiLibrary", + "PythonPyPiLibraryDict", + "PythonPyPiLibraryParam", + "RCranLibrary", + "RCranLibraryDict", + "RCranLibraryParam", + "RuntimeEngine", + "RuntimeEngineParam", + "S3StorageInfo", + "S3StorageInfoDict", + "S3StorageInfoParam", + "VolumesStorageInfo", + "VolumesStorageInfoDict", + "VolumesStorageInfoParam", + "WorkloadType", + "WorkloadTypeDict", + "WorkloadTypeParam", + "WorkspaceStorageInfo", + "WorkspaceStorageInfoDict", + "WorkspaceStorageInfoParam", ] -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - +from databricks.bundles.clusters._models.adlsgen2_info import ( + Adlsgen2Info, + Adlsgen2InfoDict, + Adlsgen2InfoParam, +) +from databricks.bundles.clusters._models.auto_scale import ( + AutoScale, + AutoScaleDict, + AutoScaleParam, +) +from databricks.bundles.clusters._models.aws_attributes import ( + AwsAttributes, + AwsAttributesDict, + AwsAttributesParam, +) +from databricks.bundles.clusters._models.aws_availability import ( + AwsAvailability, + AwsAvailabilityParam, +) +from databricks.bundles.clusters._models.azure_attributes import ( + AzureAttributes, + AzureAttributesDict, + AzureAttributesParam, +) +from databricks.bundles.clusters._models.azure_availability import ( + AzureAvailability, + AzureAvailabilityParam, +) +from databricks.bundles.clusters._models.clients_types import ( + ClientsTypes, + ClientsTypesDict, + ClientsTypesParam, +) +from databricks.bundles.clusters._models.cluster import ( + Cluster, + ClusterDict, + ClusterParam, +) +from databricks.bundles.clusters._models.cluster_log_conf import ( + ClusterLogConf, + ClusterLogConfDict, + ClusterLogConfParam, +) +from databricks.bundles.clusters._models.cluster_permission import ( + ClusterPermission, + ClusterPermissionDict, + ClusterPermissionParam, +) +from databricks.bundles.clusters._models.cluster_permission_level import ( + ClusterPermissionLevel, + ClusterPermissionLevelParam, +) +from databricks.bundles.clusters._models.confidential_compute_type import ( + ConfidentialComputeType, + ConfidentialComputeTypeParam, +) +from databricks.bundles.clusters._models.data_security_mode import ( + DataSecurityMode, + DataSecurityModeParam, +) +from databricks.bundles.clusters._models.dbfs_storage_info import ( + DbfsStorageInfo, + DbfsStorageInfoDict, + DbfsStorageInfoParam, +) +from databricks.bundles.clusters._models.dependency_mode import ( + DependencyMode, + DependencyModeParam, +) +from databricks.bundles.clusters._models.docker_basic_auth import ( + DockerBasicAuth, + DockerBasicAuthDict, + DockerBasicAuthParam, +) +from databricks.bundles.clusters._models.docker_image import ( + DockerImage, + DockerImageDict, + DockerImageParam, +) +from databricks.bundles.clusters._models.ebs_volume_type import ( + EbsVolumeType, + EbsVolumeTypeParam, +) +from databricks.bundles.clusters._models.gcp_attributes import ( + GcpAttributes, + GcpAttributesDict, + GcpAttributesParam, +) +from databricks.bundles.clusters._models.gcp_availability import ( + GcpAvailability, + GcpAvailabilityParam, +) +from databricks.bundles.clusters._models.gcs_storage_info import ( + GcsStorageInfo, + GcsStorageInfoDict, + GcsStorageInfoParam, +) +from databricks.bundles.clusters._models.init_script_info import ( + InitScriptInfo, + InitScriptInfoDict, + InitScriptInfoParam, +) from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.clusters._models.library import ( + Library, + LibraryDict, + LibraryParam, +) +from databricks.bundles.clusters._models.lifecycle_with_started import ( + LifecycleWithStarted, + LifecycleWithStartedDict, + LifecycleWithStartedParam, +) +from databricks.bundles.clusters._models.local_file_info import ( + LocalFileInfo, + LocalFileInfoDict, + LocalFileInfoParam, +) +from databricks.bundles.clusters._models.log_analytics_info import ( + LogAnalyticsInfo, + LogAnalyticsInfoDict, + LogAnalyticsInfoParam, +) +from databricks.bundles.clusters._models.maven_library import ( + MavenLibrary, + MavenLibraryDict, + MavenLibraryParam, +) +from databricks.bundles.clusters._models.node_type_flexibility import ( + NodeTypeFlexibility, + NodeTypeFlexibilityDict, + NodeTypeFlexibilityParam, +) +from databricks.bundles.clusters._models.python_py_pi_library import ( + PythonPyPiLibrary, + PythonPyPiLibraryDict, + PythonPyPiLibraryParam, +) +from databricks.bundles.clusters._models.r_cran_library import ( + RCranLibrary, + RCranLibraryDict, + RCranLibraryParam, +) +from databricks.bundles.clusters._models.runtime_engine import ( + RuntimeEngine, + RuntimeEngineParam, +) +from databricks.bundles.clusters._models.s3_storage_info import ( + S3StorageInfo, + S3StorageInfoDict, + S3StorageInfoParam, +) +from databricks.bundles.clusters._models.volumes_storage_info import ( + VolumesStorageInfo, + VolumesStorageInfoDict, + VolumesStorageInfoParam, +) +from databricks.bundles.clusters._models.workload_type import ( + WorkloadType, + WorkloadTypeDict, + WorkloadTypeParam, +) +from databricks.bundles.clusters._models.workspace_storage_info import ( + WorkspaceStorageInfo, + WorkspaceStorageInfoDict, + WorkspaceStorageInfoParam, +) diff --git a/python/databricks/bundles/clusters/_models/adlsgen2_info.py b/python/databricks/bundles/clusters/_models/adlsgen2_info.py index 0f943c35844..87c883adeb9 100644 --- a/python/databricks/bundles/clusters/_models/adlsgen2_info.py +++ b/python/databricks/bundles/clusters/_models/adlsgen2_info.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Adlsgen2Info: """ A storage location in Adls Gen2 """ + destination: VariableOr[str] """ abfss destination, e.g. `abfss://@.dfs.core.windows.net/`. """ @classmethod - def from_dict(cls, value: 'Adlsgen2InfoDict') -> 'Self': + def from_dict(cls, value: "Adlsgen2InfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'Adlsgen2InfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "Adlsgen2InfoDict": + return _transform_to_json_value(self) # type:ignore class Adlsgen2InfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/auto_scale.py b/python/databricks/bundles/clusters/_models/auto_scale.py index 1d436e99351..10ca50c26d7 100644 --- a/python/databricks/bundles/clusters/_models/auto_scale.py +++ b/python/databricks/bundles/clusters/_models/auto_scale.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AutoScale: """""" @@ -75,12 +28,11 @@ class AutoScale: """ @classmethod - def from_dict(cls, value: 'AutoScaleDict') -> 'Self': + def from_dict(cls, value: "AutoScaleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AutoScaleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AutoScaleDict": + return _transform_to_json_value(self) # type:ignore class AutoScaleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/aws_attributes.py b/python/databricks/bundles/clusters/_models/aws_attributes.py index bf9e194de16..2ff19359882 100644 --- a/python/databricks/bundles/clusters/_models/aws_attributes.py +++ b/python/databricks/bundles/clusters/_models/aws_attributes.py @@ -1,68 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.clusters._models.aws_availability import ( + AwsAvailability, + AwsAvailabilityParam, +) +from databricks.bundles.clusters._models.ebs_volume_type import ( + EbsVolumeType, + EbsVolumeTypeParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AwsAttributes: """ Attributes set during cluster creation which are related to Amazon Web Services. """ + availability: VariableOrOptional[AwsAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -161,12 +123,11 @@ class AwsAttributes: """ @classmethod - def from_dict(cls, value: 'AwsAttributesDict') -> 'Self': + def from_dict(cls, value: "AwsAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AwsAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AwsAttributesDict": + return _transform_to_json_value(self) # type:ignore class AwsAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/aws_availability.py b/python/databricks/bundles/clusters/_models/aws_availability.py index c9b2dffc181..4c3810f067e 100644 --- a/python/databricks/bundles/clusters/_models/aws_availability.py +++ b/python/databricks/bundles/clusters/_models/aws_availability.py @@ -1,67 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam +from typing import Literal class AwsAvailability(Enum): """ Availability type used for all subsequent nodes past the `first_on_demand` ones. - + Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ + SPOT = "SPOT" ON_DEMAND = "ON_DEMAND" SPOT_WITH_FALLBACK = "SPOT_WITH_FALLBACK" -AwsAvailabilityParam = Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability + +AwsAvailabilityParam = ( + Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability +) diff --git a/python/databricks/bundles/clusters/_models/azure_attributes.py b/python/databricks/bundles/clusters/_models/azure_attributes.py index 0c57c96417a..f5c5cbee36d 100644 --- a/python/databricks/bundles/clusters/_models/azure_attributes.py +++ b/python/databricks/bundles/clusters/_models/azure_attributes.py @@ -1,68 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.clusters._models.azure_availability import ( + AzureAvailability, + AzureAvailabilityParam, +) +from databricks.bundles.clusters._models.log_analytics_info import ( + LogAnalyticsInfo, + LogAnalyticsInfoParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AzureAttributes: """ Attributes set during cluster creation which are related to Microsoft Azure. """ + availability: VariableOrOptional[AzureAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -110,12 +72,11 @@ class AzureAttributes: """ @classmethod - def from_dict(cls, value: 'AzureAttributesDict') -> 'Self': + def from_dict(cls, value: "AzureAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AzureAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AzureAttributesDict": + return _transform_to_json_value(self) # type:ignore class AzureAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/azure_availability.py b/python/databricks/bundles/clusters/_models/azure_availability.py index a61c7c57906..a03cb1fdddf 100644 --- a/python/databricks/bundles/clusters/_models/azure_availability.py +++ b/python/databricks/bundles/clusters/_models/azure_availability.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam +from typing import Literal class AzureAvailability(Enum): @@ -59,8 +9,13 @@ class AzureAvailability(Enum): Availability type used for all subsequent nodes past the `first_on_demand` ones. Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ + SPOT_AZURE = "SPOT_AZURE" ON_DEMAND_AZURE = "ON_DEMAND_AZURE" SPOT_WITH_FALLBACK_AZURE = "SPOT_WITH_FALLBACK_AZURE" -AzureAvailabilityParam = Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] | AzureAvailability + +AzureAvailabilityParam = ( + Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] + | AzureAvailability +) diff --git a/python/databricks/bundles/clusters/_models/clients_types.py b/python/databricks/bundles/clusters/_models/clients_types.py index af4a62b9842..87f92446ba5 100644 --- a/python/databricks/bundles/clusters/_models/clients_types.py +++ b/python/databricks/bundles/clusters/_models/clients_types.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ClientsTypes: """""" @@ -73,12 +26,11 @@ class ClientsTypes: """ @classmethod - def from_dict(cls, value: 'ClientsTypesDict') -> 'Self': + def from_dict(cls, value: "ClientsTypesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClientsTypesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClientsTypesDict": + return _transform_to_json_value(self) # type:ignore class ClientsTypesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/cluster.py b/python/databricks/bundles/clusters/_models/cluster.py index 20546fbe5b4..cc3e9fcde04 100644 --- a/python/databricks/bundles/clusters/_models/cluster.py +++ b/python/databricks/bundles/clusters/_models/cluster.py @@ -1,68 +1,85 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.clusters._models.auto_scale import ( + AutoScale, + AutoScaleParam, +) +from databricks.bundles.clusters._models.aws_attributes import ( + AwsAttributes, + AwsAttributesParam, +) +from databricks.bundles.clusters._models.azure_attributes import ( + AzureAttributes, + AzureAttributesParam, +) +from databricks.bundles.clusters._models.cluster_log_conf import ( + ClusterLogConf, + ClusterLogConfParam, +) +from databricks.bundles.clusters._models.cluster_permission import ( + ClusterPermission, + ClusterPermissionParam, +) +from databricks.bundles.clusters._models.data_security_mode import ( + DataSecurityMode, + DataSecurityModeParam, +) +from databricks.bundles.clusters._models.dependency_mode import ( + DependencyMode, + DependencyModeParam, +) +from databricks.bundles.clusters._models.docker_image import ( + DockerImage, + DockerImageParam, +) +from databricks.bundles.clusters._models.gcp_attributes import ( + GcpAttributes, + GcpAttributesParam, +) +from databricks.bundles.clusters._models.init_script_info import ( + InitScriptInfo, + InitScriptInfoParam, +) +from databricks.bundles.clusters._models.kind import Kind, KindParam +from databricks.bundles.clusters._models.library import Library, LibraryParam +from databricks.bundles.clusters._models.lifecycle_with_started import ( + LifecycleWithStarted, + LifecycleWithStartedParam, +) +from databricks.bundles.clusters._models.node_type_flexibility import ( + NodeTypeFlexibility, + NodeTypeFlexibilityParam, +) +from databricks.bundles.clusters._models.runtime_engine import ( + RuntimeEngine, + RuntimeEngineParam, +) +from databricks.bundles.clusters._models.workload_type import ( + WorkloadType, + WorkloadTypeParam, +) from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import ( + VariableOrDict, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Cluster(Resource): """ Contains a snapshot of the latest user specified settings that were used to create/edit the cluster. """ + apply_policy_default_values: VariableOrOptional[bool] = None """ When set to true, fixed and default values from the policy will be used for fields that are omitted. When set to false, only fixed values from the policy will be applied. @@ -343,12 +360,11 @@ class Cluster(Resource): """ @classmethod - def from_dict(cls, value: 'ClusterDict') -> 'Self': + def from_dict(cls, value: "ClusterDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClusterDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClusterDict": + return _transform_to_json_value(self) # type:ignore class ClusterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/cluster_log_conf.py b/python/databricks/bundles/clusters/_models/cluster_log_conf.py index 18926bd14dc..c087a2ad463 100644 --- a/python/databricks/bundles/clusters/_models/cluster_log_conf.py +++ b/python/databricks/bundles/clusters/_models/cluster_log_conf.py @@ -1,68 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.clusters._models.dbfs_storage_info import ( + DbfsStorageInfo, + DbfsStorageInfoParam, +) +from databricks.bundles.clusters._models.s3_storage_info import ( + S3StorageInfo, + S3StorageInfoParam, +) +from databricks.bundles.clusters._models.volumes_storage_info import ( + VolumesStorageInfo, + VolumesStorageInfoParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ClusterLogConf: """ Cluster log delivery config """ + dbfs: VariableOrOptional[DbfsStorageInfo] = None """ destination needs to be provided. e.g. @@ -84,12 +50,11 @@ class ClusterLogConf: """ @classmethod - def from_dict(cls, value: 'ClusterLogConfDict') -> 'Self': + def from_dict(cls, value: "ClusterLogConfDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClusterLogConfDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClusterLogConfDict": + return _transform_to_json_value(self) # type:ignore class ClusterLogConfDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/cluster_permission.py b/python/databricks/bundles/clusters/_models/cluster_permission.py index c40850089e0..537e136780a 100644 --- a/python/databricks/bundles/clusters/_models/cluster_permission.py +++ b/python/databricks/bundles/clusters/_models/cluster_permission.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.clusters._models.cluster_permission_level import ( + ClusterPermissionLevel, + ClusterPermissionLevelParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ClusterPermission: """""" @@ -83,12 +40,11 @@ class ClusterPermission: """ @classmethod - def from_dict(cls, value: 'ClusterPermissionDict') -> 'Self': + def from_dict(cls, value: "ClusterPermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClusterPermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClusterPermissionDict": + return _transform_to_json_value(self) # type:ignore class ClusterPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/cluster_permission_level.py b/python/databricks/bundles/clusters/_models/cluster_permission_level.py index 5f9266fab7e..bb8b1347d92 100644 --- a/python/databricks/bundles/clusters/_models/cluster_permission_level.py +++ b/python/databricks/bundles/clusters/_models/cluster_permission_level.py @@ -1,65 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam +from typing import Literal class ClusterPermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" -ClusterPermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO"] | ClusterPermissionLevel + +ClusterPermissionLevelParam = ( + Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO"] | ClusterPermissionLevel +) diff --git a/python/databricks/bundles/clusters/_models/confidential_compute_type.py b/python/databricks/bundles/clusters/_models/confidential_compute_type.py index beed47d71d9..e881295bd47 100644 --- a/python/databricks/bundles/clusters/_models/confidential_compute_type.py +++ b/python/databricks/bundles/clusters/_models/confidential_compute_type.py @@ -1,69 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam +from typing import Literal class ConfidentialComputeType(Enum): """ :meta private: [EXPERIMENTAL] - + Confidential computing technology for GCP instances. Aligns with gcloud's --confidential-compute-type flag and the REST API's confidentialInstanceConfig.confidentialInstanceType field. See: https://cloud.google.com/confidential-computing/confidential-vm/docs/create-a-confidential-vm-instance """ + CONFIDENTIAL_COMPUTE_TYPE_NONE = "CONFIDENTIAL_COMPUTE_TYPE_NONE" SEV_SNP = "SEV_SNP" -ConfidentialComputeTypeParam = Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType + +ConfidentialComputeTypeParam = ( + Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType +) diff --git a/python/databricks/bundles/clusters/_models/data_security_mode.py b/python/databricks/bundles/clusters/_models/data_security_mode.py index 88976123873..ae1aafcf085 100644 --- a/python/databricks/bundles/clusters/_models/data_security_mode.py +++ b/python/databricks/bundles/clusters/_models/data_security_mode.py @@ -1,81 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam +from typing import Literal class DataSecurityMode(Enum): """ Data security mode decides what data governance model to use when accessing data from a cluster. - + * `DATA_SECURITY_MODE_AUTO`: Databricks will choose the most appropriate access mode depending on your compute configuration. * `DATA_SECURITY_MODE_STANDARD`: A secure cluster that can be shared by multiple users. Cluster users are fully isolated so that they cannot see each other’s data and credentials. Most data governance features are supported in this mode. But programming languages and cluster features might be limited. * `DATA_SECURITY_MODE_DEDICATED`: A secure cluster that can only be exclusively used by a single user specified in `single_user_name`. Most programming languages, cluster features and data governance features are available in this mode. - + The following modes are legacy aliases for the above modes: - + * `USER_ISOLATION`: Legacy alias for `DATA_SECURITY_MODE_STANDARD`. * `SINGLE_USER`: Legacy alias for `DATA_SECURITY_MODE_DEDICATED`. - + The following modes are deprecated starting with Databricks Runtime 15.0 and will be removed for future Databricks Runtime versions: - + * `LEGACY_TABLE_ACL`: This mode is for users migrating from legacy Table ACL clusters. * `LEGACY_PASSTHROUGH`: This mode is for users migrating from legacy Passthrough on high concurrency clusters. * `LEGACY_SINGLE_USER`: This mode is for users migrating from legacy Passthrough on standard clusters. * `LEGACY_SINGLE_USER_STANDARD`: This mode provides a way that doesn’t have UC nor passthrough enabled. """ + NONE = "NONE" SINGLE_USER = "SINGLE_USER" USER_ISOLATION = "USER_ISOLATION" @@ -87,4 +38,19 @@ class DataSecurityMode(Enum): DATA_SECURITY_MODE_DEDICATED = "DATA_SECURITY_MODE_DEDICATED" DATA_SECURITY_MODE_AUTO = "DATA_SECURITY_MODE_AUTO" -DataSecurityModeParam = Literal["NONE", "SINGLE_USER", "USER_ISOLATION", "LEGACY_TABLE_ACL", "LEGACY_PASSTHROUGH", "LEGACY_SINGLE_USER", "LEGACY_SINGLE_USER_STANDARD", "DATA_SECURITY_MODE_STANDARD", "DATA_SECURITY_MODE_DEDICATED", "DATA_SECURITY_MODE_AUTO"] | DataSecurityMode + +DataSecurityModeParam = ( + Literal[ + "NONE", + "SINGLE_USER", + "USER_ISOLATION", + "LEGACY_TABLE_ACL", + "LEGACY_PASSTHROUGH", + "LEGACY_SINGLE_USER", + "LEGACY_SINGLE_USER_STANDARD", + "DATA_SECURITY_MODE_STANDARD", + "DATA_SECURITY_MODE_DEDICATED", + "DATA_SECURITY_MODE_AUTO", + ] + | DataSecurityMode +) diff --git a/python/databricks/bundles/clusters/_models/dbfs_storage_info.py b/python/databricks/bundles/clusters/_models/dbfs_storage_info.py index 8d5bf7ed0bb..29744acf26a 100644 --- a/python/databricks/bundles/clusters/_models/dbfs_storage_info.py +++ b/python/databricks/bundles/clusters/_models/dbfs_storage_info.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DbfsStorageInfo: """ A storage location in DBFS """ + destination: VariableOr[str] """ dbfs destination, e.g. `dbfs:/my/path` """ @classmethod - def from_dict(cls, value: 'DbfsStorageInfoDict') -> 'Self': + def from_dict(cls, value: "DbfsStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DbfsStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DbfsStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class DbfsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/dependency_mode.py b/python/databricks/bundles/clusters/_models/dependency_mode.py index c54af29c916..1fdc22cfa4c 100644 --- a/python/databricks/bundles/clusters/_models/dependency_mode.py +++ b/python/databricks/bundles/clusters/_models/dependency_mode.py @@ -1,69 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam +from typing import Literal class DependencyMode(Enum): """ Controls dependency configuration for the cluster. - + * `DEPENDENCY_MODE_AUTO`: Databricks will choose the most appropriate dependency mode based on your compute configuration. * `DEPENDENCY_MODE_ENVIRONMENTS`: Enables a unified dependency management experience across classic and serverless, resulting in increased stability and performance. Supported only on DBR 19+ in Standard access mode. * `DEPENDENCY_MODE_CLUSTER_LIBRARIES`: Legacy mode: dependencies come from cluster libraries and init scripts. """ + DEPENDENCY_MODE_ENVIRONMENTS = "DEPENDENCY_MODE_ENVIRONMENTS" DEPENDENCY_MODE_CLUSTER_LIBRARIES = "DEPENDENCY_MODE_CLUSTER_LIBRARIES" DEPENDENCY_MODE_AUTO = "DEPENDENCY_MODE_AUTO" -DependencyModeParam = Literal["DEPENDENCY_MODE_ENVIRONMENTS", "DEPENDENCY_MODE_CLUSTER_LIBRARIES", "DEPENDENCY_MODE_AUTO"] | DependencyMode + +DependencyModeParam = ( + Literal[ + "DEPENDENCY_MODE_ENVIRONMENTS", + "DEPENDENCY_MODE_CLUSTER_LIBRARIES", + "DEPENDENCY_MODE_AUTO", + ] + | DependencyMode +) diff --git a/python/databricks/bundles/clusters/_models/docker_basic_auth.py b/python/databricks/bundles/clusters/_models/docker_basic_auth.py index b235d24953b..552ea90a83b 100644 --- a/python/databricks/bundles/clusters/_models/docker_basic_auth.py +++ b/python/databricks/bundles/clusters/_models/docker_basic_auth.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DockerBasicAuth: """""" @@ -73,12 +26,11 @@ class DockerBasicAuth: """ @classmethod - def from_dict(cls, value: 'DockerBasicAuthDict') -> 'Self': + def from_dict(cls, value: "DockerBasicAuthDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DockerBasicAuthDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DockerBasicAuthDict": + return _transform_to_json_value(self) # type:ignore class DockerBasicAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/docker_image.py b/python/databricks/bundles/clusters/_models/docker_image.py index 1e65b80f881..f7c47633e07 100644 --- a/python/databricks/bundles/clusters/_models/docker_image.py +++ b/python/databricks/bundles/clusters/_models/docker_image.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.clusters._models.docker_basic_auth import ( + DockerBasicAuth, + DockerBasicAuthParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DockerImage: """""" @@ -73,12 +30,11 @@ class DockerImage: """ @classmethod - def from_dict(cls, value: 'DockerImageDict') -> 'Self': + def from_dict(cls, value: "DockerImageDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DockerImageDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DockerImageDict": + return _transform_to_json_value(self) # type:ignore class DockerImageDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/ebs_volume_type.py b/python/databricks/bundles/clusters/_models/ebs_volume_type.py index 55599e421cd..d8f3ef13f89 100644 --- a/python/databricks/bundles/clusters/_models/ebs_volume_type.py +++ b/python/databricks/bundles/clusters/_models/ebs_volume_type.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam +from typing import Literal class EbsVolumeType(Enum): @@ -59,7 +9,11 @@ class EbsVolumeType(Enum): All EBS volume types that Databricks supports. See https://aws.amazon.com/ebs/details/ for details. """ + GENERAL_PURPOSE_SSD = "GENERAL_PURPOSE_SSD" THROUGHPUT_OPTIMIZED_HDD = "THROUGHPUT_OPTIMIZED_HDD" -EbsVolumeTypeParam = Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType + +EbsVolumeTypeParam = ( + Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType +) diff --git a/python/databricks/bundles/clusters/_models/gcp_attributes.py b/python/databricks/bundles/clusters/_models/gcp_attributes.py index ba74a57205b..e31188d405e 100644 --- a/python/databricks/bundles/clusters/_models/gcp_attributes.py +++ b/python/databricks/bundles/clusters/_models/gcp_attributes.py @@ -1,68 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.clusters._models.confidential_compute_type import ( + ConfidentialComputeType, + ConfidentialComputeTypeParam, +) +from databricks.bundles.clusters._models.gcp_availability import ( + GcpAvailability, + GcpAvailabilityParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GcpAttributes: """ Attributes set during cluster creation which are related to GCP. """ + availability: VariableOrOptional[GcpAvailability] = None """ This field determines whether the spark executors will be scheduled to run on preemptible @@ -129,12 +91,11 @@ class GcpAttributes: """ @classmethod - def from_dict(cls, value: 'GcpAttributesDict') -> 'Self': + def from_dict(cls, value: "GcpAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GcpAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GcpAttributesDict": + return _transform_to_json_value(self) # type:ignore class GcpAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/gcp_availability.py b/python/databricks/bundles/clusters/_models/gcp_availability.py index 2893d6d8e6a..0d391b87fe3 100644 --- a/python/databricks/bundles/clusters/_models/gcp_availability.py +++ b/python/databricks/bundles/clusters/_models/gcp_availability.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam +from typing import Literal class GcpAvailability(Enum): @@ -59,8 +9,13 @@ class GcpAvailability(Enum): This field determines whether the instance pool will contain preemptible VMs, on-demand VMs, or preemptible VMs with a fallback to on-demand VMs if the former is unavailable. """ + PREEMPTIBLE_GCP = "PREEMPTIBLE_GCP" ON_DEMAND_GCP = "ON_DEMAND_GCP" PREEMPTIBLE_WITH_FALLBACK_GCP = "PREEMPTIBLE_WITH_FALLBACK_GCP" -GcpAvailabilityParam = Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] | GcpAvailability + +GcpAvailabilityParam = ( + Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] + | GcpAvailability +) diff --git a/python/databricks/bundles/clusters/_models/gcs_storage_info.py b/python/databricks/bundles/clusters/_models/gcs_storage_info.py index d8b53dd6310..e8f1059373b 100644 --- a/python/databricks/bundles/clusters/_models/gcs_storage_info.py +++ b/python/databricks/bundles/clusters/_models/gcs_storage_info.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GcsStorageInfo: """ A storage location in Google Cloud Platform's GCS """ + destination: VariableOr[str] """ GCS destination/URI, e.g. `gs://my-bucket/some-prefix` """ @classmethod - def from_dict(cls, value: 'GcsStorageInfoDict') -> 'Self': + def from_dict(cls, value: "GcsStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GcsStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GcsStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class GcsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/init_script_info.py b/python/databricks/bundles/clusters/_models/init_script_info.py index 75c8fbd8436..6d78d16bf92 100644 --- a/python/databricks/bundles/clusters/_models/init_script_info.py +++ b/python/databricks/bundles/clusters/_models/init_script_info.py @@ -1,68 +1,50 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.clusters._models.adlsgen2_info import ( + Adlsgen2Info, + Adlsgen2InfoParam, +) +from databricks.bundles.clusters._models.dbfs_storage_info import ( + DbfsStorageInfo, + DbfsStorageInfoParam, +) +from databricks.bundles.clusters._models.gcs_storage_info import ( + GcsStorageInfo, + GcsStorageInfoParam, +) +from databricks.bundles.clusters._models.local_file_info import ( + LocalFileInfo, + LocalFileInfoParam, +) +from databricks.bundles.clusters._models.s3_storage_info import ( + S3StorageInfo, + S3StorageInfoParam, +) +from databricks.bundles.clusters._models.volumes_storage_info import ( + VolumesStorageInfo, + VolumesStorageInfoParam, +) +from databricks.bundles.clusters._models.workspace_storage_info import ( + WorkspaceStorageInfo, + WorkspaceStorageInfoParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InitScriptInfo: """ Config for an individual init script """ + abfss: VariableOrOptional[Adlsgen2Info] = None """ Contains the Azure Data Lake Storage destination path @@ -107,12 +89,11 @@ class InitScriptInfo: """ @classmethod - def from_dict(cls, value: 'InitScriptInfoDict') -> 'Self': + def from_dict(cls, value: "InitScriptInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InitScriptInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InitScriptInfoDict": + return _transform_to_json_value(self) # type:ignore class InitScriptInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/kind.py b/python/databricks/bundles/clusters/_models/kind.py index 32672d3d1e3..8614e25dd52 100644 --- a/python/databricks/bundles/clusters/_models/kind.py +++ b/python/databricks/bundles/clusters/_models/kind.py @@ -1,71 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam +from typing import Literal class Kind(Enum): """ The kind of compute described by this compute specification. - + Depending on `kind`, different validations and default values will be applied. - + Clusters with `kind = CLASSIC_PREVIEW` support the following fields, whereas clusters with no specified `kind` do not. * [is_single_node](/api/workspace/clusters/create#is_single_node) * [use_ml_runtime](/api/workspace/clusters/create#use_ml_runtime) - + By using the [simple form](https://docs.databricks.com/compute/simple-form.html), your clusters are automatically using `kind = CLASSIC_PREVIEW`. """ + CLASSIC_PREVIEW = "CLASSIC_PREVIEW" + KindParam = Literal["CLASSIC_PREVIEW"] | Kind diff --git a/python/databricks/bundles/clusters/_models/library.py b/python/databricks/bundles/clusters/_models/library.py index 0436500dfdb..8961f8fe744 100644 --- a/python/databricks/bundles/clusters/_models/library.py +++ b/python/databricks/bundles/clusters/_models/library.py @@ -1,63 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict + +from databricks.bundles.clusters._models.maven_library import ( + MavenLibrary, + MavenLibraryParam, +) +from databricks.bundles.clusters._models.python_py_pi_library import ( + PythonPyPiLibrary, + PythonPyPiLibraryParam, +) +from databricks.bundles.clusters._models.r_cran_library import ( + RCranLibrary, + RCranLibraryParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Library: """""" @@ -109,12 +74,11 @@ class Library: """ @classmethod - def from_dict(cls, value: 'LibraryDict') -> 'Self': + def from_dict(cls, value: "LibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LibraryDict": + return _transform_to_json_value(self) # type:ignore class LibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/lifecycle_with_started.py b/python/databricks/bundles/clusters/_models/lifecycle_with_started.py index 1c947497092..f9cb9b8d49c 100644 --- a/python/databricks/bundles/clusters/_models/lifecycle_with_started.py +++ b/python/databricks/bundles/clusters/_models/lifecycle_with_started.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LifecycleWithStarted: """""" @@ -73,12 +26,11 @@ class LifecycleWithStarted: """ @classmethod - def from_dict(cls, value: 'LifecycleWithStartedDict') -> 'Self': + def from_dict(cls, value: "LifecycleWithStartedDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleWithStartedDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleWithStartedDict": + return _transform_to_json_value(self) # type:ignore class LifecycleWithStartedDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/local_file_info.py b/python/databricks/bundles/clusters/_models/local_file_info.py index 0b52c0d10fa..875f2fe8353 100644 --- a/python/databricks/bundles/clusters/_models/local_file_info.py +++ b/python/databricks/bundles/clusters/_models/local_file_info.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LocalFileInfo: """""" @@ -68,12 +21,11 @@ class LocalFileInfo: """ @classmethod - def from_dict(cls, value: 'LocalFileInfoDict') -> 'Self': + def from_dict(cls, value: "LocalFileInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LocalFileInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LocalFileInfoDict": + return _transform_to_json_value(self) # type:ignore class LocalFileInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/log_analytics_info.py b/python/databricks/bundles/clusters/_models/log_analytics_info.py index ba78f2ca12c..67cdf85b2b9 100644 --- a/python/databricks/bundles/clusters/_models/log_analytics_info.py +++ b/python/databricks/bundles/clusters/_models/log_analytics_info.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LogAnalyticsInfo: """""" @@ -73,12 +26,11 @@ class LogAnalyticsInfo: """ @classmethod - def from_dict(cls, value: 'LogAnalyticsInfoDict') -> 'Self': + def from_dict(cls, value: "LogAnalyticsInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LogAnalyticsInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LogAnalyticsInfoDict": + return _transform_to_json_value(self) # type:ignore class LogAnalyticsInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/maven_library.py b/python/databricks/bundles/clusters/_models/maven_library.py index f5cad160966..25be1599c28 100644 --- a/python/databricks/bundles/clusters/_models/maven_library.py +++ b/python/databricks/bundles/clusters/_models/maven_library.py @@ -1,63 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MavenLibrary: """""" @@ -82,12 +39,11 @@ class MavenLibrary: """ @classmethod - def from_dict(cls, value: 'MavenLibraryDict') -> 'Self': + def from_dict(cls, value: "MavenLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MavenLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MavenLibraryDict": + return _transform_to_json_value(self) # type:ignore class MavenLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/node_type_flexibility.py b/python/databricks/bundles/clusters/_models/node_type_flexibility.py index 00e00238d4e..418ea9337bd 100644 --- a/python/databricks/bundles/clusters/_models/node_type_flexibility.py +++ b/python/databricks/bundles/clusters/_models/node_type_flexibility.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class NodeTypeFlexibility: """ Configuration for flexible node types, allowing fallback to alternate node types during cluster launch and upscale. """ + alternate_node_type_ids: VariableOrList[str] = field(default_factory=list) """ A list of node type IDs to use as fallbacks when the primary node type is unavailable. @@ -75,12 +29,11 @@ class NodeTypeFlexibility: """ @classmethod - def from_dict(cls, value: 'NodeTypeFlexibilityDict') -> 'Self': + def from_dict(cls, value: "NodeTypeFlexibilityDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'NodeTypeFlexibilityDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "NodeTypeFlexibilityDict": + return _transform_to_json_value(self) # type:ignore class NodeTypeFlexibilityDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/python_py_pi_library.py b/python/databricks/bundles/clusters/_models/python_py_pi_library.py index 94d29d699c6..037c2029f50 100644 --- a/python/databricks/bundles/clusters/_models/python_py_pi_library.py +++ b/python/databricks/bundles/clusters/_models/python_py_pi_library.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PythonPyPiLibrary: """""" @@ -75,12 +28,11 @@ class PythonPyPiLibrary: """ @classmethod - def from_dict(cls, value: 'PythonPyPiLibraryDict') -> 'Self': + def from_dict(cls, value: "PythonPyPiLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PythonPyPiLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PythonPyPiLibraryDict": + return _transform_to_json_value(self) # type:ignore class PythonPyPiLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/r_cran_library.py b/python/databricks/bundles/clusters/_models/r_cran_library.py index a12f8fd9db7..0bd2c4d5296 100644 --- a/python/databricks/bundles/clusters/_models/r_cran_library.py +++ b/python/databricks/bundles/clusters/_models/r_cran_library.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RCranLibrary: """""" @@ -73,12 +26,11 @@ class RCranLibrary: """ @classmethod - def from_dict(cls, value: 'RCranLibraryDict') -> 'Self': + def from_dict(cls, value: "RCranLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RCranLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RCranLibraryDict": + return _transform_to_json_value(self) # type:ignore class RCranLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/runtime_engine.py b/python/databricks/bundles/clusters/_models/runtime_engine.py index e452a2537e2..2e1559fa801 100644 --- a/python/databricks/bundles/clusters/_models/runtime_engine.py +++ b/python/databricks/bundles/clusters/_models/runtime_engine.py @@ -1,57 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam +from typing import Literal class RuntimeEngine(Enum): @@ -59,4 +9,5 @@ class RuntimeEngine(Enum): STANDARD = "STANDARD" PHOTON = "PHOTON" + RuntimeEngineParam = Literal["NULL", "STANDARD", "PHOTON"] | RuntimeEngine diff --git a/python/databricks/bundles/clusters/_models/s3_storage_info.py b/python/databricks/bundles/clusters/_models/s3_storage_info.py index b4725668e6c..071abd5b6f3 100644 --- a/python/databricks/bundles/clusters/_models/s3_storage_info.py +++ b/python/databricks/bundles/clusters/_models/s3_storage_info.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class S3StorageInfo: """ A storage location in Amazon S3 """ + destination: VariableOr[str] """ S3 destination, e.g. `s3://my-bucket/some-prefix` Note that logs will be delivered using @@ -110,12 +64,11 @@ class S3StorageInfo: """ @classmethod - def from_dict(cls, value: 'S3StorageInfoDict') -> 'Self': + def from_dict(cls, value: "S3StorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'S3StorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "S3StorageInfoDict": + return _transform_to_json_value(self) # type:ignore class S3StorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/volumes_storage_info.py b/python/databricks/bundles/clusters/_models/volumes_storage_info.py index 286fcf86728..1eb4fff7bc2 100644 --- a/python/databricks/bundles/clusters/_models/volumes_storage_info.py +++ b/python/databricks/bundles/clusters/_models/volumes_storage_info.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class VolumesStorageInfo: """ A storage location back by UC Volumes. """ + destination: VariableOr[str] """ UC Volumes destination, e.g. `/Volumes/catalog/schema/vol1/init-scripts/setup-datadog.sh` @@ -70,12 +24,11 @@ class VolumesStorageInfo: """ @classmethod - def from_dict(cls, value: 'VolumesStorageInfoDict') -> 'Self': + def from_dict(cls, value: "VolumesStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'VolumesStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "VolumesStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class VolumesStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/workload_type.py b/python/databricks/bundles/clusters/_models/workload_type.py index 28867bb95f6..6760e7d7edd 100644 --- a/python/databricks/bundles/clusters/_models/workload_type.py +++ b/python/databricks/bundles/clusters/_models/workload_type.py @@ -1,80 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource +from databricks.bundles.clusters._models.clients_types import ( + ClientsTypes, + ClientsTypesParam, +) from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class WorkloadType: """ Cluster Attributes showing for clusters workload types. """ + clients: VariableOr[ClientsTypes] """ defined what type of clients can use the cluster. E.g. Notebooks, Jobs """ @classmethod - def from_dict(cls, value: 'WorkloadTypeDict') -> 'Self': + def from_dict(cls, value: "WorkloadTypeDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'WorkloadTypeDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "WorkloadTypeDict": + return _transform_to_json_value(self) # type:ignore class WorkloadTypeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/clusters/_models/workspace_storage_info.py b/python/databricks/bundles/clusters/_models/workspace_storage_info.py index 7d309c7a53a..49aef9256c8 100644 --- a/python/databricks/bundles/clusters/_models/workspace_storage_info.py +++ b/python/databricks/bundles/clusters/_models/workspace_storage_info.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.clusters._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.clusters._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.clusters._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.clusters._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.clusters._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.clusters._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.clusters._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.clusters._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.clusters._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.clusters._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.clusters._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.clusters._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.clusters._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.clusters._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.clusters._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.clusters._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.clusters._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.clusters._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.clusters._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.clusters._models.cluster import Cluster, ClusterDict, ClusterParam -from databricks.bundles.clusters._models.cluster_permission import ClusterPermission, ClusterPermissionDict, ClusterPermissionParam -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.clusters._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.clusters._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel, ClusterPermissionLevelParam - -from databricks.bundles.clusters._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.clusters._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.clusters._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.clusters._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.clusters._models.kind import Kind, KindParam - -from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class WorkspaceStorageInfo: """ A storage location in Workspace Filesystem (WSFS) """ + destination: VariableOr[str] """ wsfs destination, e.g. `workspace:/cluster-init-scripts/setup-datadog.sh` """ @classmethod - def from_dict(cls, value: 'WorkspaceStorageInfoDict') -> 'Self': + def from_dict(cls, value: "WorkspaceStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'WorkspaceStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "WorkspaceStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class WorkspaceStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/core/__init__.py b/python/databricks/bundles/core/__init__.py index ede73a09089..b40cb9f9684 100644 --- a/python/databricks/bundles/core/__init__.py +++ b/python/databricks/bundles/core/__init__.py @@ -58,24 +58,6 @@ Diagnostics, Severity, ) -from databricks.bundles.core._load import ( - load_resources_from_current_package_module, - load_resources_from_module, - load_resources_from_modules, - load_resources_from_package_module, -) -from databricks.bundles.core._location import Location -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._resource_mutator import ResourceMutator -from databricks.bundles.core._resources import Resources -from databricks.bundles.core._variable import ( - Variable, - VariableOr, - VariableOrDict, - VariableOrList, - VariableOrOptional, - variables, -) from databricks.bundles.core._generated import ( alert_mutator, app_mutator, @@ -109,3 +91,21 @@ vector_search_index_mutator, volume_mutator, ) +from databricks.bundles.core._load import ( + load_resources_from_current_package_module, + load_resources_from_module, + load_resources_from_modules, + load_resources_from_package_module, +) +from databricks.bundles.core._location import Location +from databricks.bundles.core._resource import Resource +from databricks.bundles.core._resource_mutator import ResourceMutator +from databricks.bundles.core._resources import Resources +from databricks.bundles.core._variable import ( + Variable, + VariableOr, + VariableOrDict, + VariableOrList, + VariableOrOptional, + variables, +) diff --git a/python/databricks/bundles/core/_generated/__init__.py b/python/databricks/bundles/core/_generated/__init__.py index 31a1548612b..6a56d2eb856 100644 --- a/python/databricks/bundles/core/_generated/__init__.py +++ b/python/databricks/bundles/core/_generated/__init__.py @@ -4,34 +4,109 @@ from databricks.bundles.core._generated.alerts import _AlertResources, alert_mutator from databricks.bundles.core._generated.apps import _AppResources, app_mutator -from databricks.bundles.core._generated.catalogs import _CatalogResources, catalog_mutator -from databricks.bundles.core._generated.cluster_policies import _ClusterPolicyResources, cluster_policy_mutator -from databricks.bundles.core._generated.clusters import _ClusterResources, cluster_mutator -from databricks.bundles.core._generated.dashboards import _DashboardResources, dashboard_mutator -from databricks.bundles.core._generated.database_catalogs import _DatabaseCatalogResources, database_catalog_mutator -from databricks.bundles.core._generated.database_instances import _DatabaseInstanceResources, database_instance_mutator -from databricks.bundles.core._generated.experiments import _MlflowExperimentResources, mlflow_experiment_mutator -from databricks.bundles.core._generated.external_locations import _ExternalLocationResources, external_location_mutator -from databricks.bundles.core._generated.features import _FeatureResources, feature_mutator -from databricks.bundles.core._generated.genie_spaces import _GenieSpaceResources, genie_space_mutator -from databricks.bundles.core._generated.instance_pools import _InstancePoolResources, instance_pool_mutator -from databricks.bundles.core._generated.job_runs import _JobRunResources, job_run_mutator +from databricks.bundles.core._generated.catalogs import ( + _CatalogResources, + catalog_mutator, +) +from databricks.bundles.core._generated.cluster_policies import ( + _ClusterPolicyResources, + cluster_policy_mutator, +) +from databricks.bundles.core._generated.clusters import ( + _ClusterResources, + cluster_mutator, +) +from databricks.bundles.core._generated.dashboards import ( + _DashboardResources, + dashboard_mutator, +) +from databricks.bundles.core._generated.database_catalogs import ( + _DatabaseCatalogResources, + database_catalog_mutator, +) +from databricks.bundles.core._generated.database_instances import ( + _DatabaseInstanceResources, + database_instance_mutator, +) +from databricks.bundles.core._generated.experiments import ( + _MlflowExperimentResources, + mlflow_experiment_mutator, +) +from databricks.bundles.core._generated.external_locations import ( + _ExternalLocationResources, + external_location_mutator, +) +from databricks.bundles.core._generated.features import ( + _FeatureResources, + feature_mutator, +) +from databricks.bundles.core._generated.genie_spaces import ( + _GenieSpaceResources, + genie_space_mutator, +) +from databricks.bundles.core._generated.instance_pools import ( + _InstancePoolResources, + instance_pool_mutator, +) +from databricks.bundles.core._generated.job_runs import ( + _JobRunResources, + job_run_mutator, +) from databricks.bundles.core._generated.jobs import _JobResources, job_mutator -from databricks.bundles.core._generated.mcp_services import _McpServiceResources, mcp_service_mutator -from databricks.bundles.core._generated.model_provider_services import _ModelProviderServiceResources, model_provider_service_mutator -from databricks.bundles.core._generated.model_services import _ModelServiceResources, model_service_mutator -from databricks.bundles.core._generated.model_serving_endpoints import _ModelServingEndpointResources, model_serving_endpoint_mutator -from databricks.bundles.core._generated.models import _MlflowModelResources, mlflow_model_mutator -from databricks.bundles.core._generated.pipelines import _PipelineResources, pipeline_mutator -from databricks.bundles.core._generated.quality_monitors import _QualityMonitorResources, quality_monitor_mutator -from databricks.bundles.core._generated.registered_models import _RegisteredModelResources, registered_model_mutator +from databricks.bundles.core._generated.mcp_services import ( + _McpServiceResources, + mcp_service_mutator, +) +from databricks.bundles.core._generated.model_provider_services import ( + _ModelProviderServiceResources, + model_provider_service_mutator, +) +from databricks.bundles.core._generated.model_services import ( + _ModelServiceResources, + model_service_mutator, +) +from databricks.bundles.core._generated.model_serving_endpoints import ( + _ModelServingEndpointResources, + model_serving_endpoint_mutator, +) +from databricks.bundles.core._generated.models import ( + _MlflowModelResources, + mlflow_model_mutator, +) +from databricks.bundles.core._generated.pipelines import ( + _PipelineResources, + pipeline_mutator, +) +from databricks.bundles.core._generated.quality_monitors import ( + _QualityMonitorResources, + quality_monitor_mutator, +) +from databricks.bundles.core._generated.registered_models import ( + _RegisteredModelResources, + registered_model_mutator, +) from databricks.bundles.core._generated.schemas import _SchemaResources, schema_mutator -from databricks.bundles.core._generated.secret_scopes import _SecretScopeResources, secret_scope_mutator +from databricks.bundles.core._generated.secret_scopes import ( + _SecretScopeResources, + secret_scope_mutator, +) from databricks.bundles.core._generated.secrets import _SecretResources, secret_mutator -from databricks.bundles.core._generated.sql_warehouses import _SqlWarehouseResources, sql_warehouse_mutator -from databricks.bundles.core._generated.synced_database_tables import _SyncedDatabaseTableResources, synced_database_table_mutator -from databricks.bundles.core._generated.vector_search_endpoints import _VectorSearchEndpointResources, vector_search_endpoint_mutator -from databricks.bundles.core._generated.vector_search_indexes import _VectorSearchIndexResources, vector_search_index_mutator +from databricks.bundles.core._generated.sql_warehouses import ( + _SqlWarehouseResources, + sql_warehouse_mutator, +) +from databricks.bundles.core._generated.synced_database_tables import ( + _SyncedDatabaseTableResources, + synced_database_table_mutator, +) +from databricks.bundles.core._generated.vector_search_endpoints import ( + _VectorSearchEndpointResources, + vector_search_endpoint_mutator, +) +from databricks.bundles.core._generated.vector_search_indexes import ( + _VectorSearchIndexResources, + vector_search_index_mutator, +) from databricks.bundles.core._generated.volumes import _VolumeResources, volume_mutator if TYPE_CHECKING: diff --git a/python/databricks/bundles/core/_generated/alerts.py b/python/databricks/bundles/core/_generated/alerts.py index 95a8da2a6a2..94e4af9a136 100644 --- a/python/databricks/bundles/core/_generated/alerts.py +++ b/python/databricks/bundles/core/_generated/alerts.py @@ -9,13 +9,13 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: - from databricks.bundles.core._resource_type import _ResourceType from databricks.bundles.alerts._models.alert import Alert, AlertParam + from databricks.bundles.core._resource_type import _ResourceType def _resource_type() -> "_ResourceType": - from databricks.bundles.core._resource_type import _ResourceType from databricks.bundles.alerts._models.alert import Alert + from databricks.bundles.core._resource_type import _ResourceType return _ResourceType( resource_type=Alert, diff --git a/python/databricks/bundles/core/_generated/apps.py b/python/databricks/bundles/core/_generated/apps.py index bb3c7b4e3cf..51d14e2c602 100644 --- a/python/databricks/bundles/core/_generated/apps.py +++ b/python/databricks/bundles/core/_generated/apps.py @@ -9,13 +9,13 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: - from databricks.bundles.core._resource_type import _ResourceType from databricks.bundles.apps._models.app import App, AppParam + from databricks.bundles.core._resource_type import _ResourceType def _resource_type() -> "_ResourceType": - from databricks.bundles.core._resource_type import _ResourceType from databricks.bundles.apps._models.app import App + from databricks.bundles.core._resource_type import _ResourceType return _ResourceType( resource_type=App, diff --git a/python/databricks/bundles/core/_generated/catalogs.py b/python/databricks/bundles/core/_generated/catalogs.py index 80869aa9b66..55d6bbcd310 100644 --- a/python/databricks/bundles/core/_generated/catalogs.py +++ b/python/databricks/bundles/core/_generated/catalogs.py @@ -9,13 +9,13 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: - from databricks.bundles.core._resource_type import _ResourceType from databricks.bundles.catalogs._models.catalog import Catalog, CatalogParam + from databricks.bundles.core._resource_type import _ResourceType def _resource_type() -> "_ResourceType": - from databricks.bundles.core._resource_type import _ResourceType from databricks.bundles.catalogs._models.catalog import Catalog + from databricks.bundles.core._resource_type import _ResourceType return _ResourceType( resource_type=Catalog, diff --git a/python/databricks/bundles/core/_generated/cluster_policies.py b/python/databricks/bundles/core/_generated/cluster_policies.py index 6899f2f9d4f..82da760997b 100644 --- a/python/databricks/bundles/core/_generated/cluster_policies.py +++ b/python/databricks/bundles/core/_generated/cluster_policies.py @@ -9,13 +9,16 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: + from databricks.bundles.cluster_policies._models.cluster_policy import ( + ClusterPolicy, + ClusterPolicyParam, + ) from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy, ClusterPolicyParam def _resource_type() -> "_ResourceType": - from databricks.bundles.core._resource_type import _ResourceType from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy + from databricks.bundles.core._resource_type import _ResourceType return _ResourceType( resource_type=ClusterPolicy, @@ -64,7 +67,9 @@ def add_cluster_policy( :param cluster_policy: the cluster_policy to add, can be ClusterPolicy or dict :param location: optional location of the cluster_policy in the source code """ - from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy + from databricks.bundles.cluster_policies._models.cluster_policy import ( + ClusterPolicy, + ) cluster_policy = _transform(ClusterPolicy, cluster_policy) path = ("resources", "cluster_policies", resource_name) diff --git a/python/databricks/bundles/core/_generated/clusters.py b/python/databricks/bundles/core/_generated/clusters.py index 5fa9216e6a1..44c94108ac4 100644 --- a/python/databricks/bundles/core/_generated/clusters.py +++ b/python/databricks/bundles/core/_generated/clusters.py @@ -9,13 +9,13 @@ from databricks.bundles.core._transform import _transform if TYPE_CHECKING: - from databricks.bundles.core._resource_type import _ResourceType from databricks.bundles.clusters._models.cluster import Cluster, ClusterParam + from databricks.bundles.core._resource_type import _ResourceType def _resource_type() -> "_ResourceType": - from databricks.bundles.core._resource_type import _ResourceType from databricks.bundles.clusters._models.cluster import Cluster + from databricks.bundles.core._resource_type import _ResourceType return _ResourceType( resource_type=Cluster, diff --git a/python/databricks/bundles/core/_generated/dashboards.py b/python/databricks/bundles/core/_generated/dashboards.py index ae185162c23..75cbb368354 100644 --- a/python/databricks/bundles/core/_generated/dashboards.py +++ b/python/databricks/bundles/core/_generated/dashboards.py @@ -10,7 +10,10 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardParam + from databricks.bundles.dashboards._models.dashboard import ( + Dashboard, + DashboardParam, + ) def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/database_catalogs.py b/python/databricks/bundles/core/_generated/database_catalogs.py index 6ff8d0d023b..a11e165b17f 100644 --- a/python/databricks/bundles/core/_generated/database_catalogs.py +++ b/python/databricks/bundles/core/_generated/database_catalogs.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog, DatabaseCatalogParam + from databricks.bundles.database_catalogs._models.database_catalog import ( + DatabaseCatalog, + DatabaseCatalogParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog + from databricks.bundles.database_catalogs._models.database_catalog import ( + DatabaseCatalog, + ) return _ResourceType( resource_type=DatabaseCatalog, @@ -64,7 +69,9 @@ def add_database_catalog( :param database_catalog: the database_catalog to add, can be DatabaseCatalog or dict :param location: optional location of the database_catalog in the source code """ - from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog + from databricks.bundles.database_catalogs._models.database_catalog import ( + DatabaseCatalog, + ) database_catalog = _transform(DatabaseCatalog, database_catalog) path = ("resources", "database_catalogs", resource_name) @@ -110,6 +117,8 @@ def my_database_catalog_mutator(bundle: Bundle, database_catalog: DatabaseCatalo :param function: Function that mutates database_catalogs. """ - from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog + from databricks.bundles.database_catalogs._models.database_catalog import ( + DatabaseCatalog, + ) return ResourceMutator(resource_type=DatabaseCatalog, function=function) diff --git a/python/databricks/bundles/core/_generated/database_instances.py b/python/databricks/bundles/core/_generated/database_instances.py index 8eff3dbdb55..9352ce5a37f 100644 --- a/python/databricks/bundles/core/_generated/database_instances.py +++ b/python/databricks/bundles/core/_generated/database_instances.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceParam + from databricks.bundles.database_instances._models.database_instance import ( + DatabaseInstance, + DatabaseInstanceParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.database_instances._models.database_instance import DatabaseInstance + from databricks.bundles.database_instances._models.database_instance import ( + DatabaseInstance, + ) return _ResourceType( resource_type=DatabaseInstance, @@ -64,7 +69,9 @@ def add_database_instance( :param database_instance: the database_instance to add, can be DatabaseInstance or dict :param location: optional location of the database_instance in the source code """ - from databricks.bundles.database_instances._models.database_instance import DatabaseInstance + from databricks.bundles.database_instances._models.database_instance import ( + DatabaseInstance, + ) database_instance = _transform(DatabaseInstance, database_instance) path = ("resources", "database_instances", resource_name) @@ -95,7 +102,9 @@ def database_instance_mutator( ) -> ResourceMutator["DatabaseInstance"]: ... -def database_instance_mutator(function: Callable) -> ResourceMutator["DatabaseInstance"]: +def database_instance_mutator( + function: Callable, +) -> ResourceMutator["DatabaseInstance"]: """ Decorator for defining mutator for database_instances. Function should return a new instance of the database_instance with the desired changes, instead of mutating the input database_instance. @@ -110,6 +119,8 @@ def my_database_instance_mutator(bundle: Bundle, database_instance: DatabaseInst :param function: Function that mutates database_instances. """ - from databricks.bundles.database_instances._models.database_instance import DatabaseInstance + from databricks.bundles.database_instances._models.database_instance import ( + DatabaseInstance, + ) return ResourceMutator(resource_type=DatabaseInstance, function=function) diff --git a/python/databricks/bundles/core/_generated/experiments.py b/python/databricks/bundles/core/_generated/experiments.py index 3b8868d51f8..ca0860f3155 100644 --- a/python/databricks/bundles/core/_generated/experiments.py +++ b/python/databricks/bundles/core/_generated/experiments.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentParam + from databricks.bundles.experiments._models.mlflow_experiment import ( + MlflowExperiment, + MlflowExperimentParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment + from databricks.bundles.experiments._models.mlflow_experiment import ( + MlflowExperiment, + ) return _ResourceType( resource_type=MlflowExperiment, @@ -64,7 +69,9 @@ def add_mlflow_experiment( :param mlflow_experiment: the mlflow_experiment to add, can be MlflowExperiment or dict :param location: optional location of the mlflow_experiment in the source code """ - from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment + from databricks.bundles.experiments._models.mlflow_experiment import ( + MlflowExperiment, + ) mlflow_experiment = _transform(MlflowExperiment, mlflow_experiment) path = ("resources", "experiments", resource_name) @@ -95,7 +102,9 @@ def mlflow_experiment_mutator( ) -> ResourceMutator["MlflowExperiment"]: ... -def mlflow_experiment_mutator(function: Callable) -> ResourceMutator["MlflowExperiment"]: +def mlflow_experiment_mutator( + function: Callable, +) -> ResourceMutator["MlflowExperiment"]: """ Decorator for defining mutator for experiments. Function should return a new instance of the mlflow_experiment with the desired changes, instead of mutating the input mlflow_experiment. @@ -110,6 +119,8 @@ def my_mlflow_experiment_mutator(bundle: Bundle, mlflow_experiment: MlflowExperi :param function: Function that mutates experiments. """ - from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment + from databricks.bundles.experiments._models.mlflow_experiment import ( + MlflowExperiment, + ) return ResourceMutator(resource_type=MlflowExperiment, function=function) diff --git a/python/databricks/bundles/core/_generated/external_locations.py b/python/databricks/bundles/core/_generated/external_locations.py index 92d29e4a857..d30243989a2 100644 --- a/python/databricks/bundles/core/_generated/external_locations.py +++ b/python/databricks/bundles/core/_generated/external_locations.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationParam + from databricks.bundles.external_locations._models.external_location import ( + ExternalLocation, + ExternalLocationParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.external_locations._models.external_location import ExternalLocation + from databricks.bundles.external_locations._models.external_location import ( + ExternalLocation, + ) return _ResourceType( resource_type=ExternalLocation, @@ -64,7 +69,9 @@ def add_external_location( :param external_location: the external_location to add, can be ExternalLocation or dict :param location: optional location of the external_location in the source code """ - from databricks.bundles.external_locations._models.external_location import ExternalLocation + from databricks.bundles.external_locations._models.external_location import ( + ExternalLocation, + ) external_location = _transform(ExternalLocation, external_location) path = ("resources", "external_locations", resource_name) @@ -95,7 +102,9 @@ def external_location_mutator( ) -> ResourceMutator["ExternalLocation"]: ... -def external_location_mutator(function: Callable) -> ResourceMutator["ExternalLocation"]: +def external_location_mutator( + function: Callable, +) -> ResourceMutator["ExternalLocation"]: """ Decorator for defining mutator for external_locations. Function should return a new instance of the external_location with the desired changes, instead of mutating the input external_location. @@ -110,6 +119,8 @@ def my_external_location_mutator(bundle: Bundle, external_location: ExternalLoca :param function: Function that mutates external_locations. """ - from databricks.bundles.external_locations._models.external_location import ExternalLocation + from databricks.bundles.external_locations._models.external_location import ( + ExternalLocation, + ) return ResourceMutator(resource_type=ExternalLocation, function=function) diff --git a/python/databricks/bundles/core/_generated/genie_spaces.py b/python/databricks/bundles/core/_generated/genie_spaces.py index 7c6bea33299..37fd6cad561 100644 --- a/python/databricks/bundles/core/_generated/genie_spaces.py +++ b/python/databricks/bundles/core/_generated/genie_spaces.py @@ -10,7 +10,10 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceParam + from databricks.bundles.genie_spaces._models.genie_space import ( + GenieSpace, + GenieSpaceParam, + ) def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/instance_pools.py b/python/databricks/bundles/core/_generated/instance_pools.py index dd996369e5d..0ce75fc3d2e 100644 --- a/python/databricks/bundles/core/_generated/instance_pools.py +++ b/python/databricks/bundles/core/_generated/instance_pools.py @@ -10,7 +10,10 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolParam + from databricks.bundles.instance_pools._models.instance_pool import ( + InstancePool, + InstancePoolParam, + ) def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/mcp_services.py b/python/databricks/bundles/core/_generated/mcp_services.py index 213a3768bc9..5e5bf32ae08 100644 --- a/python/databricks/bundles/core/_generated/mcp_services.py +++ b/python/databricks/bundles/core/_generated/mcp_services.py @@ -10,7 +10,10 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceParam + from databricks.bundles.mcp_services._models.mcp_service import ( + McpService, + McpServiceParam, + ) def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/model_provider_services.py b/python/databricks/bundles/core/_generated/model_provider_services.py index 4275e42138d..89f5b631fcc 100644 --- a/python/databricks/bundles/core/_generated/model_provider_services.py +++ b/python/databricks/bundles/core/_generated/model_provider_services.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceParam + from databricks.bundles.model_provider_services._models.model_provider_service import ( + ModelProviderService, + ModelProviderServiceParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService + from databricks.bundles.model_provider_services._models.model_provider_service import ( + ModelProviderService, + ) return _ResourceType( resource_type=ModelProviderService, @@ -64,9 +69,13 @@ def add_model_provider_service( :param model_provider_service: the model_provider_service to add, can be ModelProviderService or dict :param location: optional location of the model_provider_service in the source code """ - from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService + from databricks.bundles.model_provider_services._models.model_provider_service import ( + ModelProviderService, + ) - model_provider_service = _transform(ModelProviderService, model_provider_service) + model_provider_service = _transform( + ModelProviderService, model_provider_service + ) path = ("resources", "model_provider_services", resource_name) location = location or Location.from_stack_frame(depth=1) @@ -80,7 +89,9 @@ def add_model_provider_service( if location: self.add_location(path, location) - self._resources["model_provider_services"][resource_name] = model_provider_service + self._resources["model_provider_services"][resource_name] = ( + model_provider_service + ) @overload @@ -95,7 +106,9 @@ def model_provider_service_mutator( ) -> ResourceMutator["ModelProviderService"]: ... -def model_provider_service_mutator(function: Callable) -> ResourceMutator["ModelProviderService"]: +def model_provider_service_mutator( + function: Callable, +) -> ResourceMutator["ModelProviderService"]: """ Decorator for defining mutator for model_provider_services. Function should return a new instance of the model_provider_service with the desired changes, instead of mutating the input model_provider_service. @@ -110,6 +123,8 @@ def my_model_provider_service_mutator(bundle: Bundle, model_provider_service: Mo :param function: Function that mutates model_provider_services. """ - from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService + from databricks.bundles.model_provider_services._models.model_provider_service import ( + ModelProviderService, + ) return ResourceMutator(resource_type=ModelProviderService, function=function) diff --git a/python/databricks/bundles/core/_generated/model_services.py b/python/databricks/bundles/core/_generated/model_services.py index 7b09fedda52..2b37ebe7d6b 100644 --- a/python/databricks/bundles/core/_generated/model_services.py +++ b/python/databricks/bundles/core/_generated/model_services.py @@ -10,7 +10,10 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceParam + from databricks.bundles.model_services._models.model_service import ( + ModelService, + ModelServiceParam, + ) def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/model_serving_endpoints.py b/python/databricks/bundles/core/_generated/model_serving_endpoints.py index 06d8fcc0616..56a9ecde661 100644 --- a/python/databricks/bundles/core/_generated/model_serving_endpoints.py +++ b/python/databricks/bundles/core/_generated/model_serving_endpoints.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointParam + from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( + ModelServingEndpoint, + ModelServingEndpointParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint + from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( + ModelServingEndpoint, + ) return _ResourceType( resource_type=ModelServingEndpoint, @@ -64,9 +69,13 @@ def add_model_serving_endpoint( :param model_serving_endpoint: the model_serving_endpoint to add, can be ModelServingEndpoint or dict :param location: optional location of the model_serving_endpoint in the source code """ - from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint + from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( + ModelServingEndpoint, + ) - model_serving_endpoint = _transform(ModelServingEndpoint, model_serving_endpoint) + model_serving_endpoint = _transform( + ModelServingEndpoint, model_serving_endpoint + ) path = ("resources", "model_serving_endpoints", resource_name) location = location or Location.from_stack_frame(depth=1) @@ -80,7 +89,9 @@ def add_model_serving_endpoint( if location: self.add_location(path, location) - self._resources["model_serving_endpoints"][resource_name] = model_serving_endpoint + self._resources["model_serving_endpoints"][resource_name] = ( + model_serving_endpoint + ) @overload @@ -95,7 +106,9 @@ def model_serving_endpoint_mutator( ) -> ResourceMutator["ModelServingEndpoint"]: ... -def model_serving_endpoint_mutator(function: Callable) -> ResourceMutator["ModelServingEndpoint"]: +def model_serving_endpoint_mutator( + function: Callable, +) -> ResourceMutator["ModelServingEndpoint"]: """ Decorator for defining mutator for model_serving_endpoints. Function should return a new instance of the model_serving_endpoint with the desired changes, instead of mutating the input model_serving_endpoint. @@ -110,6 +123,8 @@ def my_model_serving_endpoint_mutator(bundle: Bundle, model_serving_endpoint: Mo :param function: Function that mutates model_serving_endpoints. """ - from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint + from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( + ModelServingEndpoint, + ) return ResourceMutator(resource_type=ModelServingEndpoint, function=function) diff --git a/python/databricks/bundles/core/_generated/models.py b/python/databricks/bundles/core/_generated/models.py index ab4caad1bca..df7a4c6d6a3 100644 --- a/python/databricks/bundles/core/_generated/models.py +++ b/python/databricks/bundles/core/_generated/models.py @@ -10,7 +10,10 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelParam + from databricks.bundles.models._models.mlflow_model import ( + MlflowModel, + MlflowModelParam, + ) def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/quality_monitors.py b/python/databricks/bundles/core/_generated/quality_monitors.py index d8c46b01e42..ba3f52a75ae 100644 --- a/python/databricks/bundles/core/_generated/quality_monitors.py +++ b/python/databricks/bundles/core/_generated/quality_monitors.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorParam + from databricks.bundles.quality_monitors._models.quality_monitor import ( + QualityMonitor, + QualityMonitorParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor + from databricks.bundles.quality_monitors._models.quality_monitor import ( + QualityMonitor, + ) return _ResourceType( resource_type=QualityMonitor, @@ -64,7 +69,9 @@ def add_quality_monitor( :param quality_monitor: the quality_monitor to add, can be QualityMonitor or dict :param location: optional location of the quality_monitor in the source code """ - from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor + from databricks.bundles.quality_monitors._models.quality_monitor import ( + QualityMonitor, + ) quality_monitor = _transform(QualityMonitor, quality_monitor) path = ("resources", "quality_monitors", resource_name) @@ -110,6 +117,8 @@ def my_quality_monitor_mutator(bundle: Bundle, quality_monitor: QualityMonitor) :param function: Function that mutates quality_monitors. """ - from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor + from databricks.bundles.quality_monitors._models.quality_monitor import ( + QualityMonitor, + ) return ResourceMutator(resource_type=QualityMonitor, function=function) diff --git a/python/databricks/bundles/core/_generated/registered_models.py b/python/databricks/bundles/core/_generated/registered_models.py index 42bfb70bf88..cdc492beeb3 100644 --- a/python/databricks/bundles/core/_generated/registered_models.py +++ b/python/databricks/bundles/core/_generated/registered_models.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelParam + from databricks.bundles.registered_models._models.registered_model import ( + RegisteredModel, + RegisteredModelParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.registered_models._models.registered_model import RegisteredModel + from databricks.bundles.registered_models._models.registered_model import ( + RegisteredModel, + ) return _ResourceType( resource_type=RegisteredModel, @@ -64,7 +69,9 @@ def add_registered_model( :param registered_model: the registered_model to add, can be RegisteredModel or dict :param location: optional location of the registered_model in the source code """ - from databricks.bundles.registered_models._models.registered_model import RegisteredModel + from databricks.bundles.registered_models._models.registered_model import ( + RegisteredModel, + ) registered_model = _transform(RegisteredModel, registered_model) path = ("resources", "registered_models", resource_name) @@ -110,6 +117,8 @@ def my_registered_model_mutator(bundle: Bundle, registered_model: RegisteredMode :param function: Function that mutates registered_models. """ - from databricks.bundles.registered_models._models.registered_model import RegisteredModel + from databricks.bundles.registered_models._models.registered_model import ( + RegisteredModel, + ) return ResourceMutator(resource_type=RegisteredModel, function=function) diff --git a/python/databricks/bundles/core/_generated/secret_scopes.py b/python/databricks/bundles/core/_generated/secret_scopes.py index 4635888bc8f..a8be3660a6a 100644 --- a/python/databricks/bundles/core/_generated/secret_scopes.py +++ b/python/databricks/bundles/core/_generated/secret_scopes.py @@ -10,7 +10,10 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeParam + from databricks.bundles.secret_scopes._models.secret_scope import ( + SecretScope, + SecretScopeParam, + ) def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/sql_warehouses.py b/python/databricks/bundles/core/_generated/sql_warehouses.py index 7630d8af559..05f8bb0cf76 100644 --- a/python/databricks/bundles/core/_generated/sql_warehouses.py +++ b/python/databricks/bundles/core/_generated/sql_warehouses.py @@ -10,7 +10,10 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseParam + from databricks.bundles.sql_warehouses._models.sql_warehouse import ( + SqlWarehouse, + SqlWarehouseParam, + ) def _resource_type() -> "_ResourceType": diff --git a/python/databricks/bundles/core/_generated/synced_database_tables.py b/python/databricks/bundles/core/_generated/synced_database_tables.py index 90139c00d58..2d0c7e598c0 100644 --- a/python/databricks/bundles/core/_generated/synced_database_tables.py +++ b/python/databricks/bundles/core/_generated/synced_database_tables.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableParam + from databricks.bundles.synced_database_tables._models.synced_database_table import ( + SyncedDatabaseTable, + SyncedDatabaseTableParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable + from databricks.bundles.synced_database_tables._models.synced_database_table import ( + SyncedDatabaseTable, + ) return _ResourceType( resource_type=SyncedDatabaseTable, @@ -64,7 +69,9 @@ def add_synced_database_table( :param synced_database_table: the synced_database_table to add, can be SyncedDatabaseTable or dict :param location: optional location of the synced_database_table in the source code """ - from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable + from databricks.bundles.synced_database_tables._models.synced_database_table import ( + SyncedDatabaseTable, + ) synced_database_table = _transform(SyncedDatabaseTable, synced_database_table) path = ("resources", "synced_database_tables", resource_name) @@ -80,7 +87,9 @@ def add_synced_database_table( if location: self.add_location(path, location) - self._resources["synced_database_tables"][resource_name] = synced_database_table + self._resources["synced_database_tables"][resource_name] = ( + synced_database_table + ) @overload @@ -95,7 +104,9 @@ def synced_database_table_mutator( ) -> ResourceMutator["SyncedDatabaseTable"]: ... -def synced_database_table_mutator(function: Callable) -> ResourceMutator["SyncedDatabaseTable"]: +def synced_database_table_mutator( + function: Callable, +) -> ResourceMutator["SyncedDatabaseTable"]: """ Decorator for defining mutator for synced_database_tables. Function should return a new instance of the synced_database_table with the desired changes, instead of mutating the input synced_database_table. @@ -110,6 +121,8 @@ def my_synced_database_table_mutator(bundle: Bundle, synced_database_table: Sync :param function: Function that mutates synced_database_tables. """ - from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable + from databricks.bundles.synced_database_tables._models.synced_database_table import ( + SyncedDatabaseTable, + ) return ResourceMutator(resource_type=SyncedDatabaseTable, function=function) diff --git a/python/databricks/bundles/core/_generated/vector_search_endpoints.py b/python/databricks/bundles/core/_generated/vector_search_endpoints.py index 54f6777c0f7..839831dafc6 100644 --- a/python/databricks/bundles/core/_generated/vector_search_endpoints.py +++ b/python/databricks/bundles/core/_generated/vector_search_endpoints.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointParam + from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( + VectorSearchEndpoint, + VectorSearchEndpointParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint + from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( + VectorSearchEndpoint, + ) return _ResourceType( resource_type=VectorSearchEndpoint, @@ -64,9 +69,13 @@ def add_vector_search_endpoint( :param vector_search_endpoint: the vector_search_endpoint to add, can be VectorSearchEndpoint or dict :param location: optional location of the vector_search_endpoint in the source code """ - from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint + from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( + VectorSearchEndpoint, + ) - vector_search_endpoint = _transform(VectorSearchEndpoint, vector_search_endpoint) + vector_search_endpoint = _transform( + VectorSearchEndpoint, vector_search_endpoint + ) path = ("resources", "vector_search_endpoints", resource_name) location = location or Location.from_stack_frame(depth=1) @@ -80,7 +89,9 @@ def add_vector_search_endpoint( if location: self.add_location(path, location) - self._resources["vector_search_endpoints"][resource_name] = vector_search_endpoint + self._resources["vector_search_endpoints"][resource_name] = ( + vector_search_endpoint + ) @overload @@ -95,7 +106,9 @@ def vector_search_endpoint_mutator( ) -> ResourceMutator["VectorSearchEndpoint"]: ... -def vector_search_endpoint_mutator(function: Callable) -> ResourceMutator["VectorSearchEndpoint"]: +def vector_search_endpoint_mutator( + function: Callable, +) -> ResourceMutator["VectorSearchEndpoint"]: """ Decorator for defining mutator for vector_search_endpoints. Function should return a new instance of the vector_search_endpoint with the desired changes, instead of mutating the input vector_search_endpoint. @@ -110,6 +123,8 @@ def my_vector_search_endpoint_mutator(bundle: Bundle, vector_search_endpoint: Ve :param function: Function that mutates vector_search_endpoints. """ - from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint + from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( + VectorSearchEndpoint, + ) return ResourceMutator(resource_type=VectorSearchEndpoint, function=function) diff --git a/python/databricks/bundles/core/_generated/vector_search_indexes.py b/python/databricks/bundles/core/_generated/vector_search_indexes.py index d85c65ced41..7c84a0d00c8 100644 --- a/python/databricks/bundles/core/_generated/vector_search_indexes.py +++ b/python/databricks/bundles/core/_generated/vector_search_indexes.py @@ -10,12 +10,17 @@ if TYPE_CHECKING: from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexParam + from databricks.bundles.vector_search_indexes._models.vector_search_index import ( + VectorSearchIndex, + VectorSearchIndexParam, + ) def _resource_type() -> "_ResourceType": from databricks.bundles.core._resource_type import _ResourceType - from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex + from databricks.bundles.vector_search_indexes._models.vector_search_index import ( + VectorSearchIndex, + ) return _ResourceType( resource_type=VectorSearchIndex, @@ -64,7 +69,9 @@ def add_vector_search_index( :param vector_search_index: the vector_search_index to add, can be VectorSearchIndex or dict :param location: optional location of the vector_search_index in the source code """ - from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex + from databricks.bundles.vector_search_indexes._models.vector_search_index import ( + VectorSearchIndex, + ) vector_search_index = _transform(VectorSearchIndex, vector_search_index) path = ("resources", "vector_search_indexes", resource_name) @@ -80,7 +87,9 @@ def add_vector_search_index( if location: self.add_location(path, location) - self._resources["vector_search_indexes"][resource_name] = vector_search_index + self._resources["vector_search_indexes"][resource_name] = ( + vector_search_index + ) @overload @@ -95,7 +104,9 @@ def vector_search_index_mutator( ) -> ResourceMutator["VectorSearchIndex"]: ... -def vector_search_index_mutator(function: Callable) -> ResourceMutator["VectorSearchIndex"]: +def vector_search_index_mutator( + function: Callable, +) -> ResourceMutator["VectorSearchIndex"]: """ Decorator for defining mutator for vector_search_indexes. Function should return a new instance of the vector_search_index with the desired changes, instead of mutating the input vector_search_index. @@ -110,6 +121,8 @@ def my_vector_search_index_mutator(bundle: Bundle, vector_search_index: VectorSe :param function: Function that mutates vector_search_indexes. """ - from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex + from databricks.bundles.vector_search_indexes._models.vector_search_index import ( + VectorSearchIndex, + ) return ResourceMutator(resource_type=VectorSearchIndex, function=function) diff --git a/python/databricks/bundles/dashboards/__init__.py b/python/databricks/bundles/dashboards/__init__.py index 8e8c855760d..126427fcca6 100644 --- a/python/databricks/bundles/dashboards/__init__.py +++ b/python/databricks/bundles/dashboards/__init__.py @@ -1,26 +1,42 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Dashboard', - 'DashboardDict', - 'DashboardParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'LifecycleState', - 'LifecycleStateParam', - 'Permission', - 'PermissionDict', - 'PermissionLevel', - 'PermissionLevelParam', - 'PermissionParam', + "Dashboard", + "DashboardDict", + "DashboardParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "LifecycleState", + "LifecycleStateParam", + "Permission", + "PermissionDict", + "PermissionLevel", + "PermissionLevelParam", + "PermissionParam", ] -from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam -from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam - -from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.dashboards._models.dashboard import ( + Dashboard, + DashboardDict, + DashboardParam, +) +from databricks.bundles.dashboards._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.dashboards._models.lifecycle_state import ( + LifecycleState, + LifecycleStateParam, +) +from databricks.bundles.dashboards._models.permission import ( + Permission, + PermissionDict, + PermissionParam, +) +from databricks.bundles.dashboards._models.permission_level import ( + PermissionLevel, + PermissionLevelParam, +) diff --git a/python/databricks/bundles/dashboards/_models/dashboard.py b/python/databricks/bundles/dashboards/_models/dashboard.py index 6c1eae4c71f..2a3e5f03364 100644 --- a/python/databricks/bundles/dashboards/_models/dashboard.py +++ b/python/databricks/bundles/dashboards/_models/dashboard.py @@ -1,24 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, Any, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam - -from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.dashboards._models.lifecycle_state import ( + LifecycleState, + LifecycleStateParam, +) +from databricks.bundles.dashboards._models.permission import Permission, PermissionParam if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Dashboard(Resource): """""" @@ -110,12 +109,11 @@ class Dashboard(Resource): """ @classmethod - def from_dict(cls, value: 'DashboardDict') -> 'Self': + def from_dict(cls, value: "DashboardDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DashboardDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DashboardDict": + return _transform_to_json_value(self) # type:ignore class DashboardDict(TypedDict, total=False): diff --git a/python/databricks/bundles/dashboards/_models/lifecycle.py b/python/databricks/bundles/dashboards/_models/lifecycle.py index 60ce33ace0e..697776a198e 100644 --- a/python/databricks/bundles/dashboards/_models/lifecycle.py +++ b/python/databricks/bundles/dashboards/_models/lifecycle.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam -from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam - -from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -29,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/dashboards/_models/lifecycle_state.py b/python/databricks/bundles/dashboards/_models/lifecycle_state.py index ae4590aead7..15654e77a1f 100644 --- a/python/databricks/bundles/dashboards/_models/lifecycle_state.py +++ b/python/databricks/bundles/dashboards/_models/lifecycle_state.py @@ -1,22 +1,12 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam -from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam +from typing import Literal class LifecycleState(Enum): ACTIVE = "ACTIVE" TRASHED = "TRASHED" + LifecycleStateParam = Literal["ACTIVE", "TRASHED"] | LifecycleState diff --git a/python/databricks/bundles/dashboards/_models/permission.py b/python/databricks/bundles/dashboards/_models/permission.py index 9b05c19befb..fb4ea994f6d 100644 --- a/python/databricks/bundles/dashboards/_models/permission.py +++ b/python/databricks/bundles/dashboards/_models/permission.py @@ -1,24 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam -from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam - -from databricks.bundles.dashboards._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.dashboards._models.permission_level import ( + PermissionLevel, + PermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Permission: """""" @@ -44,12 +40,11 @@ class Permission: """ @classmethod - def from_dict(cls, value: 'PermissionDict') -> 'Self': + def from_dict(cls, value: "PermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PermissionDict": + return _transform_to_json_value(self) # type:ignore class PermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/dashboards/_models/permission_level.py b/python/databricks/bundles/dashboards/_models/permission_level.py index 0ba31996221..c6111911b29 100644 --- a/python/databricks/bundles/dashboards/_models/permission_level.py +++ b/python/databricks/bundles/dashboards/_models/permission_level.py @@ -1,24 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.dashboards._models.dashboard import Dashboard, DashboardDict, DashboardParam -from databricks.bundles.dashboards._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.dashboards._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState, LifecycleStateParam +from typing import Literal class PermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" @@ -40,4 +30,29 @@ class PermissionLevel(Enum): CAN_MONITOR_ONLY = "CAN_MONITOR_ONLY" CAN_CREATE_APP = "CAN_CREATE_APP" -PermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW", "CAN_READ", "CAN_RUN", "CAN_EDIT", "CAN_USE", "CAN_MANAGE_STAGING_VERSIONS", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_EDIT_METADATA", "CAN_VIEW_METADATA", "CAN_BIND", "CAN_QUERY", "CAN_MONITOR", "CAN_CREATE", "CAN_MONITOR_ONLY", "CAN_CREATE_APP"] | PermissionLevel + +PermissionLevelParam = ( + Literal[ + "CAN_MANAGE", + "CAN_RESTART", + "CAN_ATTACH_TO", + "IS_OWNER", + "CAN_MANAGE_RUN", + "CAN_VIEW", + "CAN_READ", + "CAN_RUN", + "CAN_EDIT", + "CAN_USE", + "CAN_MANAGE_STAGING_VERSIONS", + "CAN_MANAGE_PRODUCTION_VERSIONS", + "CAN_EDIT_METADATA", + "CAN_VIEW_METADATA", + "CAN_BIND", + "CAN_QUERY", + "CAN_MONITOR", + "CAN_CREATE", + "CAN_MONITOR_ONLY", + "CAN_CREATE_APP", + ] + | PermissionLevel +) diff --git a/python/databricks/bundles/database_catalogs/__init__.py b/python/databricks/bundles/database_catalogs/__init__.py index e81e527d6c9..fdd1d167302 100644 --- a/python/databricks/bundles/database_catalogs/__init__.py +++ b/python/databricks/bundles/database_catalogs/__init__.py @@ -1,14 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'DatabaseCatalog', - 'DatabaseCatalogDict', - 'DatabaseCatalogParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', + "DatabaseCatalog", + "DatabaseCatalogDict", + "DatabaseCatalogParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", ] -from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog, DatabaseCatalogDict, DatabaseCatalogParam -from databricks.bundles.database_catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.database_catalogs._models.database_catalog import ( + DatabaseCatalog, + DatabaseCatalogDict, + DatabaseCatalogParam, +) +from databricks.bundles.database_catalogs._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) diff --git a/python/databricks/bundles/database_catalogs/_models/database_catalog.py b/python/databricks/bundles/database_catalogs/_models/database_catalog.py index b943d184c64..a224ac7a316 100644 --- a/python/databricks/bundles/database_catalogs/_models/database_catalog.py +++ b/python/databricks/bundles/database_catalogs/_models/database_catalog.py @@ -1,19 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.database_catalogs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.database_catalogs._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DatabaseCatalog(Resource): """""" @@ -44,12 +46,11 @@ class DatabaseCatalog(Resource): """ @classmethod - def from_dict(cls, value: 'DatabaseCatalogDict') -> 'Self': + def from_dict(cls, value: "DatabaseCatalogDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DatabaseCatalogDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DatabaseCatalogDict": + return _transform_to_json_value(self) # type:ignore class DatabaseCatalogDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_catalogs/_models/lifecycle.py b/python/databricks/bundles/database_catalogs/_models/lifecycle.py index f010c457f95..697776a198e 100644 --- a/python/databricks/bundles/database_catalogs/_models/lifecycle.py +++ b/python/databricks/bundles/database_catalogs/_models/lifecycle.py @@ -1,19 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog, DatabaseCatalogDict, DatabaseCatalogParam +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -24,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/__init__.py b/python/databricks/bundles/database_instances/__init__.py index 1b865f0ac01..13674b646a5 100644 --- a/python/databricks/bundles/database_instances/__init__.py +++ b/python/databricks/bundles/database_instances/__init__.py @@ -1,30 +1,52 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'CustomTag', - 'CustomTagDict', - 'CustomTagParam', - 'DatabaseInstance', - 'DatabaseInstanceDict', - 'DatabaseInstanceParam', - 'DatabaseInstanceRef', - 'DatabaseInstanceRefDict', - 'DatabaseInstanceRefParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'Permission', - 'PermissionDict', - 'PermissionLevel', - 'PermissionLevelParam', - 'PermissionParam', + "CustomTag", + "CustomTagDict", + "CustomTagParam", + "DatabaseInstance", + "DatabaseInstanceDict", + "DatabaseInstanceParam", + "DatabaseInstanceRef", + "DatabaseInstanceRefDict", + "DatabaseInstanceRefParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "Permission", + "PermissionDict", + "PermissionLevel", + "PermissionLevelParam", + "PermissionParam", ] -from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam -from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam -from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam -from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.database_instances._models.custom_tag import ( + CustomTag, + CustomTagDict, + CustomTagParam, +) +from databricks.bundles.database_instances._models.database_instance import ( + DatabaseInstance, + DatabaseInstanceDict, + DatabaseInstanceParam, +) +from databricks.bundles.database_instances._models.database_instance_ref import ( + DatabaseInstanceRef, + DatabaseInstanceRefDict, + DatabaseInstanceRefParam, +) +from databricks.bundles.database_instances._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.database_instances._models.permission import ( + Permission, + PermissionDict, + PermissionParam, +) +from databricks.bundles.database_instances._models.permission_level import ( + PermissionLevel, + PermissionLevelParam, +) diff --git a/python/databricks/bundles/database_instances/_models/custom_tag.py b/python/databricks/bundles/database_instances/_models/custom_tag.py index 583b3321ad3..e08eb459bf7 100644 --- a/python/databricks/bundles/database_instances/_models/custom_tag.py +++ b/python/databricks/bundles/database_instances/_models/custom_tag.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam -from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam -from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class CustomTag: """""" @@ -38,12 +30,11 @@ class CustomTag: """ @classmethod - def from_dict(cls, value: 'CustomTagDict') -> 'Self': + def from_dict(cls, value: "CustomTagDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CustomTagDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CustomTagDict": + return _transform_to_json_value(self) # type:ignore class CustomTagDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/database_instance.py b/python/databricks/bundles/database_instances/_models/database_instance.py index e0bdd84dc47..beea60bfd1f 100644 --- a/python/databricks/bundles/database_instances/_models/database_instance.py +++ b/python/databricks/bundles/database_instances/_models/database_instance.py @@ -1,29 +1,43 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam -from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam -from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.database_instances._models.custom_tag import ( + CustomTag, + CustomTagParam, +) +from databricks.bundles.database_instances._models.database_instance_ref import ( + DatabaseInstanceRef, + DatabaseInstanceRefParam, +) +from databricks.bundles.database_instances._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.database_instances._models.permission import ( + Permission, + PermissionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DatabaseInstance(Resource): """ A DatabaseInstance represents a logical Postgres instance, comprised of both compute and storage. """ + name: VariableOr[str] """ [Public Preview] The name of the instance. This is the unique identifier for the instance. @@ -95,12 +109,11 @@ class DatabaseInstance(Resource): """ @classmethod - def from_dict(cls, value: 'DatabaseInstanceDict') -> 'Self': + def from_dict(cls, value: "DatabaseInstanceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DatabaseInstanceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DatabaseInstanceDict": + return _transform_to_json_value(self) # type:ignore class DatabaseInstanceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/database_instance_ref.py b/python/databricks/bundles/database_instances/_models/database_instance_ref.py index 13663dd4a6a..8c3e582f280 100644 --- a/python/databricks/bundles/database_instances/_models/database_instance_ref.py +++ b/python/databricks/bundles/database_instances/_models/database_instance_ref.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam -from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam -from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DatabaseInstanceRef: """ @@ -31,6 +23,7 @@ class DatabaseInstanceRef: If both lsn and branch_time are specified, the lsn will be used to create the child instance. """ + branch_time: VariableOrOptional[str] = None """ [Public Preview] Branch time of the ref database instance. @@ -56,12 +49,11 @@ class DatabaseInstanceRef: """ @classmethod - def from_dict(cls, value: 'DatabaseInstanceRefDict') -> 'Self': + def from_dict(cls, value: "DatabaseInstanceRefDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DatabaseInstanceRefDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DatabaseInstanceRefDict": + return _transform_to_json_value(self) # type:ignore class DatabaseInstanceRefDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/lifecycle.py b/python/databricks/bundles/database_instances/_models/lifecycle.py index 9eda940b47c..697776a198e 100644 --- a/python/databricks/bundles/database_instances/_models/lifecycle.py +++ b/python/databricks/bundles/database_instances/_models/lifecycle.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam -from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam -from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam -from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -29,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/permission.py b/python/databricks/bundles/database_instances/_models/permission.py index b09522188ca..da0f1e44ad2 100644 --- a/python/databricks/bundles/database_instances/_models/permission.py +++ b/python/databricks/bundles/database_instances/_models/permission.py @@ -1,24 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam -from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam -from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam -from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.database_instances._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.database_instances._models.permission_level import ( + PermissionLevel, + PermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Permission: """""" @@ -44,12 +40,11 @@ class Permission: """ @classmethod - def from_dict(cls, value: 'PermissionDict') -> 'Self': + def from_dict(cls, value: "PermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PermissionDict": + return _transform_to_json_value(self) # type:ignore class PermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/database_instances/_models/permission_level.py b/python/databricks/bundles/database_instances/_models/permission_level.py index 8295a3097cb..c6111911b29 100644 --- a/python/databricks/bundles/database_instances/_models/permission_level.py +++ b/python/databricks/bundles/database_instances/_models/permission_level.py @@ -1,24 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.database_instances._models.custom_tag import CustomTag, CustomTagDict, CustomTagParam -from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef, DatabaseInstanceRefDict, DatabaseInstanceRefParam -from databricks.bundles.database_instances._models.database_instance import DatabaseInstance, DatabaseInstanceDict, DatabaseInstanceParam -from databricks.bundles.database_instances._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.database_instances._models.permission import Permission, PermissionDict, PermissionParam class PermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" @@ -40,4 +30,29 @@ class PermissionLevel(Enum): CAN_MONITOR_ONLY = "CAN_MONITOR_ONLY" CAN_CREATE_APP = "CAN_CREATE_APP" -PermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW", "CAN_READ", "CAN_RUN", "CAN_EDIT", "CAN_USE", "CAN_MANAGE_STAGING_VERSIONS", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_EDIT_METADATA", "CAN_VIEW_METADATA", "CAN_BIND", "CAN_QUERY", "CAN_MONITOR", "CAN_CREATE", "CAN_MONITOR_ONLY", "CAN_CREATE_APP"] | PermissionLevel + +PermissionLevelParam = ( + Literal[ + "CAN_MANAGE", + "CAN_RESTART", + "CAN_ATTACH_TO", + "IS_OWNER", + "CAN_MANAGE_RUN", + "CAN_VIEW", + "CAN_READ", + "CAN_RUN", + "CAN_EDIT", + "CAN_USE", + "CAN_MANAGE_STAGING_VERSIONS", + "CAN_MANAGE_PRODUCTION_VERSIONS", + "CAN_EDIT_METADATA", + "CAN_VIEW_METADATA", + "CAN_BIND", + "CAN_QUERY", + "CAN_MONITOR", + "CAN_CREATE", + "CAN_MONITOR_ONLY", + "CAN_CREATE_APP", + ] + | PermissionLevel +) diff --git a/python/databricks/bundles/experiments/__init__.py b/python/databricks/bundles/experiments/__init__.py index 9a4be4fc294..e1ae26cc521 100644 --- a/python/databricks/bundles/experiments/__init__.py +++ b/python/databricks/bundles/experiments/__init__.py @@ -1,34 +1,60 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'ExperimentPermissionLevel', - 'ExperimentPermissionLevelParam', - 'ExperimentTag', - 'ExperimentTagDict', - 'ExperimentTagParam', - 'ExperimentTraceLocation', - 'ExperimentTraceLocationDict', - 'ExperimentTraceLocationParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'MlflowExperiment', - 'MlflowExperimentDict', - 'MlflowExperimentParam', - 'MlflowExperimentPermission', - 'MlflowExperimentPermissionDict', - 'MlflowExperimentPermissionParam', - 'UcTraceLocation', - 'UcTraceLocationDict', - 'UcTraceLocationParam', + "ExperimentPermissionLevel", + "ExperimentPermissionLevelParam", + "ExperimentTag", + "ExperimentTagDict", + "ExperimentTagParam", + "ExperimentTraceLocation", + "ExperimentTraceLocationDict", + "ExperimentTraceLocationParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "MlflowExperiment", + "MlflowExperimentDict", + "MlflowExperimentParam", + "MlflowExperimentPermission", + "MlflowExperimentPermissionDict", + "MlflowExperimentPermissionParam", + "UcTraceLocation", + "UcTraceLocationDict", + "UcTraceLocationParam", ] -from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam -from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam -from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam -from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam -from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam -from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam - +from databricks.bundles.experiments._models.experiment_permission_level import ( + ExperimentPermissionLevel, + ExperimentPermissionLevelParam, +) +from databricks.bundles.experiments._models.experiment_tag import ( + ExperimentTag, + ExperimentTagDict, + ExperimentTagParam, +) +from databricks.bundles.experiments._models.experiment_trace_location import ( + ExperimentTraceLocation, + ExperimentTraceLocationDict, + ExperimentTraceLocationParam, +) +from databricks.bundles.experiments._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.experiments._models.mlflow_experiment import ( + MlflowExperiment, + MlflowExperimentDict, + MlflowExperimentParam, +) +from databricks.bundles.experiments._models.mlflow_experiment_permission import ( + MlflowExperimentPermission, + MlflowExperimentPermissionDict, + MlflowExperimentPermissionParam, +) +from databricks.bundles.experiments._models.uc_trace_location import ( + UcTraceLocation, + UcTraceLocationDict, + UcTraceLocationParam, +) diff --git a/python/databricks/bundles/experiments/_models/experiment_permission_level.py b/python/databricks/bundles/experiments/_models/experiment_permission_level.py index 275cee3f610..056727ac92c 100644 --- a/python/databricks/bundles/experiments/_models/experiment_permission_level.py +++ b/python/databricks/bundles/experiments/_models/experiment_permission_level.py @@ -1,27 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam -from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam -from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam -from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam -from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam class ExperimentPermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_EDIT = "CAN_EDIT" CAN_READ = "CAN_READ" -ExperimentPermissionLevelParam = Literal["CAN_MANAGE", "CAN_EDIT", "CAN_READ"] | ExperimentPermissionLevel + +ExperimentPermissionLevelParam = ( + Literal["CAN_MANAGE", "CAN_EDIT", "CAN_READ"] | ExperimentPermissionLevel +) diff --git a/python/databricks/bundles/experiments/_models/experiment_tag.py b/python/databricks/bundles/experiments/_models/experiment_tag.py index 8f96768c572..2c78a181b17 100644 --- a/python/databricks/bundles/experiments/_models/experiment_tag.py +++ b/python/databricks/bundles/experiments/_models/experiment_tag.py @@ -1,30 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam -from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam -from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam -from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam -from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ExperimentTag: """ A tag for an experiment. """ + key: VariableOrOptional[str] = None """ The tag key. @@ -36,12 +28,11 @@ class ExperimentTag: """ @classmethod - def from_dict(cls, value: 'ExperimentTagDict') -> 'Self': + def from_dict(cls, value: "ExperimentTagDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ExperimentTagDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ExperimentTagDict": + return _transform_to_json_value(self) # type:ignore class ExperimentTagDict(TypedDict, total=False): diff --git a/python/databricks/bundles/experiments/_models/experiment_trace_location.py b/python/databricks/bundles/experiments/_models/experiment_trace_location.py index 7585d164f35..cbc9dffb6cf 100644 --- a/python/databricks/bundles/experiments/_models/experiment_trace_location.py +++ b/python/databricks/bundles/experiments/_models/experiment_trace_location.py @@ -1,32 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam -from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam -from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam -from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam -from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.experiments._models.uc_trace_location import ( + UcTraceLocation, + UcTraceLocationParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ExperimentTraceLocation: """ :meta private: [EXPERIMENTAL] - + The storage location for an experiment's traces. """ + uc_trace_location: VariableOrOptional[UcTraceLocation] = None """ :meta private: [EXPERIMENTAL] @@ -36,12 +32,11 @@ class ExperimentTraceLocation: """ @classmethod - def from_dict(cls, value: 'ExperimentTraceLocationDict') -> 'Self': + def from_dict(cls, value: "ExperimentTraceLocationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ExperimentTraceLocationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ExperimentTraceLocationDict": + return _transform_to_json_value(self) # type:ignore class ExperimentTraceLocationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/experiments/_models/lifecycle.py b/python/databricks/bundles/experiments/_models/lifecycle.py index 520a7ac7b3d..697776a198e 100644 --- a/python/databricks/bundles/experiments/_models/lifecycle.py +++ b/python/databricks/bundles/experiments/_models/lifecycle.py @@ -1,25 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam -from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam -from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam -from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam -from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam -from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -30,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/experiments/_models/mlflow_experiment.py b/python/databricks/bundles/experiments/_models/mlflow_experiment.py index be5d319936d..8a9d5368e99 100644 --- a/python/databricks/bundles/experiments/_models/mlflow_experiment.py +++ b/python/databricks/bundles/experiments/_models/mlflow_experiment.py @@ -1,25 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam -from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam -from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam -from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam -from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.experiments._models.experiment_tag import ( + ExperimentTag, + ExperimentTagParam, +) +from databricks.bundles.experiments._models.experiment_trace_location import ( + ExperimentTraceLocation, + ExperimentTraceLocationParam, +) +from databricks.bundles.experiments._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.experiments._models.mlflow_experiment_permission import ( + MlflowExperimentPermission, + MlflowExperimentPermissionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MlflowExperiment(Resource): """""" @@ -40,7 +52,9 @@ class MlflowExperiment(Resource): Settings that control the deployment lifecycle of the resource, such as preventing it from being destroyed. """ - permissions: VariableOrList[MlflowExperimentPermission] = field(default_factory=list) + permissions: VariableOrList[MlflowExperimentPermission] = field( + default_factory=list + ) """ The permissions to apply to this resource. """ @@ -64,12 +78,11 @@ class MlflowExperiment(Resource): """ @classmethod - def from_dict(cls, value: 'MlflowExperimentDict') -> 'Self': + def from_dict(cls, value: "MlflowExperimentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MlflowExperimentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MlflowExperimentDict": + return _transform_to_json_value(self) # type:ignore class MlflowExperimentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/experiments/_models/mlflow_experiment_permission.py b/python/databricks/bundles/experiments/_models/mlflow_experiment_permission.py index fbaa465cccf..041b02d91b3 100644 --- a/python/databricks/bundles/experiments/_models/mlflow_experiment_permission.py +++ b/python/databricks/bundles/experiments/_models/mlflow_experiment_permission.py @@ -1,25 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam -from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam -from databricks.bundles.experiments._models.uc_trace_location import UcTraceLocation, UcTraceLocationDict, UcTraceLocationParam -from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam -from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.experiments._models.experiment_permission_level import ( + ExperimentPermissionLevel, + ExperimentPermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MlflowExperimentPermission: """""" @@ -45,12 +40,11 @@ class MlflowExperimentPermission: """ @classmethod - def from_dict(cls, value: 'MlflowExperimentPermissionDict') -> 'Self': + def from_dict(cls, value: "MlflowExperimentPermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MlflowExperimentPermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MlflowExperimentPermissionDict": + return _transform_to_json_value(self) # type:ignore class MlflowExperimentPermissionDict(TypedDict, total=False): @@ -77,4 +71,6 @@ class MlflowExperimentPermissionDict(TypedDict, total=False): """ -MlflowExperimentPermissionParam = MlflowExperimentPermissionDict | MlflowExperimentPermission +MlflowExperimentPermissionParam = ( + MlflowExperimentPermissionDict | MlflowExperimentPermission +) diff --git a/python/databricks/bundles/experiments/_models/uc_trace_location.py b/python/databricks/bundles/experiments/_models/uc_trace_location.py index 8abb137df3e..b7b8148823e 100644 --- a/python/databricks/bundles/experiments/_models/uc_trace_location.py +++ b/python/databricks/bundles/experiments/_models/uc_trace_location.py @@ -1,33 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.experiments._models.experiment_tag import ExperimentTag, ExperimentTagDict, ExperimentTagParam -from databricks.bundles.experiments._models.experiment_trace_location import ExperimentTraceLocation, ExperimentTraceLocationDict, ExperimentTraceLocationParam -from databricks.bundles.experiments._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment, MlflowExperimentDict, MlflowExperimentParam -from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission, MlflowExperimentPermissionDict, MlflowExperimentPermissionParam -from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel, ExperimentPermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class UcTraceLocation: """ :meta private: [EXPERIMENTAL] - + A Unity Catalog trace storage location. Traces are stored as Delta tables in the specified catalog and schema. """ + catalog: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -55,12 +47,11 @@ class UcTraceLocation: """ @classmethod - def from_dict(cls, value: 'UcTraceLocationDict') -> 'Self': + def from_dict(cls, value: "UcTraceLocationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'UcTraceLocationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "UcTraceLocationDict": + return _transform_to_json_value(self) # type:ignore class UcTraceLocationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/__init__.py b/python/databricks/bundles/external_locations/__init__.py index 7a69b1d9836..d6aa3bf908c 100644 --- a/python/databricks/bundles/external_locations/__init__.py +++ b/python/databricks/bundles/external_locations/__init__.py @@ -1,50 +1,90 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'AwsSqsQueue', - 'AwsSqsQueueDict', - 'AwsSqsQueueParam', - 'AzureQueueStorage', - 'AzureQueueStorageDict', - 'AzureQueueStorageParam', - 'EncryptionDetails', - 'EncryptionDetailsDict', - 'EncryptionDetailsParam', - 'ExternalLocation', - 'ExternalLocationDict', - 'ExternalLocationParam', - 'FileEventQueue', - 'FileEventQueueDict', - 'FileEventQueueParam', - 'GcpPubsub', - 'GcpPubsubDict', - 'GcpPubsubParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'SseEncryptionDetails', - 'SseEncryptionDetailsAlgorithm', - 'SseEncryptionDetailsAlgorithmParam', - 'SseEncryptionDetailsDict', - 'SseEncryptionDetailsParam', + "AwsSqsQueue", + "AwsSqsQueueDict", + "AwsSqsQueueParam", + "AzureQueueStorage", + "AzureQueueStorageDict", + "AzureQueueStorageParam", + "EncryptionDetails", + "EncryptionDetailsDict", + "EncryptionDetailsParam", + "ExternalLocation", + "ExternalLocationDict", + "ExternalLocationParam", + "FileEventQueue", + "FileEventQueueDict", + "FileEventQueueParam", + "GcpPubsub", + "GcpPubsubDict", + "GcpPubsubParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", + "SseEncryptionDetails", + "SseEncryptionDetailsAlgorithm", + "SseEncryptionDetailsAlgorithmParam", + "SseEncryptionDetailsDict", + "SseEncryptionDetailsParam", ] -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.external_locations._models.aws_sqs_queue import ( + AwsSqsQueue, + AwsSqsQueueDict, + AwsSqsQueueParam, +) +from databricks.bundles.external_locations._models.azure_queue_storage import ( + AzureQueueStorage, + AzureQueueStorageDict, + AzureQueueStorageParam, +) +from databricks.bundles.external_locations._models.encryption_details import ( + EncryptionDetails, + EncryptionDetailsDict, + EncryptionDetailsParam, +) +from databricks.bundles.external_locations._models.external_location import ( + ExternalLocation, + ExternalLocationDict, + ExternalLocationParam, +) +from databricks.bundles.external_locations._models.file_event_queue import ( + FileEventQueue, + FileEventQueueDict, + FileEventQueueParam, +) +from databricks.bundles.external_locations._models.gcp_pubsub import ( + GcpPubsub, + GcpPubsubDict, + GcpPubsubParam, +) +from databricks.bundles.external_locations._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.external_locations._models.privilege import ( + Privilege, + PrivilegeParam, +) +from databricks.bundles.external_locations._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) +from databricks.bundles.external_locations._models.sse_encryption_details import ( + SseEncryptionDetails, + SseEncryptionDetailsDict, + SseEncryptionDetailsParam, +) +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import ( + SseEncryptionDetailsAlgorithm, + SseEncryptionDetailsAlgorithmParam, +) diff --git a/python/databricks/bundles/external_locations/_models/aws_sqs_queue.py b/python/databricks/bundles/external_locations/_models/aws_sqs_queue.py index 437490dad63..53c7a0b7062 100644 --- a/python/databricks/bundles/external_locations/_models/aws_sqs_queue.py +++ b/python/databricks/bundles/external_locations/_models/aws_sqs_queue.py @@ -1,30 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AwsSqsQueue: """""" @@ -36,12 +22,11 @@ class AwsSqsQueue: """ @classmethod - def from_dict(cls, value: 'AwsSqsQueueDict') -> 'Self': + def from_dict(cls, value: "AwsSqsQueueDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AwsSqsQueueDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AwsSqsQueueDict": + return _transform_to_json_value(self) # type:ignore class AwsSqsQueueDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/azure_queue_storage.py b/python/databricks/bundles/external_locations/_models/azure_queue_storage.py index 3357af42478..a8d182421f7 100644 --- a/python/databricks/bundles/external_locations/_models/azure_queue_storage.py +++ b/python/databricks/bundles/external_locations/_models/azure_queue_storage.py @@ -1,30 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AzureQueueStorage: """""" @@ -50,12 +36,11 @@ class AzureQueueStorage: """ @classmethod - def from_dict(cls, value: 'AzureQueueStorageDict') -> 'Self': + def from_dict(cls, value: "AzureQueueStorageDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AzureQueueStorageDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AzureQueueStorageDict": + return _transform_to_json_value(self) # type:ignore class AzureQueueStorageDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/encryption_details.py b/python/databricks/bundles/external_locations/_models/encryption_details.py index 5d0df12237e..d1b67fa76bd 100644 --- a/python/databricks/bundles/external_locations/_models/encryption_details.py +++ b/python/databricks/bundles/external_locations/_models/encryption_details.py @@ -1,47 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.external_locations._models.sse_encryption_details import ( + SseEncryptionDetails, + SseEncryptionDetailsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EncryptionDetails: """ Encryption options that apply to clients connecting to cloud storage. """ + sse_encryption_details: VariableOrOptional[SseEncryptionDetails] = None """ Server-Side Encryption properties for clients communicating with AWS s3. """ @classmethod - def from_dict(cls, value: 'EncryptionDetailsDict') -> 'Self': + def from_dict(cls, value: "EncryptionDetailsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EncryptionDetailsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EncryptionDetailsDict": + return _transform_to_json_value(self) # type:ignore class EncryptionDetailsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/external_location.py b/python/databricks/bundles/external_locations/_models/external_location.py index 9e0c99840a6..ea72a5da3dd 100644 --- a/python/databricks/bundles/external_locations/_models/external_location.py +++ b/python/databricks/bundles/external_locations/_models/external_location.py @@ -1,30 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.external_locations._models.encryption_details import ( + EncryptionDetails, + EncryptionDetailsParam, +) +from databricks.bundles.external_locations._models.file_event_queue import ( + FileEventQueue, + FileEventQueueParam, +) +from databricks.bundles.external_locations._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.external_locations._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ExternalLocation(Resource): """""" @@ -91,12 +98,11 @@ class ExternalLocation(Resource): """ @classmethod - def from_dict(cls, value: 'ExternalLocationDict') -> 'Self': + def from_dict(cls, value: "ExternalLocationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ExternalLocationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ExternalLocationDict": + return _transform_to_json_value(self) # type:ignore class ExternalLocationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/file_event_queue.py b/python/databricks/bundles/external_locations/_models/file_event_queue.py index bb224e885bc..78a791766cd 100644 --- a/python/databricks/bundles/external_locations/_models/file_event_queue.py +++ b/python/databricks/bundles/external_locations/_models/file_event_queue.py @@ -1,30 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.external_locations._models.aws_sqs_queue import ( + AwsSqsQueue, + AwsSqsQueueParam, +) +from databricks.bundles.external_locations._models.azure_queue_storage import ( + AzureQueueStorage, + AzureQueueStorageParam, +) +from databricks.bundles.external_locations._models.gcp_pubsub import ( + GcpPubsub, + GcpPubsubParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FileEventQueue: """""" @@ -42,12 +40,11 @@ class FileEventQueue: provided_sqs: VariableOrOptional[AwsSqsQueue] = None @classmethod - def from_dict(cls, value: 'FileEventQueueDict') -> 'Self': + def from_dict(cls, value: "FileEventQueueDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FileEventQueueDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FileEventQueueDict": + return _transform_to_json_value(self) # type:ignore class FileEventQueueDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/gcp_pubsub.py b/python/databricks/bundles/external_locations/_models/gcp_pubsub.py index ac2055c98d6..926b27d2a55 100644 --- a/python/databricks/bundles/external_locations/_models/gcp_pubsub.py +++ b/python/databricks/bundles/external_locations/_models/gcp_pubsub.py @@ -1,30 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GcpPubsub: """""" @@ -36,12 +22,11 @@ class GcpPubsub: """ @classmethod - def from_dict(cls, value: 'GcpPubsubDict') -> 'Self': + def from_dict(cls, value: "GcpPubsubDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GcpPubsubDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GcpPubsubDict": + return _transform_to_json_value(self) # type:ignore class GcpPubsubDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/lifecycle.py b/python/databricks/bundles/external_locations/_models/lifecycle.py index 842d400f831..697776a198e 100644 --- a/python/databricks/bundles/external_locations/_models/lifecycle.py +++ b/python/databricks/bundles/external_locations/_models/lifecycle.py @@ -1,30 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -35,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/privilege.py b/python/databricks/bundles/external_locations/_models/privilege.py index 41cf98001ae..df45211a76a 100644 --- a/python/databricks/bundles/external_locations/_models/privilege.py +++ b/python/databricks/bundles/external_locations/_models/privilege.py @@ -1,24 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam +from typing import Literal class Privilege(Enum): @@ -75,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/external_locations/_models/privilege_assignment.py b/python/databricks/bundles/external_locations/_models/privilege_assignment.py index f6b498fea1e..f48f9511ac0 100644 --- a/python/databricks/bundles/external_locations/_models/privilege_assignment.py +++ b/python/databricks/bundles/external_locations/_models/privilege_assignment.py @@ -1,30 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.external_locations._models.privilege import ( + Privilege, + PrivilegeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -41,12 +31,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/sse_encryption_details.py b/python/databricks/bundles/external_locations/_models/sse_encryption_details.py index 46c78a5774c..339b98b8b26 100644 --- a/python/databricks/bundles/external_locations/_models/sse_encryption_details.py +++ b/python/databricks/bundles/external_locations/_models/sse_encryption_details.py @@ -1,35 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import SseEncryptionDetailsAlgorithm, SseEncryptionDetailsAlgorithmParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.external_locations._models.sse_encryption_details_algorithm import ( + SseEncryptionDetailsAlgorithm, + SseEncryptionDetailsAlgorithmParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SseEncryptionDetails: """ Server-Side Encryption properties for clients communicating with AWS s3. """ + algorithm: VariableOrOptional[SseEncryptionDetailsAlgorithm] = None """ Sets the value of the 'x-amz-server-side-encryption' header in S3 request. @@ -42,12 +33,11 @@ class SseEncryptionDetails: """ @classmethod - def from_dict(cls, value: 'SseEncryptionDetailsDict') -> 'Self': + def from_dict(cls, value: "SseEncryptionDetailsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SseEncryptionDetailsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SseEncryptionDetailsDict": + return _transform_to_json_value(self) # type:ignore class SseEncryptionDetailsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/external_locations/_models/sse_encryption_details_algorithm.py b/python/databricks/bundles/external_locations/_models/sse_encryption_details_algorithm.py index 9bc52f98294..d5d63383a74 100644 --- a/python/databricks/bundles/external_locations/_models/sse_encryption_details_algorithm.py +++ b/python/databricks/bundles/external_locations/_models/sse_encryption_details_algorithm.py @@ -1,31 +1,18 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.external_locations._models.aws_sqs_queue import AwsSqsQueue, AwsSqsQueueDict, AwsSqsQueueParam -from databricks.bundles.external_locations._models.azure_queue_storage import AzureQueueStorage, AzureQueueStorageDict, AzureQueueStorageParam -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails, EncryptionDetailsDict, EncryptionDetailsParam -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue, FileEventQueueDict, FileEventQueueParam -from databricks.bundles.external_locations._models.gcp_pubsub import GcpPubsub, GcpPubsubDict, GcpPubsubParam -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.external_locations._models.sse_encryption_details import SseEncryptionDetails, SseEncryptionDetailsDict, SseEncryptionDetailsParam -from databricks.bundles.external_locations._models.external_location import ExternalLocation, ExternalLocationDict, ExternalLocationParam -from databricks.bundles.external_locations._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.external_locations._models.privilege import Privilege, PrivilegeParam +from typing import Literal class SseEncryptionDetailsAlgorithm(Enum): """ SSE algorithm to use for encrypting S3 objects """ + AWS_SSE_S3 = "AWS_SSE_S3" AWS_SSE_KMS = "AWS_SSE_KMS" -SseEncryptionDetailsAlgorithmParam = Literal["AWS_SSE_S3", "AWS_SSE_KMS"] | SseEncryptionDetailsAlgorithm + +SseEncryptionDetailsAlgorithmParam = ( + Literal["AWS_SSE_S3", "AWS_SSE_KMS"] | SseEncryptionDetailsAlgorithm +) diff --git a/python/databricks/bundles/features/__init__.py b/python/databricks/bundles/features/__init__.py index e8eae18922b..ecf951b1e12 100644 --- a/python/databricks/bundles/features/__init__.py +++ b/python/databricks/bundles/features/__init__.py @@ -1,186 +1,364 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'AggregationFunction', - 'AggregationFunctionDict', - 'AggregationFunctionParam', - 'ApproxCountDistinctFunction', - 'ApproxCountDistinctFunctionDict', - 'ApproxCountDistinctFunctionParam', - 'ApproxPercentileFunction', - 'ApproxPercentileFunctionDict', - 'ApproxPercentileFunctionParam', - 'AvgFunction', - 'AvgFunctionDict', - 'AvgFunctionParam', - 'ColumnSelection', - 'ColumnSelectionDict', - 'ColumnSelectionParam', - 'CountFunction', - 'CountFunctionDict', - 'CountFunctionParam', - 'CustomUdf', - 'CustomUdfDict', - 'CustomUdfParam', - 'DataSource', - 'DataSourceDict', - 'DataSourceParam', - 'DeltaTableSource', - 'DeltaTableSourceDict', - 'DeltaTableSourceParam', - 'EntityColumn', - 'EntityColumnDict', - 'EntityColumnParam', - 'Feature', - 'FeatureDict', - 'FeatureParam', - 'FeatureReference', - 'FeatureReferenceDict', - 'FeatureReferenceParam', - 'FeatureViewSource', - 'FeatureViewSourceDict', - 'FeatureViewSourceParam', - 'FieldDefinition', - 'FieldDefinitionDict', - 'FieldDefinitionParam', - 'FirstDistinctFunction', - 'FirstDistinctFunctionDict', - 'FirstDistinctFunctionParam', - 'FirstFunction', - 'FirstFunctionDict', - 'FirstFunctionParam', - 'FirstNFunction', - 'FirstNFunctionDict', - 'FirstNFunctionParam', - 'FlatSchema', - 'FlatSchemaDict', - 'FlatSchemaParam', - 'Function', - 'FunctionDict', - 'FunctionParam', - 'InputBinding', - 'InputBindingDict', - 'InputBindingParam', - 'JobContext', - 'JobContextDict', - 'JobContextParam', - 'KafkaSource', - 'KafkaSourceDict', - 'KafkaSourceParam', - 'LastDistinctFunction', - 'LastDistinctFunctionDict', - 'LastDistinctFunctionParam', - 'LastFunction', - 'LastFunctionDict', - 'LastFunctionParam', - 'LastNFunction', - 'LastNFunctionDict', - 'LastNFunctionParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'LineageContext', - 'LineageContextDict', - 'LineageContextParam', - 'MaxFunction', - 'MaxFunctionDict', - 'MaxFunctionParam', - 'MinFunction', - 'MinFunctionDict', - 'MinFunctionParam', - 'RequestSource', - 'RequestSourceDict', - 'RequestSourceParam', - 'RollingWindow', - 'RollingWindowDict', - 'RollingWindowParam', - 'SawtoothWindow', - 'SawtoothWindowDict', - 'SawtoothWindowParam', - 'ScalarDataType', - 'ScalarDataTypeParam', - 'SdpSource', - 'SdpSourceDict', - 'SdpSourceParam', - 'SlidingWindow', - 'SlidingWindowDict', - 'SlidingWindowParam', - 'SourceLateness', - 'SourceLatenessDict', - 'SourceLatenessParam', - 'StddevPopFunction', - 'StddevPopFunctionDict', - 'StddevPopFunctionParam', - 'StddevSampFunction', - 'StddevSampFunctionDict', - 'StddevSampFunctionParam', - 'StreamSource', - 'StreamSourceDict', - 'StreamSourceParam', - 'SumFunction', - 'SumFunctionDict', - 'SumFunctionParam', - 'TimeWindow', - 'TimeWindowDict', - 'TimeWindowParam', - 'TimeseriesColumn', - 'TimeseriesColumnDict', - 'TimeseriesColumnParam', - 'TumblingWindow', - 'TumblingWindowDict', - 'TumblingWindowParam', - 'VarPopFunction', - 'VarPopFunctionDict', - 'VarPopFunctionParam', - 'VarSampFunction', - 'VarSampFunctionDict', - 'VarSampFunctionParam', + "AggregationFunction", + "AggregationFunctionDict", + "AggregationFunctionParam", + "ApproxCountDistinctFunction", + "ApproxCountDistinctFunctionDict", + "ApproxCountDistinctFunctionParam", + "ApproxPercentileFunction", + "ApproxPercentileFunctionDict", + "ApproxPercentileFunctionParam", + "AvgFunction", + "AvgFunctionDict", + "AvgFunctionParam", + "ColumnSelection", + "ColumnSelectionDict", + "ColumnSelectionParam", + "CountFunction", + "CountFunctionDict", + "CountFunctionParam", + "CustomUdf", + "CustomUdfDict", + "CustomUdfParam", + "DataSource", + "DataSourceDict", + "DataSourceParam", + "DeltaTableSource", + "DeltaTableSourceDict", + "DeltaTableSourceParam", + "EntityColumn", + "EntityColumnDict", + "EntityColumnParam", + "Feature", + "FeatureDict", + "FeatureParam", + "FeatureReference", + "FeatureReferenceDict", + "FeatureReferenceParam", + "FeatureViewSource", + "FeatureViewSourceDict", + "FeatureViewSourceParam", + "FieldDefinition", + "FieldDefinitionDict", + "FieldDefinitionParam", + "FirstDistinctFunction", + "FirstDistinctFunctionDict", + "FirstDistinctFunctionParam", + "FirstFunction", + "FirstFunctionDict", + "FirstFunctionParam", + "FirstNFunction", + "FirstNFunctionDict", + "FirstNFunctionParam", + "FlatSchema", + "FlatSchemaDict", + "FlatSchemaParam", + "Function", + "FunctionDict", + "FunctionParam", + "InputBinding", + "InputBindingDict", + "InputBindingParam", + "JobContext", + "JobContextDict", + "JobContextParam", + "KafkaSource", + "KafkaSourceDict", + "KafkaSourceParam", + "LastDistinctFunction", + "LastDistinctFunctionDict", + "LastDistinctFunctionParam", + "LastFunction", + "LastFunctionDict", + "LastFunctionParam", + "LastNFunction", + "LastNFunctionDict", + "LastNFunctionParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "LineageContext", + "LineageContextDict", + "LineageContextParam", + "MaxFunction", + "MaxFunctionDict", + "MaxFunctionParam", + "MinFunction", + "MinFunctionDict", + "MinFunctionParam", + "RequestSource", + "RequestSourceDict", + "RequestSourceParam", + "RollingWindow", + "RollingWindowDict", + "RollingWindowParam", + "SawtoothWindow", + "SawtoothWindowDict", + "SawtoothWindowParam", + "ScalarDataType", + "ScalarDataTypeParam", + "SdpSource", + "SdpSourceDict", + "SdpSourceParam", + "SlidingWindow", + "SlidingWindowDict", + "SlidingWindowParam", + "SourceLateness", + "SourceLatenessDict", + "SourceLatenessParam", + "StddevPopFunction", + "StddevPopFunctionDict", + "StddevPopFunctionParam", + "StddevSampFunction", + "StddevSampFunctionDict", + "StddevSampFunctionParam", + "StreamSource", + "StreamSourceDict", + "StreamSourceParam", + "SumFunction", + "SumFunctionDict", + "SumFunctionParam", + "TimeWindow", + "TimeWindowDict", + "TimeWindowParam", + "TimeseriesColumn", + "TimeseriesColumnDict", + "TimeseriesColumnParam", + "TumblingWindow", + "TumblingWindowDict", + "TumblingWindowParam", + "VarPopFunction", + "VarPopFunctionDict", + "VarPopFunctionParam", + "VarSampFunction", + "VarSampFunctionDict", + "VarSampFunctionParam", ] -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.features._models.aggregation_function import ( + AggregationFunction, + AggregationFunctionDict, + AggregationFunctionParam, +) +from databricks.bundles.features._models.approx_count_distinct_function import ( + ApproxCountDistinctFunction, + ApproxCountDistinctFunctionDict, + ApproxCountDistinctFunctionParam, +) +from databricks.bundles.features._models.approx_percentile_function import ( + ApproxPercentileFunction, + ApproxPercentileFunctionDict, + ApproxPercentileFunctionParam, +) +from databricks.bundles.features._models.avg_function import ( + AvgFunction, + AvgFunctionDict, + AvgFunctionParam, +) +from databricks.bundles.features._models.column_selection import ( + ColumnSelection, + ColumnSelectionDict, + ColumnSelectionParam, +) +from databricks.bundles.features._models.count_function import ( + CountFunction, + CountFunctionDict, + CountFunctionParam, +) +from databricks.bundles.features._models.custom_udf import ( + CustomUdf, + CustomUdfDict, + CustomUdfParam, +) +from databricks.bundles.features._models.data_source import ( + DataSource, + DataSourceDict, + DataSourceParam, +) +from databricks.bundles.features._models.delta_table_source import ( + DeltaTableSource, + DeltaTableSourceDict, + DeltaTableSourceParam, +) +from databricks.bundles.features._models.entity_column import ( + EntityColumn, + EntityColumnDict, + EntityColumnParam, +) +from databricks.bundles.features._models.feature import ( + Feature, + FeatureDict, + FeatureParam, +) +from databricks.bundles.features._models.feature_reference import ( + FeatureReference, + FeatureReferenceDict, + FeatureReferenceParam, +) +from databricks.bundles.features._models.feature_view_source import ( + FeatureViewSource, + FeatureViewSourceDict, + FeatureViewSourceParam, +) +from databricks.bundles.features._models.field_definition import ( + FieldDefinition, + FieldDefinitionDict, + FieldDefinitionParam, +) +from databricks.bundles.features._models.first_distinct_function import ( + FirstDistinctFunction, + FirstDistinctFunctionDict, + FirstDistinctFunctionParam, +) +from databricks.bundles.features._models.first_function import ( + FirstFunction, + FirstFunctionDict, + FirstFunctionParam, +) +from databricks.bundles.features._models.first_n_function import ( + FirstNFunction, + FirstNFunctionDict, + FirstNFunctionParam, +) +from databricks.bundles.features._models.flat_schema import ( + FlatSchema, + FlatSchemaDict, + FlatSchemaParam, +) +from databricks.bundles.features._models.function import ( + Function, + FunctionDict, + FunctionParam, +) +from databricks.bundles.features._models.input_binding import ( + InputBinding, + InputBindingDict, + InputBindingParam, +) +from databricks.bundles.features._models.job_context import ( + JobContext, + JobContextDict, + JobContextParam, +) +from databricks.bundles.features._models.kafka_source import ( + KafkaSource, + KafkaSourceDict, + KafkaSourceParam, +) +from databricks.bundles.features._models.last_distinct_function import ( + LastDistinctFunction, + LastDistinctFunctionDict, + LastDistinctFunctionParam, +) +from databricks.bundles.features._models.last_function import ( + LastFunction, + LastFunctionDict, + LastFunctionParam, +) +from databricks.bundles.features._models.last_n_function import ( + LastNFunction, + LastNFunctionDict, + LastNFunctionParam, +) +from databricks.bundles.features._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.features._models.lineage_context import ( + LineageContext, + LineageContextDict, + LineageContextParam, +) +from databricks.bundles.features._models.max_function import ( + MaxFunction, + MaxFunctionDict, + MaxFunctionParam, +) +from databricks.bundles.features._models.min_function import ( + MinFunction, + MinFunctionDict, + MinFunctionParam, +) +from databricks.bundles.features._models.request_source import ( + RequestSource, + RequestSourceDict, + RequestSourceParam, +) +from databricks.bundles.features._models.rolling_window import ( + RollingWindow, + RollingWindowDict, + RollingWindowParam, +) +from databricks.bundles.features._models.sawtooth_window import ( + SawtoothWindow, + SawtoothWindowDict, + SawtoothWindowParam, +) +from databricks.bundles.features._models.scalar_data_type import ( + ScalarDataType, + ScalarDataTypeParam, +) +from databricks.bundles.features._models.sdp_source import ( + SdpSource, + SdpSourceDict, + SdpSourceParam, +) +from databricks.bundles.features._models.sliding_window import ( + SlidingWindow, + SlidingWindowDict, + SlidingWindowParam, +) +from databricks.bundles.features._models.source_lateness import ( + SourceLateness, + SourceLatenessDict, + SourceLatenessParam, +) +from databricks.bundles.features._models.stddev_pop_function import ( + StddevPopFunction, + StddevPopFunctionDict, + StddevPopFunctionParam, +) +from databricks.bundles.features._models.stddev_samp_function import ( + StddevSampFunction, + StddevSampFunctionDict, + StddevSampFunctionParam, +) +from databricks.bundles.features._models.stream_source import ( + StreamSource, + StreamSourceDict, + StreamSourceParam, +) +from databricks.bundles.features._models.sum_function import ( + SumFunction, + SumFunctionDict, + SumFunctionParam, +) +from databricks.bundles.features._models.time_window import ( + TimeWindow, + TimeWindowDict, + TimeWindowParam, +) +from databricks.bundles.features._models.timeseries_column import ( + TimeseriesColumn, + TimeseriesColumnDict, + TimeseriesColumnParam, +) +from databricks.bundles.features._models.tumbling_window import ( + TumblingWindow, + TumblingWindowDict, + TumblingWindowParam, +) +from databricks.bundles.features._models.var_pop_function import ( + VarPopFunction, + VarPopFunctionDict, + VarPopFunctionParam, +) +from databricks.bundles.features._models.var_samp_function import ( + VarSampFunction, + VarSampFunctionDict, + VarSampFunctionParam, +) diff --git a/python/databricks/bundles/features/_models/aggregation_function.py b/python/databricks/bundles/features/_models/aggregation_function.py index fe593e4d8ff..292f03909fa 100644 --- a/python/databricks/bundles/features/_models/aggregation_function.py +++ b/python/databricks/bundles/features/_models/aggregation_function.py @@ -1,68 +1,91 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.features._models.approx_count_distinct_function import ( + ApproxCountDistinctFunction, + ApproxCountDistinctFunctionParam, +) +from databricks.bundles.features._models.approx_percentile_function import ( + ApproxPercentileFunction, + ApproxPercentileFunctionParam, +) +from databricks.bundles.features._models.avg_function import ( + AvgFunction, + AvgFunctionParam, +) +from databricks.bundles.features._models.count_function import ( + CountFunction, + CountFunctionParam, +) +from databricks.bundles.features._models.first_distinct_function import ( + FirstDistinctFunction, + FirstDistinctFunctionParam, +) +from databricks.bundles.features._models.first_function import ( + FirstFunction, + FirstFunctionParam, +) +from databricks.bundles.features._models.first_n_function import ( + FirstNFunction, + FirstNFunctionParam, +) +from databricks.bundles.features._models.last_distinct_function import ( + LastDistinctFunction, + LastDistinctFunctionParam, +) +from databricks.bundles.features._models.last_function import ( + LastFunction, + LastFunctionParam, +) +from databricks.bundles.features._models.last_n_function import ( + LastNFunction, + LastNFunctionParam, +) +from databricks.bundles.features._models.max_function import ( + MaxFunction, + MaxFunctionParam, +) +from databricks.bundles.features._models.min_function import ( + MinFunction, + MinFunctionParam, +) +from databricks.bundles.features._models.stddev_pop_function import ( + StddevPopFunction, + StddevPopFunctionParam, +) +from databricks.bundles.features._models.stddev_samp_function import ( + StddevSampFunction, + StddevSampFunctionParam, +) +from databricks.bundles.features._models.sum_function import ( + SumFunction, + SumFunctionParam, +) +from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowParam +from databricks.bundles.features._models.var_pop_function import ( + VarPopFunction, + VarPopFunctionParam, +) +from databricks.bundles.features._models.var_samp_function import ( + VarSampFunction, + VarSampFunctionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AggregationFunction: """ An aggregation function applied over a time window. """ + approx_count_distinct: VariableOrOptional[ApproxCountDistinctFunction] = None """ Computes the approximate count of distinct values. @@ -162,12 +185,11 @@ class AggregationFunction: """ @classmethod - def from_dict(cls, value: 'AggregationFunctionDict') -> 'Self': + def from_dict(cls, value: "AggregationFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AggregationFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AggregationFunctionDict": + return _transform_to_json_value(self) # type:ignore class AggregationFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/approx_count_distinct_function.py b/python/databricks/bundles/features/_models/approx_count_distinct_function.py index 6eec220608e..6244d5a431c 100644 --- a/python/databricks/bundles/features/_models/approx_count_distinct_function.py +++ b/python/databricks/bundles/features/_models/approx_count_distinct_function.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ApproxCountDistinctFunction: """ Computes the approximate count of distinct values. """ + input: VariableOr[str] """ The input column from which the approximate count of distinct values is computed. @@ -74,12 +28,11 @@ class ApproxCountDistinctFunction: """ @classmethod - def from_dict(cls, value: 'ApproxCountDistinctFunctionDict') -> 'Self': + def from_dict(cls, value: "ApproxCountDistinctFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ApproxCountDistinctFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ApproxCountDistinctFunctionDict": + return _transform_to_json_value(self) # type:ignore class ApproxCountDistinctFunctionDict(TypedDict, total=False): @@ -96,4 +49,6 @@ class ApproxCountDistinctFunctionDict(TypedDict, total=False): """ -ApproxCountDistinctFunctionParam = ApproxCountDistinctFunctionDict | ApproxCountDistinctFunction +ApproxCountDistinctFunctionParam = ( + ApproxCountDistinctFunctionDict | ApproxCountDistinctFunction +) diff --git a/python/databricks/bundles/features/_models/approx_percentile_function.py b/python/databricks/bundles/features/_models/approx_percentile_function.py index 235bd72fbb0..b4648e41209 100644 --- a/python/databricks/bundles/features/_models/approx_percentile_function.py +++ b/python/databricks/bundles/features/_models/approx_percentile_function.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ApproxPercentileFunction: """ Computes the approximate percentile of values. """ + input: VariableOr[str] """ The input column from which the approximate percentile is computed. @@ -79,12 +33,11 @@ class ApproxPercentileFunction: """ @classmethod - def from_dict(cls, value: 'ApproxPercentileFunctionDict') -> 'Self': + def from_dict(cls, value: "ApproxPercentileFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ApproxPercentileFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ApproxPercentileFunctionDict": + return _transform_to_json_value(self) # type:ignore class ApproxPercentileFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/avg_function.py b/python/databricks/bundles/features/_models/avg_function.py index f25ac863f21..5ecf0cc486b 100644 --- a/python/databricks/bundles/features/_models/avg_function.py +++ b/python/databricks/bundles/features/_models/avg_function.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AvgFunction: """ Computes the average of values. """ + input: VariableOr[str] """ The input column from which the average is computed. For Kafka sources, use dot-prefixed path @@ -72,12 +26,11 @@ class AvgFunction: """ @classmethod - def from_dict(cls, value: 'AvgFunctionDict') -> 'Self': + def from_dict(cls, value: "AvgFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AvgFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AvgFunctionDict": + return _transform_to_json_value(self) # type:ignore class AvgFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/column_selection.py b/python/databricks/bundles/features/_models/column_selection.py index 78b53500b33..d09726cd190 100644 --- a/python/databricks/bundles/features/_models/column_selection.py +++ b/python/databricks/bundles/features/_models/column_selection.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ColumnSelection: """ A ColumnSelection function, equivalent to the LAST() record of an entity over a lifetime window """ + column: VariableOr[str] """ Column name from source to select as the feature value. """ @classmethod - def from_dict(cls, value: 'ColumnSelectionDict') -> 'Self': + def from_dict(cls, value: "ColumnSelectionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ColumnSelectionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ColumnSelectionDict": + return _transform_to_json_value(self) # type:ignore class ColumnSelectionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/count_function.py b/python/databricks/bundles/features/_models/count_function.py index 20c103aee19..6cd86a68132 100644 --- a/python/databricks/bundles/features/_models/count_function.py +++ b/python/databricks/bundles/features/_models/count_function.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class CountFunction: """ Computes the count of values. """ + input: VariableOr[str] """ The input column from which the count is computed. For Kafka sources, use dot-prefixed path @@ -72,12 +26,11 @@ class CountFunction: """ @classmethod - def from_dict(cls, value: 'CountFunctionDict') -> 'Self': + def from_dict(cls, value: "CountFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CountFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CountFunctionDict": + return _transform_to_json_value(self) # type:ignore class CountFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/custom_udf.py b/python/databricks/bundles/features/_models/custom_udf.py index 2fbf69fc96d..7130a8e59cb 100644 --- a/python/databricks/bundles/features/_models/custom_udf.py +++ b/python/databricks/bundles/features/_models/custom_udf.py @@ -1,71 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrList +from databricks.bundles.features._models.input_binding import ( + InputBinding, + InputBindingParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class CustomUdf: """ :meta private: [EXPERIMENTAL] - + A CustomUdf function applies a registered Unity Catalog function row-wise to source columns, producing a single output column per row. """ + function_path: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -82,12 +40,11 @@ class CustomUdf: """ @classmethod - def from_dict(cls, value: 'CustomUdfDict') -> 'Self': + def from_dict(cls, value: "CustomUdfDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CustomUdfDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CustomUdfDict": + return _transform_to_json_value(self) # type:ignore class CustomUdfDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/data_source.py b/python/databricks/bundles/features/_models/data_source.py index 7f41d3d180b..26c945d6d7a 100644 --- a/python/databricks/bundles/features/_models/data_source.py +++ b/python/databricks/bundles/features/_models/data_source.py @@ -1,68 +1,47 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.features._models.delta_table_source import ( + DeltaTableSource, + DeltaTableSourceParam, +) +from databricks.bundles.features._models.feature_view_source import ( + FeatureViewSource, + FeatureViewSourceParam, +) +from databricks.bundles.features._models.kafka_source import ( + KafkaSource, + KafkaSourceParam, +) +from databricks.bundles.features._models.request_source import ( + RequestSource, + RequestSourceParam, +) +from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceParam +from databricks.bundles.features._models.source_lateness import ( + SourceLateness, + SourceLatenessParam, +) +from databricks.bundles.features._models.stream_source import ( + StreamSource, + StreamSourceParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DataSource: """ Specifies the data source backing a feature. Exactly one source type must be set. """ + delta_table_source: VariableOrOptional[DeltaTableSource] = None """ A Delta table data source. @@ -106,12 +85,11 @@ class DataSource: """ @classmethod - def from_dict(cls, value: 'DataSourceDict') -> 'Self': + def from_dict(cls, value: "DataSourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DataSourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DataSourceDict": + return _transform_to_json_value(self) # type:ignore class DataSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/delta_table_source.py b/python/databricks/bundles/features/_models/delta_table_source.py index a91c4dba2cc..3caedb40fa6 100644 --- a/python/databricks/bundles/features/_models/delta_table_source.py +++ b/python/databricks/bundles/features/_models/delta_table_source.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DeltaTableSource: """""" @@ -87,12 +40,11 @@ class DeltaTableSource: """ @classmethod - def from_dict(cls, value: 'DeltaTableSourceDict') -> 'Self': + def from_dict(cls, value: "DeltaTableSourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DeltaTableSourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DeltaTableSourceDict": + return _transform_to_json_value(self) # type:ignore class DeltaTableSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/entity_column.py b/python/databricks/bundles/features/_models/entity_column.py index d81033efd1e..d460e9ef4e2 100644 --- a/python/databricks/bundles/features/_models/entity_column.py +++ b/python/databricks/bundles/features/_models/entity_column.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EntityColumn: """""" @@ -73,12 +26,11 @@ class EntityColumn: """ @classmethod - def from_dict(cls, value: 'EntityColumnDict') -> 'Self': + def from_dict(cls, value: "EntityColumnDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EntityColumnDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EntityColumnDict": + return _transform_to_json_value(self) # type:ignore class EntityColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/feature.py b/python/databricks/bundles/features/_models/feature.py index d7e382c106e..d00ed799f92 100644 --- a/python/databricks/bundles/features/_models/feature.py +++ b/python/databricks/bundles/features/_models/feature.py @@ -1,63 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.features._models.data_source import ( + DataSource, + DataSourceParam, +) +from databricks.bundles.features._models.entity_column import ( + EntityColumn, + EntityColumnParam, +) +from databricks.bundles.features._models.function import ( + Function, + FunctionParam, +) +from databricks.bundles.features._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.features._models.lineage_context import ( + LineageContext, + LineageContextParam, +) +from databricks.bundles.features._models.timeseries_column import ( + TimeseriesColumn, + TimeseriesColumnParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Feature(Resource): """""" @@ -109,12 +91,11 @@ class Feature(Resource): """ @classmethod - def from_dict(cls, value: 'FeatureDict') -> 'Self': + def from_dict(cls, value: "FeatureDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FeatureDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FeatureDict": + return _transform_to_json_value(self) # type:ignore class FeatureDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/feature_reference.py b/python/databricks/bundles/features/_models/feature_reference.py index c7d40731e2f..21decdf1547 100644 --- a/python/databricks/bundles/features/_models/feature_reference.py +++ b/python/databricks/bundles/features/_models/feature_reference.py @@ -1,71 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FeatureReference: """ :meta private: [EXPERIMENTAL] - + A reference to one registered upstream Feature. A message rather than a bare name so an upstream can later be pinned more precisely (e.g. by version) without a breaking type change. """ + feature: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -74,12 +28,11 @@ class FeatureReference: """ @classmethod - def from_dict(cls, value: 'FeatureReferenceDict') -> 'Self': + def from_dict(cls, value: "FeatureReferenceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FeatureReferenceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FeatureReferenceDict": + return _transform_to_json_value(self) # type:ignore class FeatureReferenceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/feature_view_source.py b/python/databricks/bundles/features/_models/feature_view_source.py index a23d17c8b72..bca40ffcafa 100644 --- a/python/databricks/bundles/features/_models/feature_view_source.py +++ b/python/databricks/bundles/features/_models/feature_view_source.py @@ -1,70 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.features._models.feature_reference import ( + FeatureReference, + FeatureReferenceParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FeatureViewSource: """ :meta private: [EXPERIMENTAL] - + A data source composed from registered upstream Features. """ + feature_references: VariableOrList[FeatureReference] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -73,12 +31,11 @@ class FeatureViewSource: """ @classmethod - def from_dict(cls, value: 'FeatureViewSourceDict') -> 'Self': + def from_dict(cls, value: "FeatureViewSourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FeatureViewSourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FeatureViewSourceDict": + return _transform_to_json_value(self) # type:ignore class FeatureViewSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/field_definition.py b/python/databricks/bundles/features/_models/field_definition.py index ea64a48d055..3087c757232 100644 --- a/python/databricks/bundles/features/_models/field_definition.py +++ b/python/databricks/bundles/features/_models/field_definition.py @@ -1,69 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr +from databricks.bundles.features._models.scalar_data_type import ( + ScalarDataType, + ScalarDataTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FieldDefinition: """ A single field definition within a FlatSchema, specifying the field name and its scalar data type. Does not support nested or complex types (arrays, maps, structs). """ + data_type: VariableOr[ScalarDataType] """ The scalar data type of the field. @@ -75,12 +33,11 @@ class FieldDefinition: """ @classmethod - def from_dict(cls, value: 'FieldDefinitionDict') -> 'Self': + def from_dict(cls, value: "FieldDefinitionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FieldDefinitionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FieldDefinitionDict": + return _transform_to_json_value(self) # type:ignore class FieldDefinitionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/first_distinct_function.py b/python/databricks/bundles/features/_models/first_distinct_function.py index 801a2a4232b..87d460fa228 100644 --- a/python/databricks/bundles/features/_models/first_distinct_function.py +++ b/python/databricks/bundles/features/_models/first_distinct_function.py @@ -1,70 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FirstDistinctFunction: """ :meta private: [EXPERIMENTAL] - + Returns the first N distinct values, ordered by the feature's timeseries column. """ + input: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -80,12 +34,11 @@ class FirstDistinctFunction: """ @classmethod - def from_dict(cls, value: 'FirstDistinctFunctionDict') -> 'Self': + def from_dict(cls, value: "FirstDistinctFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FirstDistinctFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FirstDistinctFunctionDict": + return _transform_to_json_value(self) # type:ignore class FirstDistinctFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/first_function.py b/python/databricks/bundles/features/_models/first_function.py index feabbd4ed54..eb4ea4f1d2d 100644 --- a/python/databricks/bundles/features/_models/first_function.py +++ b/python/databricks/bundles/features/_models/first_function.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FirstFunction: """ Returns the first value. """ + input: VariableOr[str] """ The input column from which the first value is returned. """ @classmethod - def from_dict(cls, value: 'FirstFunctionDict') -> 'Self': + def from_dict(cls, value: "FirstFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FirstFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FirstFunctionDict": + return _transform_to_json_value(self) # type:ignore class FirstFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/first_n_function.py b/python/databricks/bundles/features/_models/first_n_function.py index c7158525891..3368c234f68 100644 --- a/python/databricks/bundles/features/_models/first_n_function.py +++ b/python/databricks/bundles/features/_models/first_n_function.py @@ -1,70 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FirstNFunction: """ :meta private: [EXPERIMENTAL] - + Returns the first N values, ordered by the feature's timeseries column. """ + input: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -80,12 +34,11 @@ class FirstNFunction: """ @classmethod - def from_dict(cls, value: 'FirstNFunctionDict') -> 'Self': + def from_dict(cls, value: "FirstNFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FirstNFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FirstNFunctionDict": + return _transform_to_json_value(self) # type:ignore class FirstNFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/flat_schema.py b/python/databricks/bundles/features/_models/flat_schema.py index 603812880c7..0535274cc83 100644 --- a/python/databricks/bundles/features/_models/flat_schema.py +++ b/python/databricks/bundles/features/_models/flat_schema.py @@ -1,81 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.features._models.field_definition import ( + FieldDefinition, + FieldDefinitionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FlatSchema: """ A flat (non-nested) schema for request-time fields, defined as an ordered list of field definitions. This schema only supports scalar types. """ + fields: VariableOrList[FieldDefinition] = field(default_factory=list) """ The list of fields in this schema. """ @classmethod - def from_dict(cls, value: 'FlatSchemaDict') -> 'Self': + def from_dict(cls, value: "FlatSchemaDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FlatSchemaDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FlatSchemaDict": + return _transform_to_json_value(self) # type:ignore class FlatSchemaDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/function.py b/python/databricks/bundles/features/_models/function.py index a803f7e5d2f..c68a8cd0db2 100644 --- a/python/databricks/bundles/features/_models/function.py +++ b/python/databricks/bundles/features/_models/function.py @@ -1,63 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.features._models.aggregation_function import ( + AggregationFunction, + AggregationFunctionParam, +) +from databricks.bundles.features._models.column_selection import ( + ColumnSelection, + ColumnSelectionParam, +) +from databricks.bundles.features._models.custom_udf import ( + CustomUdf, + CustomUdfParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Function: """""" @@ -80,12 +45,11 @@ class Function: """ @classmethod - def from_dict(cls, value: 'FunctionDict') -> 'Self': + def from_dict(cls, value: "FunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FunctionDict": + return _transform_to_json_value(self) # type:ignore class FunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/input_binding.py b/python/databricks/bundles/features/_models/input_binding.py index 88fade3d867..7ade8c9405e 100644 --- a/python/databricks/bundles/features/_models/input_binding.py +++ b/python/databricks/bundles/features/_models/input_binding.py @@ -1,70 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InputBinding: """ :meta private: [EXPERIMENTAL] - + Binds a single UC function parameter to a source column. """ + column: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -80,12 +34,11 @@ class InputBinding: """ @classmethod - def from_dict(cls, value: 'InputBindingDict') -> 'Self': + def from_dict(cls, value: "InputBindingDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InputBindingDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InputBindingDict": + return _transform_to_json_value(self) # type:ignore class InputBindingDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/job_context.py b/python/databricks/bundles/features/_models/job_context.py index 0f1d6880eb1..e2d70d0a0bf 100644 --- a/python/databricks/bundles/features/_models/job_context.py +++ b/python/databricks/bundles/features/_models/job_context.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobContext: """ :meta private: [EXPERIMENTAL] """ + job_id: VariableOrOptional[int] = None """ :meta private: [EXPERIMENTAL] @@ -78,12 +32,11 @@ class JobContext: """ @classmethod - def from_dict(cls, value: 'JobContextDict') -> 'Self': + def from_dict(cls, value: "JobContextDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobContextDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobContextDict": + return _transform_to_json_value(self) # type:ignore class JobContextDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/kafka_source.py b/python/databricks/bundles/features/_models/kafka_source.py index a7febacaadc..a7142a7cacb 100644 --- a/python/databricks/bundles/features/_models/kafka_source.py +++ b/python/databricks/bundles/features/_models/kafka_source.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class KafkaSource: """""" @@ -73,12 +26,11 @@ class KafkaSource: """ @classmethod - def from_dict(cls, value: 'KafkaSourceDict') -> 'Self': + def from_dict(cls, value: "KafkaSourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'KafkaSourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "KafkaSourceDict": + return _transform_to_json_value(self) # type:ignore class KafkaSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/last_distinct_function.py b/python/databricks/bundles/features/_models/last_distinct_function.py index 99ae7e73635..f0ed97454a1 100644 --- a/python/databricks/bundles/features/_models/last_distinct_function.py +++ b/python/databricks/bundles/features/_models/last_distinct_function.py @@ -1,70 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LastDistinctFunction: """ :meta private: [EXPERIMENTAL] - + Returns the last N distinct values, ordered by the feature's timeseries column. """ + input: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -80,12 +34,11 @@ class LastDistinctFunction: """ @classmethod - def from_dict(cls, value: 'LastDistinctFunctionDict') -> 'Self': + def from_dict(cls, value: "LastDistinctFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LastDistinctFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LastDistinctFunctionDict": + return _transform_to_json_value(self) # type:ignore class LastDistinctFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/last_function.py b/python/databricks/bundles/features/_models/last_function.py index 5f89e093c44..56cd70a231d 100644 --- a/python/databricks/bundles/features/_models/last_function.py +++ b/python/databricks/bundles/features/_models/last_function.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LastFunction: """ Returns the last value. """ + input: VariableOr[str] """ The input column from which the last value is returned. """ @classmethod - def from_dict(cls, value: 'LastFunctionDict') -> 'Self': + def from_dict(cls, value: "LastFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LastFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LastFunctionDict": + return _transform_to_json_value(self) # type:ignore class LastFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/last_n_function.py b/python/databricks/bundles/features/_models/last_n_function.py index 8105579bae3..03131e3182f 100644 --- a/python/databricks/bundles/features/_models/last_n_function.py +++ b/python/databricks/bundles/features/_models/last_n_function.py @@ -1,70 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LastNFunction: """ :meta private: [EXPERIMENTAL] - + Returns the last N values, ordered by the feature's timeseries column. """ + input: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -80,12 +34,11 @@ class LastNFunction: """ @classmethod - def from_dict(cls, value: 'LastNFunctionDict') -> 'Self': + def from_dict(cls, value: "LastNFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LastNFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LastNFunctionDict": + return _transform_to_json_value(self) # type:ignore class LastNFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/lifecycle.py b/python/databricks/bundles/features/_models/lifecycle.py index 8c910bb74c2..697776a198e 100644 --- a/python/databricks/bundles/features/_models/lifecycle.py +++ b/python/databricks/bundles/features/_models/lifecycle.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -68,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/lineage_context.py b/python/databricks/bundles/features/_models/lineage_context.py index 8b106bde8e9..d11cb8082d6 100644 --- a/python/databricks/bundles/features/_models/lineage_context.py +++ b/python/databricks/bundles/features/_models/lineage_context.py @@ -1,70 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.features._models.job_context import ( + JobContext, + JobContextParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LineageContext: """ :meta private: [EXPERIMENTAL] - + Lineage context information for tracking where an API was invoked. This will allow us to track lineage, which currently uses caller entity information for use across the Lineage Client and Observability in Lumberjack. """ + job_context: VariableOrOptional[JobContext] = None """ :meta private: [EXPERIMENTAL] @@ -80,12 +38,11 @@ class LineageContext: """ @classmethod - def from_dict(cls, value: 'LineageContextDict') -> 'Self': + def from_dict(cls, value: "LineageContextDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LineageContextDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LineageContextDict": + return _transform_to_json_value(self) # type:ignore class LineageContextDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/max_function.py b/python/databricks/bundles/features/_models/max_function.py index 90d0b0ddb82..dbc7c7f8f81 100644 --- a/python/databricks/bundles/features/_models/max_function.py +++ b/python/databricks/bundles/features/_models/max_function.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MaxFunction: """ Computes the maximum value. """ + input: VariableOr[str] """ The input column from which the maximum is computed. """ @classmethod - def from_dict(cls, value: 'MaxFunctionDict') -> 'Self': + def from_dict(cls, value: "MaxFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MaxFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MaxFunctionDict": + return _transform_to_json_value(self) # type:ignore class MaxFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/min_function.py b/python/databricks/bundles/features/_models/min_function.py index cad2d927012..17bebfdbc41 100644 --- a/python/databricks/bundles/features/_models/min_function.py +++ b/python/databricks/bundles/features/_models/min_function.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MinFunction: """ Computes the minimum value. """ + input: VariableOr[str] """ The input column from which the minimum is computed. """ @classmethod - def from_dict(cls, value: 'MinFunctionDict') -> 'Self': + def from_dict(cls, value: "MinFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MinFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MinFunctionDict": + return _transform_to_json_value(self) # type:ignore class MinFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/request_source.py b/python/databricks/bundles/features/_models/request_source.py index 4ad8edd8750..2ad0c4a0225 100644 --- a/python/databricks/bundles/features/_models/request_source.py +++ b/python/databricks/bundles/features/_models/request_source.py @@ -1,68 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaParam if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RequestSource: """ A request-time data source whose value is provided at inference time: offline batch scoring or online serving endpoint """ + dataframe_schema: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -77,12 +32,11 @@ class RequestSource: """ @classmethod - def from_dict(cls, value: 'RequestSourceDict') -> 'Self': + def from_dict(cls, value: "RequestSourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RequestSourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RequestSourceDict": + return _transform_to_json_value(self) # type:ignore class RequestSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/rolling_window.py b/python/databricks/bundles/features/_models/rolling_window.py index 9604168e1c8..9cbdd0b7c2f 100644 --- a/python/databricks/bundles/features/_models/rolling_window.py +++ b/python/databricks/bundles/features/_models/rolling_window.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RollingWindow: """ A rolling time window with an optional non-negative delay. """ + delay: VariableOrOptional[str] = None """ Non-negative analytic lag that evaluates the window this far in the past. Use this for timing @@ -77,12 +31,11 @@ class RollingWindow: """ @classmethod - def from_dict(cls, value: 'RollingWindowDict') -> 'Self': + def from_dict(cls, value: "RollingWindowDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RollingWindowDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RollingWindowDict": + return _transform_to_json_value(self) # type:ignore class RollingWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/sawtooth_window.py b/python/databricks/bundles/features/_models/sawtooth_window.py index ba1de2258c4..303ab073d46 100644 --- a/python/databricks/bundles/features/_models/sawtooth_window.py +++ b/python/databricks/bundles/features/_models/sawtooth_window.py @@ -1,74 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SawtoothWindow: """ :meta private: [EXPERIMENTAL] - + A sawtooth window served via the hybrid batch + streaming path. The batch pipeline maintains daily partial aggregates for the bulk of the window while the streaming pipeline maintains the most recent day(s), and serving merges them on read. Same field shape as RollingWindow, but a distinct type so the control plane can explicitly identify hybrid (sawtooth) features rather than inferring hybrid behavior from window_duration. """ + delay: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -87,12 +41,11 @@ class SawtoothWindow: """ @classmethod - def from_dict(cls, value: 'SawtoothWindowDict') -> 'Self': + def from_dict(cls, value: "SawtoothWindowDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SawtoothWindowDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SawtoothWindowDict": + return _transform_to_json_value(self) # type:ignore class SawtoothWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/scalar_data_type.py b/python/databricks/bundles/features/_models/scalar_data_type.py index e673f94d9ce..5c115502b0a 100644 --- a/python/databricks/bundles/features/_models/scalar_data_type.py +++ b/python/databricks/bundles/features/_models/scalar_data_type.py @@ -1,64 +1,15 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam class ScalarDataType(Enum): """ Scalar data types for request-time field definitions. Only flat (non-nested) types are supported. """ + INTEGER = "INTEGER" FLOAT = "FLOAT" BOOLEAN = "BOOLEAN" @@ -71,4 +22,20 @@ class ScalarDataType(Enum): BINARY = "BINARY" DECIMAL = "DECIMAL" -ScalarDataTypeParam = Literal["INTEGER", "FLOAT", "BOOLEAN", "STRING", "DOUBLE", "LONG", "TIMESTAMP", "DATE", "SHORT", "BINARY", "DECIMAL"] | ScalarDataType + +ScalarDataTypeParam = ( + Literal[ + "INTEGER", + "FLOAT", + "BOOLEAN", + "STRING", + "DOUBLE", + "LONG", + "TIMESTAMP", + "DATE", + "SHORT", + "BINARY", + "DECIMAL", + ] + | ScalarDataType +) diff --git a/python/databricks/bundles/features/_models/sdp_source.py b/python/databricks/bundles/features/_models/sdp_source.py index 0c983aa393d..235df7bcc55 100644 --- a/python/databricks/bundles/features/_models/sdp_source.py +++ b/python/databricks/bundles/features/_models/sdp_source.py @@ -1,70 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SdpSource: """ :meta private: [EXPERIMENTAL] - + A Spark Declarative Pipeline (SDP) data source. """ + dataframe_schema: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -120,12 +74,11 @@ class SdpSource: """ @classmethod - def from_dict(cls, value: 'SdpSourceDict') -> 'Self': + def from_dict(cls, value: "SdpSourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SdpSourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SdpSourceDict": + return _transform_to_json_value(self) # type:ignore class SdpSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/sliding_window.py b/python/databricks/bundles/features/_models/sliding_window.py index ea4f1736c7e..cdf67ed2af0 100644 --- a/python/databricks/bundles/features/_models/sliding_window.py +++ b/python/databricks/bundles/features/_models/sliding_window.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SlidingWindow: """""" @@ -93,12 +46,11 @@ class SlidingWindow: """ @classmethod - def from_dict(cls, value: 'SlidingWindowDict') -> 'Self': + def from_dict(cls, value: "SlidingWindowDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SlidingWindowDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SlidingWindowDict": + return _transform_to_json_value(self) # type:ignore class SlidingWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/source_lateness.py b/python/databricks/bundles/features/_models/source_lateness.py index a88758b38f6..757eac320e8 100644 --- a/python/databricks/bundles/features/_models/source_lateness.py +++ b/python/databricks/bundles/features/_models/source_lateness.py @@ -1,70 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SourceLateness: """ :meta private: [EXPERIMENTAL] - + Configures when event-time data from this source is considered complete for a Feature. """ + settling_delay: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -76,12 +30,11 @@ class SourceLateness: """ @classmethod - def from_dict(cls, value: 'SourceLatenessDict') -> 'Self': + def from_dict(cls, value: "SourceLatenessDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SourceLatenessDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SourceLatenessDict": + return _transform_to_json_value(self) # type:ignore class SourceLatenessDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/stddev_pop_function.py b/python/databricks/bundles/features/_models/stddev_pop_function.py index 319eeb1d5dd..edd63b11b93 100644 --- a/python/databricks/bundles/features/_models/stddev_pop_function.py +++ b/python/databricks/bundles/features/_models/stddev_pop_function.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class StddevPopFunction: """ Computes the population standard deviation. """ + input: VariableOr[str] """ The input column from which the population standard deviation is computed. For Kafka sources, @@ -72,12 +26,11 @@ class StddevPopFunction: """ @classmethod - def from_dict(cls, value: 'StddevPopFunctionDict') -> 'Self': + def from_dict(cls, value: "StddevPopFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'StddevPopFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "StddevPopFunctionDict": + return _transform_to_json_value(self) # type:ignore class StddevPopFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/stddev_samp_function.py b/python/databricks/bundles/features/_models/stddev_samp_function.py index bca039fbbaa..ca253c92315 100644 --- a/python/databricks/bundles/features/_models/stddev_samp_function.py +++ b/python/databricks/bundles/features/_models/stddev_samp_function.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class StddevSampFunction: """ Computes the sample standard deviation. """ + input: VariableOr[str] """ The input column from which the sample standard deviation is computed. """ @classmethod - def from_dict(cls, value: 'StddevSampFunctionDict') -> 'Self': + def from_dict(cls, value: "StddevSampFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'StddevSampFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "StddevSampFunctionDict": + return _transform_to_json_value(self) # type:ignore class StddevSampFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/stream_source.py b/python/databricks/bundles/features/_models/stream_source.py index e803b243641..7b591576426 100644 --- a/python/databricks/bundles/features/_models/stream_source.py +++ b/python/databricks/bundles/features/_models/stream_source.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class StreamSource: """ A Stream entity used as a data source for a feature. """ + full_name: VariableOr[str] """ Three-part full name of the Stream (catalog.schema.stream). @@ -91,12 +45,11 @@ class StreamSource: """ @classmethod - def from_dict(cls, value: 'StreamSourceDict') -> 'Self': + def from_dict(cls, value: "StreamSourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'StreamSourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "StreamSourceDict": + return _transform_to_json_value(self) # type:ignore class StreamSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/sum_function.py b/python/databricks/bundles/features/_models/sum_function.py index 3839fd0f349..f74c3e8a6ac 100644 --- a/python/databricks/bundles/features/_models/sum_function.py +++ b/python/databricks/bundles/features/_models/sum_function.py @@ -1,68 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SumFunction: """ Computes the sum of values. """ + input: VariableOr[str] """ The input column from which the sum is computed. For Kafka sources, use dot-prefixed path @@ -72,12 +26,11 @@ class SumFunction: """ @classmethod - def from_dict(cls, value: 'SumFunctionDict') -> 'Self': + def from_dict(cls, value: "SumFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SumFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SumFunctionDict": + return _transform_to_json_value(self) # type:ignore class SumFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/time_window.py b/python/databricks/bundles/features/_models/time_window.py index 16c6f51b3ba..85bd6a488dd 100644 --- a/python/databricks/bundles/features/_models/time_window.py +++ b/python/databricks/bundles/features/_models/time_window.py @@ -1,63 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.features._models.rolling_window import ( + RollingWindow, + RollingWindowParam, +) +from databricks.bundles.features._models.sawtooth_window import ( + SawtoothWindow, + SawtoothWindowParam, +) +from databricks.bundles.features._models.sliding_window import ( + SlidingWindow, + SlidingWindowParam, +) +from databricks.bundles.features._models.tumbling_window import ( + TumblingWindow, + TumblingWindowParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TimeWindow: """""" @@ -93,12 +62,11 @@ class TimeWindow: tumbling: VariableOrOptional[TumblingWindow] = None @classmethod - def from_dict(cls, value: 'TimeWindowDict') -> 'Self': + def from_dict(cls, value: "TimeWindowDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TimeWindowDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TimeWindowDict": + return _transform_to_json_value(self) # type:ignore class TimeWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/timeseries_column.py b/python/databricks/bundles/features/_models/timeseries_column.py index 740439aaf52..be19cf7dad0 100644 --- a/python/databricks/bundles/features/_models/timeseries_column.py +++ b/python/databricks/bundles/features/_models/timeseries_column.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TimeseriesColumn: """""" @@ -73,12 +26,11 @@ class TimeseriesColumn: """ @classmethod - def from_dict(cls, value: 'TimeseriesColumnDict') -> 'Self': + def from_dict(cls, value: "TimeseriesColumnDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TimeseriesColumnDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TimeseriesColumnDict": + return _transform_to_json_value(self) # type:ignore class TimeseriesColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/tumbling_window.py b/python/databricks/bundles/features/_models/tumbling_window.py index fc28ae9c5e0..38a873fa5cd 100644 --- a/python/databricks/bundles/features/_models/tumbling_window.py +++ b/python/databricks/bundles/features/_models/tumbling_window.py @@ -1,63 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TumblingWindow: """""" @@ -86,12 +39,11 @@ class TumblingWindow: """ @classmethod - def from_dict(cls, value: 'TumblingWindowDict') -> 'Self': + def from_dict(cls, value: "TumblingWindowDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TumblingWindowDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TumblingWindowDict": + return _transform_to_json_value(self) # type:ignore class TumblingWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/var_pop_function.py b/python/databricks/bundles/features/_models/var_pop_function.py index fa68523a468..79b6dc5f425 100644 --- a/python/databricks/bundles/features/_models/var_pop_function.py +++ b/python/databricks/bundles/features/_models/var_pop_function.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_samp_function import VarSampFunction, VarSampFunctionDict, VarSampFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class VarPopFunction: """ Computes the population variance. """ + input: VariableOr[str] """ The input column from which the population variance is computed. """ @classmethod - def from_dict(cls, value: 'VarPopFunctionDict') -> 'Self': + def from_dict(cls, value: "VarPopFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'VarPopFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "VarPopFunctionDict": + return _transform_to_json_value(self) # type:ignore class VarPopFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/features/_models/var_samp_function.py b/python/databricks/bundles/features/_models/var_samp_function.py index 4ede4549412..8f0ee87ce77 100644 --- a/python/databricks/bundles/features/_models/var_samp_function.py +++ b/python/databricks/bundles/features/_models/var_samp_function.py @@ -1,80 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.features._models.aggregation_function import AggregationFunction, AggregationFunctionDict, AggregationFunctionParam -from databricks.bundles.features._models.approx_count_distinct_function import ApproxCountDistinctFunction, ApproxCountDistinctFunctionDict, ApproxCountDistinctFunctionParam -from databricks.bundles.features._models.approx_percentile_function import ApproxPercentileFunction, ApproxPercentileFunctionDict, ApproxPercentileFunctionParam -from databricks.bundles.features._models.avg_function import AvgFunction, AvgFunctionDict, AvgFunctionParam -from databricks.bundles.features._models.column_selection import ColumnSelection, ColumnSelectionDict, ColumnSelectionParam -from databricks.bundles.features._models.count_function import CountFunction, CountFunctionDict, CountFunctionParam -from databricks.bundles.features._models.custom_udf import CustomUdf, CustomUdfDict, CustomUdfParam -from databricks.bundles.features._models.data_source import DataSource, DataSourceDict, DataSourceParam -from databricks.bundles.features._models.delta_table_source import DeltaTableSource, DeltaTableSourceDict, DeltaTableSourceParam -from databricks.bundles.features._models.entity_column import EntityColumn, EntityColumnDict, EntityColumnParam -from databricks.bundles.features._models.feature_reference import FeatureReference, FeatureReferenceDict, FeatureReferenceParam -from databricks.bundles.features._models.feature_view_source import FeatureViewSource, FeatureViewSourceDict, FeatureViewSourceParam -from databricks.bundles.features._models.field_definition import FieldDefinition, FieldDefinitionDict, FieldDefinitionParam -from databricks.bundles.features._models.first_distinct_function import FirstDistinctFunction, FirstDistinctFunctionDict, FirstDistinctFunctionParam -from databricks.bundles.features._models.first_function import FirstFunction, FirstFunctionDict, FirstFunctionParam -from databricks.bundles.features._models.first_n_function import FirstNFunction, FirstNFunctionDict, FirstNFunctionParam -from databricks.bundles.features._models.flat_schema import FlatSchema, FlatSchemaDict, FlatSchemaParam -from databricks.bundles.features._models.function import Function, FunctionDict, FunctionParam -from databricks.bundles.features._models.input_binding import InputBinding, InputBindingDict, InputBindingParam -from databricks.bundles.features._models.job_context import JobContext, JobContextDict, JobContextParam -from databricks.bundles.features._models.kafka_source import KafkaSource, KafkaSourceDict, KafkaSourceParam -from databricks.bundles.features._models.last_distinct_function import LastDistinctFunction, LastDistinctFunctionDict, LastDistinctFunctionParam -from databricks.bundles.features._models.last_function import LastFunction, LastFunctionDict, LastFunctionParam -from databricks.bundles.features._models.last_n_function import LastNFunction, LastNFunctionDict, LastNFunctionParam -from databricks.bundles.features._models.lineage_context import LineageContext, LineageContextDict, LineageContextParam -from databricks.bundles.features._models.max_function import MaxFunction, MaxFunctionDict, MaxFunctionParam -from databricks.bundles.features._models.min_function import MinFunction, MinFunctionDict, MinFunctionParam -from databricks.bundles.features._models.request_source import RequestSource, RequestSourceDict, RequestSourceParam -from databricks.bundles.features._models.rolling_window import RollingWindow, RollingWindowDict, RollingWindowParam -from databricks.bundles.features._models.sawtooth_window import SawtoothWindow, SawtoothWindowDict, SawtoothWindowParam -from databricks.bundles.features._models.sdp_source import SdpSource, SdpSourceDict, SdpSourceParam -from databricks.bundles.features._models.sliding_window import SlidingWindow, SlidingWindowDict, SlidingWindowParam -from databricks.bundles.features._models.source_lateness import SourceLateness, SourceLatenessDict, SourceLatenessParam -from databricks.bundles.features._models.stddev_pop_function import StddevPopFunction, StddevPopFunctionDict, StddevPopFunctionParam -from databricks.bundles.features._models.stddev_samp_function import StddevSampFunction, StddevSampFunctionDict, StddevSampFunctionParam -from databricks.bundles.features._models.stream_source import StreamSource, StreamSourceDict, StreamSourceParam -from databricks.bundles.features._models.sum_function import SumFunction, SumFunctionDict, SumFunctionParam -from databricks.bundles.features._models.time_window import TimeWindow, TimeWindowDict, TimeWindowParam -from databricks.bundles.features._models.timeseries_column import TimeseriesColumn, TimeseriesColumnDict, TimeseriesColumnParam -from databricks.bundles.features._models.tumbling_window import TumblingWindow, TumblingWindowDict, TumblingWindowParam -from databricks.bundles.features._models.var_pop_function import VarPopFunction, VarPopFunctionDict, VarPopFunctionParam -from databricks.bundles.features._models.feature import Feature, FeatureDict, FeatureParam -from databricks.bundles.features._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.features._models.scalar_data_type import ScalarDataType, ScalarDataTypeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class VarSampFunction: """ Computes the sample variance. """ + input: VariableOr[str] """ The input column from which the sample variance is computed. """ @classmethod - def from_dict(cls, value: 'VarSampFunctionDict') -> 'Self': + def from_dict(cls, value: "VarSampFunctionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'VarSampFunctionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "VarSampFunctionDict": + return _transform_to_json_value(self) # type:ignore class VarSampFunctionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/genie_spaces/__init__.py b/python/databricks/bundles/genie_spaces/__init__.py index 6b12939bf92..a56af37c716 100644 --- a/python/databricks/bundles/genie_spaces/__init__.py +++ b/python/databricks/bundles/genie_spaces/__init__.py @@ -1,22 +1,36 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'GenieSpace', - 'GenieSpaceDict', - 'GenieSpaceParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'Permission', - 'PermissionDict', - 'PermissionLevel', - 'PermissionLevelParam', - 'PermissionParam', + "GenieSpace", + "GenieSpaceDict", + "GenieSpaceParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "Permission", + "PermissionDict", + "PermissionLevel", + "PermissionLevelParam", + "PermissionParam", ] -from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceDict, GenieSpaceParam -from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.genie_spaces._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.genie_spaces._models.genie_space import ( + GenieSpace, + GenieSpaceDict, + GenieSpaceParam, +) +from databricks.bundles.genie_spaces._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.genie_spaces._models.permission import ( + Permission, + PermissionDict, + PermissionParam, +) +from databricks.bundles.genie_spaces._models.permission_level import ( + PermissionLevel, + PermissionLevelParam, +) diff --git a/python/databricks/bundles/genie_spaces/_models/genie_space.py b/python/databricks/bundles/genie_spaces/_models/genie_space.py index a481f6bd59d..bb1ec94c8df 100644 --- a/python/databricks/bundles/genie_spaces/_models/genie_space.py +++ b/python/databricks/bundles/genie_spaces/_models/genie_space.py @@ -1,22 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, Any, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.genie_spaces._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.genie_spaces._models.permission import ( + Permission, + PermissionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GenieSpace(Resource): """""" @@ -64,12 +64,11 @@ class GenieSpace(Resource): """ @classmethod - def from_dict(cls, value: 'GenieSpaceDict') -> 'Self': + def from_dict(cls, value: "GenieSpaceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GenieSpaceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GenieSpaceDict": + return _transform_to_json_value(self) # type:ignore class GenieSpaceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/genie_spaces/_models/lifecycle.py b/python/databricks/bundles/genie_spaces/_models/lifecycle.py index b5ad6018a8c..697776a198e 100644 --- a/python/databricks/bundles/genie_spaces/_models/lifecycle.py +++ b/python/databricks/bundles/genie_spaces/_models/lifecycle.py @@ -1,22 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceDict, GenieSpaceParam -from databricks.bundles.genie_spaces._models.permission import Permission, PermissionDict, PermissionParam -from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -27,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/genie_spaces/_models/permission.py b/python/databricks/bundles/genie_spaces/_models/permission.py index 3984cc95652..372f063d547 100644 --- a/python/databricks/bundles/genie_spaces/_models/permission.py +++ b/python/databricks/bundles/genie_spaces/_models/permission.py @@ -1,22 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceDict, GenieSpaceParam -from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel, PermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.genie_spaces._models.permission_level import ( + PermissionLevel, + PermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Permission: """""" @@ -42,12 +40,11 @@ class Permission: """ @classmethod - def from_dict(cls, value: 'PermissionDict') -> 'Self': + def from_dict(cls, value: "PermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PermissionDict": + return _transform_to_json_value(self) # type:ignore class PermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/genie_spaces/_models/permission_level.py b/python/databricks/bundles/genie_spaces/_models/permission_level.py index 74f0ee9d40a..c6111911b29 100644 --- a/python/databricks/bundles/genie_spaces/_models/permission_level.py +++ b/python/databricks/bundles/genie_spaces/_models/permission_level.py @@ -1,22 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.genie_spaces._models.genie_space import GenieSpace, GenieSpaceDict, GenieSpaceParam -from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.genie_spaces._models.permission import Permission, PermissionDict, PermissionParam class PermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_RESTART = "CAN_RESTART" CAN_ATTACH_TO = "CAN_ATTACH_TO" @@ -38,4 +30,29 @@ class PermissionLevel(Enum): CAN_MONITOR_ONLY = "CAN_MONITOR_ONLY" CAN_CREATE_APP = "CAN_CREATE_APP" -PermissionLevelParam = Literal["CAN_MANAGE", "CAN_RESTART", "CAN_ATTACH_TO", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW", "CAN_READ", "CAN_RUN", "CAN_EDIT", "CAN_USE", "CAN_MANAGE_STAGING_VERSIONS", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_EDIT_METADATA", "CAN_VIEW_METADATA", "CAN_BIND", "CAN_QUERY", "CAN_MONITOR", "CAN_CREATE", "CAN_MONITOR_ONLY", "CAN_CREATE_APP"] | PermissionLevel + +PermissionLevelParam = ( + Literal[ + "CAN_MANAGE", + "CAN_RESTART", + "CAN_ATTACH_TO", + "IS_OWNER", + "CAN_MANAGE_RUN", + "CAN_VIEW", + "CAN_READ", + "CAN_RUN", + "CAN_EDIT", + "CAN_USE", + "CAN_MANAGE_STAGING_VERSIONS", + "CAN_MANAGE_PRODUCTION_VERSIONS", + "CAN_EDIT_METADATA", + "CAN_VIEW_METADATA", + "CAN_BIND", + "CAN_QUERY", + "CAN_MONITOR", + "CAN_CREATE", + "CAN_MONITOR_ONLY", + "CAN_CREATE_APP", + ] + | PermissionLevel +) diff --git a/python/databricks/bundles/instance_pools/__init__.py b/python/databricks/bundles/instance_pools/__init__.py index 5e699b4a170..2c5a0e1c4eb 100644 --- a/python/databricks/bundles/instance_pools/__init__.py +++ b/python/databricks/bundles/instance_pools/__init__.py @@ -1,74 +1,130 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'DiskSpec', - 'DiskSpecDict', - 'DiskSpecParam', - 'DiskType', - 'DiskTypeAzureDiskVolumeType', - 'DiskTypeAzureDiskVolumeTypeParam', - 'DiskTypeDict', - 'DiskTypeEbsVolumeType', - 'DiskTypeEbsVolumeTypeParam', - 'DiskTypeParam', - 'DockerBasicAuth', - 'DockerBasicAuthDict', - 'DockerBasicAuthParam', - 'DockerImage', - 'DockerImageDict', - 'DockerImageParam', - 'GcpAvailability', - 'GcpAvailabilityParam', - 'InstancePool', - 'InstancePoolAwsAttributes', - 'InstancePoolAwsAttributesAvailability', - 'InstancePoolAwsAttributesAvailabilityParam', - 'InstancePoolAwsAttributesDict', - 'InstancePoolAwsAttributesParam', - 'InstancePoolAzureAttributes', - 'InstancePoolAzureAttributesAvailability', - 'InstancePoolAzureAttributesAvailabilityParam', - 'InstancePoolAzureAttributesDict', - 'InstancePoolAzureAttributesParam', - 'InstancePoolDict', - 'InstancePoolGcpAttributes', - 'InstancePoolGcpAttributesDict', - 'InstancePoolGcpAttributesParam', - 'InstancePoolParam', - 'InstancePoolPermission', - 'InstancePoolPermissionDict', - 'InstancePoolPermissionLevel', - 'InstancePoolPermissionLevelParam', - 'InstancePoolPermissionParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'NodeTypeFlexibility', - 'NodeTypeFlexibilityDict', - 'NodeTypeFlexibilityParam', + "DiskSpec", + "DiskSpecDict", + "DiskSpecParam", + "DiskType", + "DiskTypeAzureDiskVolumeType", + "DiskTypeAzureDiskVolumeTypeParam", + "DiskTypeDict", + "DiskTypeEbsVolumeType", + "DiskTypeEbsVolumeTypeParam", + "DiskTypeParam", + "DockerBasicAuth", + "DockerBasicAuthDict", + "DockerBasicAuthParam", + "DockerImage", + "DockerImageDict", + "DockerImageParam", + "GcpAvailability", + "GcpAvailabilityParam", + "InstancePool", + "InstancePoolAwsAttributes", + "InstancePoolAwsAttributesAvailability", + "InstancePoolAwsAttributesAvailabilityParam", + "InstancePoolAwsAttributesDict", + "InstancePoolAwsAttributesParam", + "InstancePoolAzureAttributes", + "InstancePoolAzureAttributesAvailability", + "InstancePoolAzureAttributesAvailabilityParam", + "InstancePoolAzureAttributesDict", + "InstancePoolAzureAttributesParam", + "InstancePoolDict", + "InstancePoolGcpAttributes", + "InstancePoolGcpAttributesDict", + "InstancePoolGcpAttributesParam", + "InstancePoolParam", + "InstancePoolPermission", + "InstancePoolPermissionDict", + "InstancePoolPermissionLevel", + "InstancePoolPermissionLevelParam", + "InstancePoolPermissionParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "NodeTypeFlexibility", + "NodeTypeFlexibilityDict", + "NodeTypeFlexibilityParam", ] -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.instance_pools._models.disk_spec import ( + DiskSpec, + DiskSpecDict, + DiskSpecParam, +) +from databricks.bundles.instance_pools._models.disk_type import ( + DiskType, + DiskTypeDict, + DiskTypeParam, +) +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import ( + DiskTypeAzureDiskVolumeType, + DiskTypeAzureDiskVolumeTypeParam, +) +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import ( + DiskTypeEbsVolumeType, + DiskTypeEbsVolumeTypeParam, +) +from databricks.bundles.instance_pools._models.docker_basic_auth import ( + DockerBasicAuth, + DockerBasicAuthDict, + DockerBasicAuthParam, +) +from databricks.bundles.instance_pools._models.docker_image import ( + DockerImage, + DockerImageDict, + DockerImageParam, +) +from databricks.bundles.instance_pools._models.gcp_availability import ( + GcpAvailability, + GcpAvailabilityParam, +) +from databricks.bundles.instance_pools._models.instance_pool import ( + InstancePool, + InstancePoolDict, + InstancePoolParam, +) +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import ( + InstancePoolAwsAttributes, + InstancePoolAwsAttributesDict, + InstancePoolAwsAttributesParam, +) +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import ( + InstancePoolAwsAttributesAvailability, + InstancePoolAwsAttributesAvailabilityParam, +) +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import ( + InstancePoolAzureAttributes, + InstancePoolAzureAttributesDict, + InstancePoolAzureAttributesParam, +) +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import ( + InstancePoolAzureAttributesAvailability, + InstancePoolAzureAttributesAvailabilityParam, +) +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import ( + InstancePoolGcpAttributes, + InstancePoolGcpAttributesDict, + InstancePoolGcpAttributesParam, +) +from databricks.bundles.instance_pools._models.instance_pool_permission import ( + InstancePoolPermission, + InstancePoolPermissionDict, + InstancePoolPermissionParam, +) +from databricks.bundles.instance_pools._models.instance_pool_permission_level import ( + InstancePoolPermissionLevel, + InstancePoolPermissionLevelParam, +) +from databricks.bundles.instance_pools._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.instance_pools._models.node_type_flexibility import ( + NodeTypeFlexibility, + NodeTypeFlexibilityDict, + NodeTypeFlexibilityParam, +) diff --git a/python/databricks/bundles/instance_pools/_models/disk_spec.py b/python/databricks/bundles/instance_pools/_models/disk_spec.py index 335ff6bc912..b9edbb5d4e8 100644 --- a/python/databricks/bundles/instance_pools/_models/disk_spec.py +++ b/python/databricks/bundles/instance_pools/_models/disk_spec.py @@ -1,40 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeParam if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DiskSpec: """ @@ -43,6 +20,7 @@ class DiskSpec: 2 disks, 100 GiB each, then Databricks will launch a total of 6 disks, 100 GiB each, for this cluster. """ + disk_count: VariableOrOptional[int] = None """ The number of disks launched for each instance: @@ -94,12 +72,11 @@ class DiskSpec: """ @classmethod - def from_dict(cls, value: 'DiskSpecDict') -> 'Self': + def from_dict(cls, value: "DiskSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DiskSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DiskSpecDict": + return _transform_to_json_value(self) # type:ignore class DiskSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/disk_type.py b/python/databricks/bundles/instance_pools/_models/disk_type.py index d5a3b85e516..752bf75bf78 100644 --- a/python/databricks/bundles/instance_pools/_models/disk_type.py +++ b/python/databricks/bundles/instance_pools/_models/disk_type.py @@ -1,45 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import ( + DiskTypeAzureDiskVolumeType, + DiskTypeAzureDiskVolumeTypeParam, +) +from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import ( + DiskTypeEbsVolumeType, + DiskTypeEbsVolumeTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DiskType: """ Describes the disk type. """ + azure_disk_volume_type: VariableOrOptional[DiskTypeAzureDiskVolumeType] = None """ All Azure Disk types that Databricks supports. @@ -53,12 +38,11 @@ class DiskType: """ @classmethod - def from_dict(cls, value: 'DiskTypeDict') -> 'Self': + def from_dict(cls, value: "DiskTypeDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DiskTypeDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DiskTypeDict": + return _transform_to_json_value(self) # type:ignore class DiskTypeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/disk_type_azure_disk_volume_type.py b/python/databricks/bundles/instance_pools/_models/disk_type_azure_disk_volume_type.py index 4114d62c59d..740ab2dd245 100644 --- a/python/databricks/bundles/instance_pools/_models/disk_type_azure_disk_volume_type.py +++ b/python/databricks/bundles/instance_pools/_models/disk_type_azure_disk_volume_type.py @@ -1,34 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam +from typing import Literal class DiskTypeAzureDiskVolumeType(Enum): @@ -36,7 +9,11 @@ class DiskTypeAzureDiskVolumeType(Enum): All Azure Disk types that Databricks supports. See https://docs.microsoft.com/en-us/azure/storage/storage-about-disks-and-vhds-linux#types-of-disks """ + PREMIUM_LRS = "PREMIUM_LRS" STANDARD_LRS = "STANDARD_LRS" -DiskTypeAzureDiskVolumeTypeParam = Literal["PREMIUM_LRS", "STANDARD_LRS"] | DiskTypeAzureDiskVolumeType + +DiskTypeAzureDiskVolumeTypeParam = ( + Literal["PREMIUM_LRS", "STANDARD_LRS"] | DiskTypeAzureDiskVolumeType +) diff --git a/python/databricks/bundles/instance_pools/_models/disk_type_ebs_volume_type.py b/python/databricks/bundles/instance_pools/_models/disk_type_ebs_volume_type.py index 3d3c96487d8..cb5ae6b7a05 100644 --- a/python/databricks/bundles/instance_pools/_models/disk_type_ebs_volume_type.py +++ b/python/databricks/bundles/instance_pools/_models/disk_type_ebs_volume_type.py @@ -1,34 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam +from typing import Literal class DiskTypeEbsVolumeType(Enum): @@ -36,7 +9,11 @@ class DiskTypeEbsVolumeType(Enum): All EBS volume types that Databricks supports. See https://aws.amazon.com/ebs/details/ for details. """ + GENERAL_PURPOSE_SSD = "GENERAL_PURPOSE_SSD" THROUGHPUT_OPTIMIZED_HDD = "THROUGHPUT_OPTIMIZED_HDD" -DiskTypeEbsVolumeTypeParam = Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | DiskTypeEbsVolumeType + +DiskTypeEbsVolumeTypeParam = ( + Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | DiskTypeEbsVolumeType +) diff --git a/python/databricks/bundles/instance_pools/_models/docker_basic_auth.py b/python/databricks/bundles/instance_pools/_models/docker_basic_auth.py index 6f165be26c1..552ea90a83b 100644 --- a/python/databricks/bundles/instance_pools/_models/docker_basic_auth.py +++ b/python/databricks/bundles/instance_pools/_models/docker_basic_auth.py @@ -1,40 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DockerBasicAuth: """""" @@ -50,12 +26,11 @@ class DockerBasicAuth: """ @classmethod - def from_dict(cls, value: 'DockerBasicAuthDict') -> 'Self': + def from_dict(cls, value: "DockerBasicAuthDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DockerBasicAuthDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DockerBasicAuthDict": + return _transform_to_json_value(self) # type:ignore class DockerBasicAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/docker_image.py b/python/databricks/bundles/instance_pools/_models/docker_image.py index 857ed779ba4..15b43d46554 100644 --- a/python/databricks/bundles/instance_pools/_models/docker_image.py +++ b/python/databricks/bundles/instance_pools/_models/docker_image.py @@ -1,40 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.instance_pools._models.docker_basic_auth import ( + DockerBasicAuth, + DockerBasicAuthParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DockerImage: """""" @@ -50,12 +30,11 @@ class DockerImage: """ @classmethod - def from_dict(cls, value: 'DockerImageDict') -> 'Self': + def from_dict(cls, value: "DockerImageDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DockerImageDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DockerImageDict": + return _transform_to_json_value(self) # type:ignore class DockerImageDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/gcp_availability.py b/python/databricks/bundles/instance_pools/_models/gcp_availability.py index 452b66695fd..0d391b87fe3 100644 --- a/python/databricks/bundles/instance_pools/_models/gcp_availability.py +++ b/python/databricks/bundles/instance_pools/_models/gcp_availability.py @@ -1,34 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam +from typing import Literal class GcpAvailability(Enum): @@ -36,8 +9,13 @@ class GcpAvailability(Enum): This field determines whether the instance pool will contain preemptible VMs, on-demand VMs, or preemptible VMs with a fallback to on-demand VMs if the former is unavailable. """ + PREEMPTIBLE_GCP = "PREEMPTIBLE_GCP" ON_DEMAND_GCP = "ON_DEMAND_GCP" PREEMPTIBLE_WITH_FALLBACK_GCP = "PREEMPTIBLE_WITH_FALLBACK_GCP" -GcpAvailabilityParam = Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] | GcpAvailability + +GcpAvailabilityParam = ( + Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] + | GcpAvailability +) diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool.py b/python/databricks/bundles/instance_pools/_models/instance_pool.py index 2a61a8ae312..2b10f210284 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool.py @@ -1,40 +1,51 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecParam +from databricks.bundles.instance_pools._models.docker_image import ( + DockerImage, + DockerImageParam, +) +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import ( + InstancePoolAwsAttributes, + InstancePoolAwsAttributesParam, +) +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import ( + InstancePoolAzureAttributes, + InstancePoolAzureAttributesParam, +) +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import ( + InstancePoolGcpAttributes, + InstancePoolGcpAttributesParam, +) +from databricks.bundles.instance_pools._models.instance_pool_permission import ( + InstancePoolPermission, + InstancePoolPermissionParam, +) +from databricks.bundles.instance_pools._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.instance_pools._models.node_type_flexibility import ( + NodeTypeFlexibility, + NodeTypeFlexibilityParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InstancePool(Resource): """""" @@ -152,12 +163,11 @@ class InstancePool(Resource): """ @classmethod - def from_dict(cls, value: 'InstancePoolDict') -> 'Self': + def from_dict(cls, value: "InstancePoolDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InstancePoolDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InstancePoolDict": + return _transform_to_json_value(self) # type:ignore class InstancePoolDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes.py b/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes.py index ced88bc7faf..25fe305787e 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes.py @@ -1,45 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import ( + InstancePoolAwsAttributesAvailability, + InstancePoolAwsAttributesAvailabilityParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InstancePoolAwsAttributes: """ Attributes set during instance pool creation which are related to Amazon Web Services. """ + availability: VariableOrOptional[InstancePoolAwsAttributesAvailability] = None """ Availability type used for the instances in the pool. Supports on-demand, spot, and @@ -99,12 +80,11 @@ class InstancePoolAwsAttributes: """ @classmethod - def from_dict(cls, value: 'InstancePoolAwsAttributesDict') -> 'Self': + def from_dict(cls, value: "InstancePoolAwsAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InstancePoolAwsAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InstancePoolAwsAttributesDict": + return _transform_to_json_value(self) # type:ignore class InstancePoolAwsAttributesDict(TypedDict, total=False): @@ -169,4 +149,6 @@ class InstancePoolAwsAttributesDict(TypedDict, total=False): """ -InstancePoolAwsAttributesParam = InstancePoolAwsAttributesDict | InstancePoolAwsAttributes +InstancePoolAwsAttributesParam = ( + InstancePoolAwsAttributesDict | InstancePoolAwsAttributes +) diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes_availability.py b/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes_availability.py index 903baed3b5f..5bc8a0350fa 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes_availability.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_aws_attributes_availability.py @@ -1,41 +1,18 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam +from typing import Literal class InstancePoolAwsAttributesAvailability(Enum): """ The set of AWS availability types supported when setting up nodes for a cluster. """ + SPOT = "SPOT" ON_DEMAND = "ON_DEMAND" -InstancePoolAwsAttributesAvailabilityParam = Literal["SPOT", "ON_DEMAND"] | InstancePoolAwsAttributesAvailability + +InstancePoolAwsAttributesAvailabilityParam = ( + Literal["SPOT", "ON_DEMAND"] | InstancePoolAwsAttributesAvailability +) diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes.py b/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes.py index 78ec65ef031..80833993289 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes.py @@ -1,45 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import ( + InstancePoolAzureAttributesAvailability, + InstancePoolAzureAttributesAvailabilityParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InstancePoolAzureAttributes: """ Attributes set during instance pool creation which are related to Azure. """ + availability: VariableOrOptional[InstancePoolAzureAttributesAvailability] = None """ Availability type used for the instances in the pool. Supports on-demand, spot, and @@ -80,12 +61,11 @@ class InstancePoolAzureAttributes: """ @classmethod - def from_dict(cls, value: 'InstancePoolAzureAttributesDict') -> 'Self': + def from_dict(cls, value: "InstancePoolAzureAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InstancePoolAzureAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InstancePoolAzureAttributesDict": + return _transform_to_json_value(self) # type:ignore class InstancePoolAzureAttributesDict(TypedDict, total=False): @@ -131,4 +111,6 @@ class InstancePoolAzureAttributesDict(TypedDict, total=False): """ -InstancePoolAzureAttributesParam = InstancePoolAzureAttributesDict | InstancePoolAzureAttributes +InstancePoolAzureAttributesParam = ( + InstancePoolAzureAttributesDict | InstancePoolAzureAttributes +) diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes_availability.py b/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes_availability.py index 822fb058e76..6d41fe997dd 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes_availability.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_azure_attributes_availability.py @@ -1,41 +1,18 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam +from typing import Literal class InstancePoolAzureAttributesAvailability(Enum): """ The set of Azure availability types supported when setting up nodes for a cluster. """ + SPOT_AZURE = "SPOT_AZURE" ON_DEMAND_AZURE = "ON_DEMAND_AZURE" -InstancePoolAzureAttributesAvailabilityParam = Literal["SPOT_AZURE", "ON_DEMAND_AZURE"] | InstancePoolAzureAttributesAvailability + +InstancePoolAzureAttributesAvailabilityParam = ( + Literal["SPOT_AZURE", "ON_DEMAND_AZURE"] | InstancePoolAzureAttributesAvailability +) diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_gcp_attributes.py b/python/databricks/bundles/instance_pools/_models/instance_pool_gcp_attributes.py index 8a93dfd8c0f..244158a007d 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_gcp_attributes.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_gcp_attributes.py @@ -1,45 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.instance_pools._models.gcp_availability import ( + GcpAvailability, + GcpAvailabilityParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InstancePoolGcpAttributes: """ Attributes set during instance pool creation which are related to GCP. """ + gcp_availability: VariableOrOptional[GcpAvailability] = None local_ssd_count: VariableOrOptional[int] = None @@ -69,12 +50,11 @@ class InstancePoolGcpAttributes: """ @classmethod - def from_dict(cls, value: 'InstancePoolGcpAttributesDict') -> 'Self': + def from_dict(cls, value: "InstancePoolGcpAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InstancePoolGcpAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InstancePoolGcpAttributesDict": + return _transform_to_json_value(self) # type:ignore class InstancePoolGcpAttributesDict(TypedDict, total=False): @@ -109,4 +89,6 @@ class InstancePoolGcpAttributesDict(TypedDict, total=False): """ -InstancePoolGcpAttributesParam = InstancePoolGcpAttributesDict | InstancePoolGcpAttributes +InstancePoolGcpAttributesParam = ( + InstancePoolGcpAttributesDict | InstancePoolGcpAttributes +) diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_permission.py b/python/databricks/bundles/instance_pools/_models/instance_pool_permission.py index dc0996107ef..a81c2602aa6 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_permission.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_permission.py @@ -1,40 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.instance_pools._models.instance_pool_permission_level import ( + InstancePoolPermissionLevel, + InstancePoolPermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InstancePoolPermission: """""" @@ -60,12 +40,11 @@ class InstancePoolPermission: """ @classmethod - def from_dict(cls, value: 'InstancePoolPermissionDict') -> 'Self': + def from_dict(cls, value: "InstancePoolPermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InstancePoolPermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InstancePoolPermissionDict": + return _transform_to_json_value(self) # type:ignore class InstancePoolPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/instance_pool_permission_level.py b/python/databricks/bundles/instance_pools/_models/instance_pool_permission_level.py index 67ebf6a3989..bb0468b742c 100644 --- a/python/databricks/bundles/instance_pools/_models/instance_pool_permission_level.py +++ b/python/databricks/bundles/instance_pools/_models/instance_pool_permission_level.py @@ -1,41 +1,18 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam +from typing import Literal class InstancePoolPermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_ATTACH_TO = "CAN_ATTACH_TO" -InstancePoolPermissionLevelParam = Literal["CAN_MANAGE", "CAN_ATTACH_TO"] | InstancePoolPermissionLevel + +InstancePoolPermissionLevelParam = ( + Literal["CAN_MANAGE", "CAN_ATTACH_TO"] | InstancePoolPermissionLevel +) diff --git a/python/databricks/bundles/instance_pools/_models/lifecycle.py b/python/databricks/bundles/instance_pools/_models/lifecycle.py index 3b29b72c225..697776a198e 100644 --- a/python/databricks/bundles/instance_pools/_models/lifecycle.py +++ b/python/databricks/bundles/instance_pools/_models/lifecycle.py @@ -1,40 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -45,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/instance_pools/_models/node_type_flexibility.py b/python/databricks/bundles/instance_pools/_models/node_type_flexibility.py index 8dffee2f2cd..418ea9337bd 100644 --- a/python/databricks/bundles/instance_pools/_models/node_type_flexibility.py +++ b/python/databricks/bundles/instance_pools/_models/node_type_flexibility.py @@ -1,45 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.instance_pools._models.disk_spec import DiskSpec, DiskSpecDict, DiskSpecParam -from databricks.bundles.instance_pools._models.disk_type import DiskType, DiskTypeDict, DiskTypeParam -from databricks.bundles.instance_pools._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.instance_pools._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes, InstancePoolAwsAttributesDict, InstancePoolAwsAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes, InstancePoolAzureAttributesDict, InstancePoolAzureAttributesParam -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes, InstancePoolGcpAttributesDict, InstancePoolGcpAttributesParam -from databricks.bundles.instance_pools._models.instance_pool import InstancePool, InstancePoolDict, InstancePoolParam -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission, InstancePoolPermissionDict, InstancePoolPermissionParam -from databricks.bundles.instance_pools._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.instance_pools._models.disk_type_azure_disk_volume_type import DiskTypeAzureDiskVolumeType, DiskTypeAzureDiskVolumeTypeParam - -from databricks.bundles.instance_pools._models.disk_type_ebs_volume_type import DiskTypeEbsVolumeType, DiskTypeEbsVolumeTypeParam - -from databricks.bundles.instance_pools._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes_availability import InstancePoolAwsAttributesAvailability, InstancePoolAwsAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes_availability import InstancePoolAzureAttributesAvailability, InstancePoolAzureAttributesAvailabilityParam - -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel, InstancePoolPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class NodeTypeFlexibility: """ Configuration for flexible node types, allowing fallback to alternate node types during cluster launch and upscale. """ + alternate_node_type_ids: VariableOrList[str] = field(default_factory=list) """ A list of node type IDs to use as fallbacks when the primary node type is unavailable. @@ -52,12 +29,11 @@ class NodeTypeFlexibility: """ @classmethod - def from_dict(cls, value: 'NodeTypeFlexibilityDict') -> 'Self': + def from_dict(cls, value: "NodeTypeFlexibilityDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'NodeTypeFlexibilityDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "NodeTypeFlexibilityDict": + return _transform_to_json_value(self) # type:ignore class NodeTypeFlexibilityDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/__init__.py b/python/databricks/bundles/job_runs/__init__.py index 2c5f5e1b96c..093f2201409 100644 --- a/python/databricks/bundles/job_runs/__init__.py +++ b/python/databricks/bundles/job_runs/__init__.py @@ -1,30 +1,48 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'JobRun', - 'JobRunDict', - 'JobRunLifecycle', - 'JobRunLifecycleDict', - 'JobRunLifecycleParam', - 'JobRunParam', - 'JobRunTrigger', - 'JobRunTriggerDict', - 'JobRunTriggerParam', - 'PerformanceTarget', - 'PerformanceTargetParam', - 'PipelineParams', - 'PipelineParamsDict', - 'PipelineParamsParam', - 'QueueSettings', - 'QueueSettingsDict', - 'QueueSettingsParam', + "JobRun", + "JobRunDict", + "JobRunLifecycle", + "JobRunLifecycleDict", + "JobRunLifecycleParam", + "JobRunParam", + "JobRunTrigger", + "JobRunTriggerDict", + "JobRunTriggerParam", + "PerformanceTarget", + "PerformanceTargetParam", + "PipelineParams", + "PipelineParamsDict", + "PipelineParamsParam", + "QueueSettings", + "QueueSettingsDict", + "QueueSettingsParam", ] -from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam -from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam -from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam -from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam - +from databricks.bundles.job_runs._models.job_run_lifecycle import ( + JobRunLifecycle, + JobRunLifecycleDict, + JobRunLifecycleParam, +) +from databricks.bundles.job_runs._models.job_run_trigger import ( + JobRunTrigger, + JobRunTriggerDict, + JobRunTriggerParam, +) +from databricks.bundles.job_runs._models.performance_target import ( + PerformanceTarget, + PerformanceTargetParam, +) +from databricks.bundles.job_runs._models.pipeline_params import ( + PipelineParams, + PipelineParamsDict, + PipelineParamsParam, +) +from databricks.bundles.job_runs._models.queue_settings import ( + QueueSettings, + QueueSettingsDict, + QueueSettingsParam, +) diff --git a/python/databricks/bundles/job_runs/_models/job_run.py b/python/databricks/bundles/job_runs/_models/job_run.py index 94dc9e6758c..e7f5a0b3f71 100644 --- a/python/databricks/bundles/job_runs/_models/job_run.py +++ b/python/databricks/bundles/job_runs/_models/job_run.py @@ -1,24 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam -from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam -from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.job_runs._models.job_run_lifecycle import ( + JobRunLifecycle, + JobRunLifecycleParam, +) +from databricks.bundles.job_runs._models.performance_target import ( + PerformanceTarget, + PerformanceTargetParam, +) +from databricks.bundles.job_runs._models.pipeline_params import ( + PipelineParams, + PipelineParamsParam, +) +from databricks.bundles.job_runs._models.queue_settings import ( + QueueSettings, + QueueSettingsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobRun(Resource): """""" @@ -77,12 +91,11 @@ class JobRun(Resource): """ @classmethod - def from_dict(cls, value: 'JobRunDict') -> 'Self': + def from_dict(cls, value: "JobRunDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobRunDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobRunDict": + return _transform_to_json_value(self) # type:ignore class JobRunDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/_models/job_run_lifecycle.py b/python/databricks/bundles/job_runs/_models/job_run_lifecycle.py index 0b99f0f5d61..cb673b0245d 100644 --- a/python/databricks/bundles/job_runs/_models/job_run_lifecycle.py +++ b/python/databricks/bundles/job_runs/_models/job_run_lifecycle.py @@ -1,24 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam -from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam -from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.job_runs._models.job_run_trigger import ( + JobRunTrigger, + JobRunTriggerParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobRunLifecycle: """""" @@ -34,12 +30,11 @@ class JobRunLifecycle: """ @classmethod - def from_dict(cls, value: 'JobRunLifecycleDict') -> 'Self': + def from_dict(cls, value: "JobRunLifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobRunLifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobRunLifecycleDict": + return _transform_to_json_value(self) # type:ignore class JobRunLifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/_models/job_run_trigger.py b/python/databricks/bundles/job_runs/_models/job_run_trigger.py index 5bac797649c..75bdf970a4e 100644 --- a/python/databricks/bundles/job_runs/_models/job_run_trigger.py +++ b/python/databricks/bundles/job_runs/_models/job_run_trigger.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam -from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam -from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobRunTrigger: """""" @@ -34,12 +26,11 @@ class JobRunTrigger: """ @classmethod - def from_dict(cls, value: 'JobRunTriggerDict') -> 'Self': + def from_dict(cls, value: "JobRunTriggerDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobRunTriggerDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobRunTriggerDict": + return _transform_to_json_value(self) # type:ignore class JobRunTriggerDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/_models/performance_target.py b/python/databricks/bundles/job_runs/_models/performance_target.py index 43cef1b8afa..8dbe7e4a435 100644 --- a/python/databricks/bundles/job_runs/_models/performance_target.py +++ b/python/databricks/bundles/job_runs/_models/performance_target.py @@ -1,19 +1,8 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam -from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam -from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam class PerformanceTarget(Enum): """ @@ -21,7 +10,11 @@ class PerformanceTarget(Enum): The performance mode on the job or pipeline should map to a performance setting that is passed to Cluster Manager (see cluster-common PerformanceTarget). """ + PERFORMANCE_OPTIMIZED = "PERFORMANCE_OPTIMIZED" STANDARD = "STANDARD" -PerformanceTargetParam = Literal["PERFORMANCE_OPTIMIZED", "STANDARD"] | PerformanceTarget + +PerformanceTargetParam = ( + Literal["PERFORMANCE_OPTIMIZED", "STANDARD"] | PerformanceTarget +) diff --git a/python/databricks/bundles/job_runs/_models/pipeline_params.py b/python/databricks/bundles/job_runs/_models/pipeline_params.py index fe3bf70db82..98550f271fb 100644 --- a/python/databricks/bundles/job_runs/_models/pipeline_params.py +++ b/python/databricks/bundles/job_runs/_models/pipeline_params.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.job_runs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam -from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam -from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam -from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PipelineParams: """""" @@ -50,12 +42,11 @@ class PipelineParams: """ @classmethod - def from_dict(cls, value: 'PipelineParamsDict') -> 'Self': + def from_dict(cls, value: "PipelineParamsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PipelineParamsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PipelineParamsDict": + return _transform_to_json_value(self) # type:ignore class PipelineParamsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/job_runs/_models/queue_settings.py b/python/databricks/bundles/job_runs/_models/queue_settings.py index a6bca6701b7..a72921aed96 100644 --- a/python/databricks/bundles/job_runs/_models/queue_settings.py +++ b/python/databricks/bundles/job_runs/_models/queue_settings.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.job_runs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.job_runs._models.job_run import JobRun, JobRunDict, JobRunParam -from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle, JobRunLifecycleDict, JobRunLifecycleParam -from databricks.bundles.job_runs._models.job_run_trigger import JobRunTrigger, JobRunTriggerDict, JobRunTriggerParam -from databricks.bundles.job_runs._models.performance_target import PerformanceTarget, PerformanceTargetParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class QueueSettings: """""" @@ -29,12 +21,11 @@ class QueueSettings: """ @classmethod - def from_dict(cls, value: 'QueueSettingsDict') -> 'Self': + def from_dict(cls, value: "QueueSettingsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'QueueSettingsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "QueueSettingsDict": + return _transform_to_json_value(self) # type:ignore class QueueSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/__init__.py b/python/databricks/bundles/jobs/__init__.py index e3fc1309b09..1df8be62977 100644 --- a/python/databricks/bundles/jobs/__init__.py +++ b/python/databricks/bundles/jobs/__init__.py @@ -1,524 +1,927 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Adlsgen2Info', - 'Adlsgen2InfoDict', - 'Adlsgen2InfoParam', - 'AiRuntimeTask', - 'AiRuntimeTaskDict', - 'AiRuntimeTaskParam', - 'AiRuntimeTaskPriorityClass', - 'AiRuntimeTaskPriorityClassParam', - 'AlertTask', - 'AlertTaskDict', - 'AlertTaskParam', - 'AlertTaskSubscriber', - 'AlertTaskSubscriberDict', - 'AlertTaskSubscriberParam', - 'AuthenticationMethod', - 'AuthenticationMethodParam', - 'AutoScale', - 'AutoScaleDict', - 'AutoScaleParam', - 'AwsAttributes', - 'AwsAttributesDict', - 'AwsAttributesParam', - 'AwsAvailability', - 'AwsAvailabilityParam', - 'AzureAttributes', - 'AzureAttributesDict', - 'AzureAttributesParam', - 'AzureAvailability', - 'AzureAvailabilityParam', - 'CleanRoomsNotebookTask', - 'CleanRoomsNotebookTaskDict', - 'CleanRoomsNotebookTaskParam', - 'ClientsTypes', - 'ClientsTypesDict', - 'ClientsTypesParam', - 'ClusterLogConf', - 'ClusterLogConfDict', - 'ClusterLogConfParam', - 'ClusterSpec', - 'ClusterSpecDict', - 'ClusterSpecParam', - 'Compute', - 'ComputeConfig', - 'ComputeConfigDict', - 'ComputeConfigParam', - 'ComputeDict', - 'ComputeParam', - 'ComputeSpec', - 'ComputeSpecAcceleratorType', - 'ComputeSpecAcceleratorTypeParam', - 'ComputeSpecDict', - 'ComputeSpecParam', - 'Condition', - 'ConditionParam', - 'ConditionTask', - 'ConditionTaskDict', - 'ConditionTaskOp', - 'ConditionTaskOpParam', - 'ConditionTaskParam', - 'ConfidentialComputeType', - 'ConfidentialComputeTypeParam', - 'Continuous', - 'ContinuousDict', - 'ContinuousParam', - 'ContinuousTriggerConfiguration', - 'ContinuousTriggerConfigurationDict', - 'ContinuousTriggerConfigurationParam', - 'CronSchedule', - 'CronScheduleDict', - 'CronScheduleParam', - 'CronTriggerConfiguration', - 'CronTriggerConfigurationDict', - 'CronTriggerConfigurationParam', - 'DashboardTask', - 'DashboardTaskDict', - 'DashboardTaskParam', - 'DataSecurityMode', - 'DataSecurityModeParam', - 'DayOfWeek', - 'DayOfWeekParam', - 'DbfsStorageInfo', - 'DbfsStorageInfoDict', - 'DbfsStorageInfoParam', - 'DbtPlatformTask', - 'DbtPlatformTaskDict', - 'DbtPlatformTaskParam', - 'DbtTask', - 'DbtTaskDict', - 'DbtTaskParam', - 'DependencyMode', - 'DependencyModeParam', - 'DeploymentSpec', - 'DeploymentSpecDict', - 'DeploymentSpecParam', - 'DockerBasicAuth', - 'DockerBasicAuthDict', - 'DockerBasicAuthParam', - 'DockerImage', - 'DockerImageDict', - 'DockerImageParam', - 'EbsVolumeType', - 'EbsVolumeTypeParam', - 'Environment', - 'EnvironmentDict', - 'EnvironmentParam', - 'FileArrivalTriggerConfiguration', - 'FileArrivalTriggerConfigurationDict', - 'FileArrivalTriggerConfigurationParam', - 'ForEachTask', - 'ForEachTaskDict', - 'ForEachTaskParam', - 'GcpAttributes', - 'GcpAttributesDict', - 'GcpAttributesParam', - 'GcpAvailability', - 'GcpAvailabilityParam', - 'GcsStorageInfo', - 'GcsStorageInfoDict', - 'GcsStorageInfoParam', - 'GenAiComputeTask', - 'GenAiComputeTaskDict', - 'GenAiComputeTaskParam', - 'GitProvider', - 'GitProviderParam', - 'GitSource', - 'GitSourceDict', - 'GitSourceParam', - 'HardwareAcceleratorType', - 'HardwareAcceleratorTypeParam', - 'InitScriptInfo', - 'InitScriptInfoDict', - 'InitScriptInfoParam', - 'Job', - 'JobCluster', - 'JobClusterDict', - 'JobClusterParam', - 'JobDict', - 'JobEmailNotifications', - 'JobEmailNotificationsDict', - 'JobEmailNotificationsParam', - 'JobEnvironment', - 'JobEnvironmentDict', - 'JobEnvironmentParam', - 'JobEnvironmentVariables', - 'JobEnvironmentVariablesDict', - 'JobEnvironmentVariablesParam', - 'JobEnvironmentVariablesSpec', - 'JobEnvironmentVariablesSpecDict', - 'JobEnvironmentVariablesSpecParam', - 'JobNotificationSettings', - 'JobNotificationSettingsDict', - 'JobNotificationSettingsParam', - 'JobParam', - 'JobParameterDefinition', - 'JobParameterDefinitionDict', - 'JobParameterDefinitionParam', - 'JobPermission', - 'JobPermissionDict', - 'JobPermissionLevel', - 'JobPermissionLevelParam', - 'JobPermissionParam', - 'JobRunAs', - 'JobRunAsDict', - 'JobRunAsParam', - 'JobsHealthMetric', - 'JobsHealthMetricParam', - 'JobsHealthOperator', - 'JobsHealthOperatorParam', - 'JobsHealthRule', - 'JobsHealthRuleDict', - 'JobsHealthRuleParam', - 'JobsHealthRules', - 'JobsHealthRulesDict', - 'JobsHealthRulesParam', - 'Kind', - 'KindParam', - 'Library', - 'LibraryDict', - 'LibraryParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'LocalFileInfo', - 'LocalFileInfoDict', - 'LocalFileInfoParam', - 'LogAnalyticsInfo', - 'LogAnalyticsInfoDict', - 'LogAnalyticsInfoParam', - 'MaintenanceWindow', - 'MaintenanceWindowDict', - 'MaintenanceWindowParam', - 'MavenLibrary', - 'MavenLibraryDict', - 'MavenLibraryParam', - 'ModelTriggerConfiguration', - 'ModelTriggerConfigurationCondition', - 'ModelTriggerConfigurationConditionParam', - 'ModelTriggerConfigurationDict', - 'ModelTriggerConfigurationParam', - 'NodeTypeFlexibility', - 'NodeTypeFlexibilityDict', - 'NodeTypeFlexibilityParam', - 'NotebookTask', - 'NotebookTaskDict', - 'NotebookTaskParam', - 'PauseStatus', - 'PauseStatusParam', - 'PerformanceTarget', - 'PerformanceTargetParam', - 'PeriodicTriggerConfiguration', - 'PeriodicTriggerConfigurationDict', - 'PeriodicTriggerConfigurationParam', - 'PeriodicTriggerConfigurationTimeUnit', - 'PeriodicTriggerConfigurationTimeUnitParam', - 'PipelineParams', - 'PipelineParamsDict', - 'PipelineParamsParam', - 'PipelineTask', - 'PipelineTaskDict', - 'PipelineTaskParam', - 'PowerBiModel', - 'PowerBiModelDict', - 'PowerBiModelParam', - 'PowerBiTable', - 'PowerBiTableDict', - 'PowerBiTableParam', - 'PowerBiTask', - 'PowerBiTaskDict', - 'PowerBiTaskParam', - 'PythonOperatorTask', - 'PythonOperatorTaskDict', - 'PythonOperatorTaskParam', - 'PythonOperatorTaskParameter', - 'PythonOperatorTaskParameterDict', - 'PythonOperatorTaskParameterParam', - 'PythonPyPiLibrary', - 'PythonPyPiLibraryDict', - 'PythonPyPiLibraryParam', - 'PythonWheelTask', - 'PythonWheelTaskDict', - 'PythonWheelTaskParam', - 'QueueSettings', - 'QueueSettingsDict', - 'QueueSettingsParam', - 'RCranLibrary', - 'RCranLibraryDict', - 'RCranLibraryParam', - 'RunIf', - 'RunIfParam', - 'RunJobTask', - 'RunJobTaskDict', - 'RunJobTaskParam', - 'RuntimeEngine', - 'RuntimeEngineParam', - 'S3StorageInfo', - 'S3StorageInfoDict', - 'S3StorageInfoParam', - 'Source', - 'SourceParam', - 'SparkJarTask', - 'SparkJarTaskDict', - 'SparkJarTaskParam', - 'SparkPythonTask', - 'SparkPythonTaskDict', - 'SparkPythonTaskParam', - 'SparkSubmitTask', - 'SparkSubmitTaskDict', - 'SparkSubmitTaskParam', - 'SparseCheckout', - 'SparseCheckoutDict', - 'SparseCheckoutParam', - 'SqlConditionConfiguration', - 'SqlConditionConfigurationDict', - 'SqlConditionConfigurationParam', - 'SqlConditionTriggerMode', - 'SqlConditionTriggerModeParam', - 'SqlTask', - 'SqlTaskAlert', - 'SqlTaskAlertDict', - 'SqlTaskAlertParam', - 'SqlTaskDashboard', - 'SqlTaskDashboardDict', - 'SqlTaskDashboardParam', - 'SqlTaskDict', - 'SqlTaskFile', - 'SqlTaskFileDict', - 'SqlTaskFileParam', - 'SqlTaskParam', - 'SqlTaskQuery', - 'SqlTaskQueryDict', - 'SqlTaskQueryParam', - 'SqlTaskSubscription', - 'SqlTaskSubscriptionDict', - 'SqlTaskSubscriptionParam', - 'StorageMode', - 'StorageModeParam', - 'Subscription', - 'SubscriptionAttachment', - 'SubscriptionAttachmentDataAttachment', - 'SubscriptionAttachmentDataAttachmentDict', - 'SubscriptionAttachmentDataAttachmentFileFormat', - 'SubscriptionAttachmentDataAttachmentFileFormatParam', - 'SubscriptionAttachmentDataAttachmentParam', - 'SubscriptionAttachmentDataAttachmentWidget', - 'SubscriptionAttachmentDataAttachmentWidgetDict', - 'SubscriptionAttachmentDataAttachmentWidgetParam', - 'SubscriptionAttachmentDict', - 'SubscriptionAttachmentPageAttachment', - 'SubscriptionAttachmentPageAttachmentDict', - 'SubscriptionAttachmentPageAttachmentPageSelection', - 'SubscriptionAttachmentPageAttachmentPageSelectionParam', - 'SubscriptionAttachmentPageAttachmentParam', - 'SubscriptionAttachmentParam', - 'SubscriptionDict', - 'SubscriptionParam', - 'SubscriptionSubscriber', - 'SubscriptionSubscriberDict', - 'SubscriptionSubscriberParam', - 'TableUpdateTriggerConfiguration', - 'TableUpdateTriggerConfigurationDict', - 'TableUpdateTriggerConfigurationParam', - 'Task', - 'TaskDependency', - 'TaskDependencyDict', - 'TaskDependencyParam', - 'TaskDict', - 'TaskEmailNotifications', - 'TaskEmailNotificationsDict', - 'TaskEmailNotificationsParam', - 'TaskNotificationSettings', - 'TaskNotificationSettingsDict', - 'TaskNotificationSettingsParam', - 'TaskParam', - 'TaskRetryMode', - 'TaskRetryModeParam', - 'TriggerConfiguration', - 'TriggerConfigurationDict', - 'TriggerConfigurationParam', - 'TriggerSettings', - 'TriggerSettingsDict', - 'TriggerSettingsParam', - 'VolumesStorageInfo', - 'VolumesStorageInfoDict', - 'VolumesStorageInfoParam', - 'Webhook', - 'WebhookDict', - 'WebhookNotifications', - 'WebhookNotificationsDict', - 'WebhookNotificationsParam', - 'WebhookParam', - 'WorkloadType', - 'WorkloadTypeDict', - 'WorkloadTypeParam', - 'WorkspaceStorageInfo', - 'WorkspaceStorageInfoDict', - 'WorkspaceStorageInfoParam', + "Adlsgen2Info", + "Adlsgen2InfoDict", + "Adlsgen2InfoParam", + "AiRuntimeTask", + "AiRuntimeTaskDict", + "AiRuntimeTaskParam", + "AiRuntimeTaskPriorityClass", + "AiRuntimeTaskPriorityClassParam", + "AlertTask", + "AlertTaskDict", + "AlertTaskParam", + "AlertTaskSubscriber", + "AlertTaskSubscriberDict", + "AlertTaskSubscriberParam", + "AuthenticationMethod", + "AuthenticationMethodParam", + "AutoScale", + "AutoScaleDict", + "AutoScaleParam", + "AwsAttributes", + "AwsAttributesDict", + "AwsAttributesParam", + "AwsAvailability", + "AwsAvailabilityParam", + "AzureAttributes", + "AzureAttributesDict", + "AzureAttributesParam", + "AzureAvailability", + "AzureAvailabilityParam", + "CleanRoomsNotebookTask", + "CleanRoomsNotebookTaskDict", + "CleanRoomsNotebookTaskParam", + "ClientsTypes", + "ClientsTypesDict", + "ClientsTypesParam", + "ClusterLogConf", + "ClusterLogConfDict", + "ClusterLogConfParam", + "ClusterSpec", + "ClusterSpecDict", + "ClusterSpecParam", + "Compute", + "ComputeConfig", + "ComputeConfigDict", + "ComputeConfigParam", + "ComputeDict", + "ComputeParam", + "ComputeSpec", + "ComputeSpecAcceleratorType", + "ComputeSpecAcceleratorTypeParam", + "ComputeSpecDict", + "ComputeSpecParam", + "Condition", + "ConditionParam", + "ConditionTask", + "ConditionTaskDict", + "ConditionTaskOp", + "ConditionTaskOpParam", + "ConditionTaskParam", + "ConfidentialComputeType", + "ConfidentialComputeTypeParam", + "Continuous", + "ContinuousDict", + "ContinuousParam", + "ContinuousTriggerConfiguration", + "ContinuousTriggerConfigurationDict", + "ContinuousTriggerConfigurationParam", + "CronSchedule", + "CronScheduleDict", + "CronScheduleParam", + "CronTriggerConfiguration", + "CronTriggerConfigurationDict", + "CronTriggerConfigurationParam", + "DashboardTask", + "DashboardTaskDict", + "DashboardTaskParam", + "DataSecurityMode", + "DataSecurityModeParam", + "DayOfWeek", + "DayOfWeekParam", + "DbfsStorageInfo", + "DbfsStorageInfoDict", + "DbfsStorageInfoParam", + "DbtPlatformTask", + "DbtPlatformTaskDict", + "DbtPlatformTaskParam", + "DbtTask", + "DbtTaskDict", + "DbtTaskParam", + "DependencyMode", + "DependencyModeParam", + "DeploymentSpec", + "DeploymentSpecDict", + "DeploymentSpecParam", + "DockerBasicAuth", + "DockerBasicAuthDict", + "DockerBasicAuthParam", + "DockerImage", + "DockerImageDict", + "DockerImageParam", + "EbsVolumeType", + "EbsVolumeTypeParam", + "Environment", + "EnvironmentDict", + "EnvironmentParam", + "FileArrivalTriggerConfiguration", + "FileArrivalTriggerConfigurationDict", + "FileArrivalTriggerConfigurationParam", + "ForEachTask", + "ForEachTaskDict", + "ForEachTaskParam", + "GcpAttributes", + "GcpAttributesDict", + "GcpAttributesParam", + "GcpAvailability", + "GcpAvailabilityParam", + "GcsStorageInfo", + "GcsStorageInfoDict", + "GcsStorageInfoParam", + "GenAiComputeTask", + "GenAiComputeTaskDict", + "GenAiComputeTaskParam", + "GitProvider", + "GitProviderParam", + "GitSource", + "GitSourceDict", + "GitSourceParam", + "HardwareAcceleratorType", + "HardwareAcceleratorTypeParam", + "InitScriptInfo", + "InitScriptInfoDict", + "InitScriptInfoParam", + "Job", + "JobCluster", + "JobClusterDict", + "JobClusterParam", + "JobDict", + "JobEmailNotifications", + "JobEmailNotificationsDict", + "JobEmailNotificationsParam", + "JobEnvironment", + "JobEnvironmentDict", + "JobEnvironmentParam", + "JobEnvironmentVariables", + "JobEnvironmentVariablesDict", + "JobEnvironmentVariablesParam", + "JobEnvironmentVariablesSpec", + "JobEnvironmentVariablesSpecDict", + "JobEnvironmentVariablesSpecParam", + "JobNotificationSettings", + "JobNotificationSettingsDict", + "JobNotificationSettingsParam", + "JobParam", + "JobParameterDefinition", + "JobParameterDefinitionDict", + "JobParameterDefinitionParam", + "JobPermission", + "JobPermissionDict", + "JobPermissionLevel", + "JobPermissionLevelParam", + "JobPermissionParam", + "JobRunAs", + "JobRunAsDict", + "JobRunAsParam", + "JobsHealthMetric", + "JobsHealthMetricParam", + "JobsHealthOperator", + "JobsHealthOperatorParam", + "JobsHealthRule", + "JobsHealthRuleDict", + "JobsHealthRuleParam", + "JobsHealthRules", + "JobsHealthRulesDict", + "JobsHealthRulesParam", + "Kind", + "KindParam", + "Library", + "LibraryDict", + "LibraryParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "LocalFileInfo", + "LocalFileInfoDict", + "LocalFileInfoParam", + "LogAnalyticsInfo", + "LogAnalyticsInfoDict", + "LogAnalyticsInfoParam", + "MaintenanceWindow", + "MaintenanceWindowDict", + "MaintenanceWindowParam", + "MavenLibrary", + "MavenLibraryDict", + "MavenLibraryParam", + "ModelTriggerConfiguration", + "ModelTriggerConfigurationCondition", + "ModelTriggerConfigurationConditionParam", + "ModelTriggerConfigurationDict", + "ModelTriggerConfigurationParam", + "NodeTypeFlexibility", + "NodeTypeFlexibilityDict", + "NodeTypeFlexibilityParam", + "NotebookTask", + "NotebookTaskDict", + "NotebookTaskParam", + "PauseStatus", + "PauseStatusParam", + "PerformanceTarget", + "PerformanceTargetParam", + "PeriodicTriggerConfiguration", + "PeriodicTriggerConfigurationDict", + "PeriodicTriggerConfigurationParam", + "PeriodicTriggerConfigurationTimeUnit", + "PeriodicTriggerConfigurationTimeUnitParam", + "PipelineParams", + "PipelineParamsDict", + "PipelineParamsParam", + "PipelineTask", + "PipelineTaskDict", + "PipelineTaskParam", + "PowerBiModel", + "PowerBiModelDict", + "PowerBiModelParam", + "PowerBiTable", + "PowerBiTableDict", + "PowerBiTableParam", + "PowerBiTask", + "PowerBiTaskDict", + "PowerBiTaskParam", + "PythonOperatorTask", + "PythonOperatorTaskDict", + "PythonOperatorTaskParam", + "PythonOperatorTaskParameter", + "PythonOperatorTaskParameterDict", + "PythonOperatorTaskParameterParam", + "PythonPyPiLibrary", + "PythonPyPiLibraryDict", + "PythonPyPiLibraryParam", + "PythonWheelTask", + "PythonWheelTaskDict", + "PythonWheelTaskParam", + "QueueSettings", + "QueueSettingsDict", + "QueueSettingsParam", + "RCranLibrary", + "RCranLibraryDict", + "RCranLibraryParam", + "RunIf", + "RunIfParam", + "RunJobTask", + "RunJobTaskDict", + "RunJobTaskParam", + "RuntimeEngine", + "RuntimeEngineParam", + "S3StorageInfo", + "S3StorageInfoDict", + "S3StorageInfoParam", + "Source", + "SourceParam", + "SparkJarTask", + "SparkJarTaskDict", + "SparkJarTaskParam", + "SparkPythonTask", + "SparkPythonTaskDict", + "SparkPythonTaskParam", + "SparkSubmitTask", + "SparkSubmitTaskDict", + "SparkSubmitTaskParam", + "SparseCheckout", + "SparseCheckoutDict", + "SparseCheckoutParam", + "SqlConditionConfiguration", + "SqlConditionConfigurationDict", + "SqlConditionConfigurationParam", + "SqlConditionTriggerMode", + "SqlConditionTriggerModeParam", + "SqlTask", + "SqlTaskAlert", + "SqlTaskAlertDict", + "SqlTaskAlertParam", + "SqlTaskDashboard", + "SqlTaskDashboardDict", + "SqlTaskDashboardParam", + "SqlTaskDict", + "SqlTaskFile", + "SqlTaskFileDict", + "SqlTaskFileParam", + "SqlTaskParam", + "SqlTaskQuery", + "SqlTaskQueryDict", + "SqlTaskQueryParam", + "SqlTaskSubscription", + "SqlTaskSubscriptionDict", + "SqlTaskSubscriptionParam", + "StorageMode", + "StorageModeParam", + "Subscription", + "SubscriptionAttachment", + "SubscriptionAttachmentDataAttachment", + "SubscriptionAttachmentDataAttachmentDict", + "SubscriptionAttachmentDataAttachmentFileFormat", + "SubscriptionAttachmentDataAttachmentFileFormatParam", + "SubscriptionAttachmentDataAttachmentParam", + "SubscriptionAttachmentDataAttachmentWidget", + "SubscriptionAttachmentDataAttachmentWidgetDict", + "SubscriptionAttachmentDataAttachmentWidgetParam", + "SubscriptionAttachmentDict", + "SubscriptionAttachmentPageAttachment", + "SubscriptionAttachmentPageAttachmentDict", + "SubscriptionAttachmentPageAttachmentPageSelection", + "SubscriptionAttachmentPageAttachmentPageSelectionParam", + "SubscriptionAttachmentPageAttachmentParam", + "SubscriptionAttachmentParam", + "SubscriptionDict", + "SubscriptionParam", + "SubscriptionSubscriber", + "SubscriptionSubscriberDict", + "SubscriptionSubscriberParam", + "TableUpdateTriggerConfiguration", + "TableUpdateTriggerConfigurationDict", + "TableUpdateTriggerConfigurationParam", + "Task", + "TaskDependency", + "TaskDependencyDict", + "TaskDependencyParam", + "TaskDict", + "TaskEmailNotifications", + "TaskEmailNotificationsDict", + "TaskEmailNotificationsParam", + "TaskNotificationSettings", + "TaskNotificationSettingsDict", + "TaskNotificationSettingsParam", + "TaskParam", + "TaskRetryMode", + "TaskRetryModeParam", + "TriggerConfiguration", + "TriggerConfigurationDict", + "TriggerConfigurationParam", + "TriggerSettings", + "TriggerSettingsDict", + "TriggerSettingsParam", + "VolumesStorageInfo", + "VolumesStorageInfoDict", + "VolumesStorageInfoParam", + "Webhook", + "WebhookDict", + "WebhookNotifications", + "WebhookNotificationsDict", + "WebhookNotificationsParam", + "WebhookParam", + "WorkloadType", + "WorkloadTypeDict", + "WorkloadTypeParam", + "WorkspaceStorageInfo", + "WorkspaceStorageInfoDict", + "WorkspaceStorageInfoParam", ] -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam +from databricks.bundles.jobs._models.adlsgen2_info import ( + Adlsgen2Info, + Adlsgen2InfoDict, + Adlsgen2InfoParam, +) +from databricks.bundles.jobs._models.ai_runtime_task import ( + AiRuntimeTask, + AiRuntimeTaskDict, + AiRuntimeTaskParam, +) +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import ( + AiRuntimeTaskPriorityClass, + AiRuntimeTaskPriorityClassParam, +) +from databricks.bundles.jobs._models.alert_task import ( + AlertTask, + AlertTaskDict, + AlertTaskParam, +) +from databricks.bundles.jobs._models.alert_task_subscriber import ( + AlertTaskSubscriber, + AlertTaskSubscriberDict, + AlertTaskSubscriberParam, +) +from databricks.bundles.jobs._models.authentication_method import ( + AuthenticationMethod, + AuthenticationMethodParam, +) +from databricks.bundles.jobs._models.auto_scale import ( + AutoScale, + AutoScaleDict, + AutoScaleParam, +) +from databricks.bundles.jobs._models.aws_attributes import ( + AwsAttributes, + AwsAttributesDict, + AwsAttributesParam, +) +from databricks.bundles.jobs._models.aws_availability import ( + AwsAvailability, + AwsAvailabilityParam, +) +from databricks.bundles.jobs._models.azure_attributes import ( + AzureAttributes, + AzureAttributesDict, + AzureAttributesParam, +) +from databricks.bundles.jobs._models.azure_availability import ( + AzureAvailability, + AzureAvailabilityParam, +) +from databricks.bundles.jobs._models.clean_rooms_notebook_task import ( + CleanRoomsNotebookTask, + CleanRoomsNotebookTaskDict, + CleanRoomsNotebookTaskParam, +) +from databricks.bundles.jobs._models.clients_types import ( + ClientsTypes, + ClientsTypesDict, + ClientsTypesParam, +) +from databricks.bundles.jobs._models.cluster_log_conf import ( + ClusterLogConf, + ClusterLogConfDict, + ClusterLogConfParam, +) +from databricks.bundles.jobs._models.cluster_spec import ( + ClusterSpec, + ClusterSpecDict, + ClusterSpecParam, +) from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - +from databricks.bundles.jobs._models.compute_config import ( + ComputeConfig, + ComputeConfigDict, + ComputeConfigParam, +) +from databricks.bundles.jobs._models.compute_spec import ( + ComputeSpec, + ComputeSpecDict, + ComputeSpecParam, +) +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ( + ComputeSpecAcceleratorType, + ComputeSpecAcceleratorTypeParam, +) from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - +from databricks.bundles.jobs._models.condition_task import ( + ConditionTask, + ConditionTaskDict, + ConditionTaskParam, +) +from databricks.bundles.jobs._models.condition_task_op import ( + ConditionTaskOp, + ConditionTaskOpParam, +) +from databricks.bundles.jobs._models.confidential_compute_type import ( + ConfidentialComputeType, + ConfidentialComputeTypeParam, +) +from databricks.bundles.jobs._models.continuous import ( + Continuous, + ContinuousDict, + ContinuousParam, +) +from databricks.bundles.jobs._models.continuous_trigger_configuration import ( + ContinuousTriggerConfiguration, + ContinuousTriggerConfigurationDict, + ContinuousTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.cron_schedule import ( + CronSchedule, + CronScheduleDict, + CronScheduleParam, +) +from databricks.bundles.jobs._models.cron_trigger_configuration import ( + CronTriggerConfiguration, + CronTriggerConfigurationDict, + CronTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.dashboard_task import ( + DashboardTask, + DashboardTaskDict, + DashboardTaskParam, +) +from databricks.bundles.jobs._models.data_security_mode import ( + DataSecurityMode, + DataSecurityModeParam, +) from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - +from databricks.bundles.jobs._models.dbfs_storage_info import ( + DbfsStorageInfo, + DbfsStorageInfoDict, + DbfsStorageInfoParam, +) +from databricks.bundles.jobs._models.dbt_platform_task import ( + DbtPlatformTask, + DbtPlatformTaskDict, + DbtPlatformTaskParam, +) +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam +from databricks.bundles.jobs._models.dependency_mode import ( + DependencyMode, + DependencyModeParam, +) +from databricks.bundles.jobs._models.deployment_spec import ( + DeploymentSpec, + DeploymentSpecDict, + DeploymentSpecParam, +) +from databricks.bundles.jobs._models.docker_basic_auth import ( + DockerBasicAuth, + DockerBasicAuthDict, + DockerBasicAuthParam, +) +from databricks.bundles.jobs._models.docker_image import ( + DockerImage, + DockerImageDict, + DockerImageParam, +) +from databricks.bundles.jobs._models.ebs_volume_type import ( + EbsVolumeType, + EbsVolumeTypeParam, +) +from databricks.bundles.jobs._models.environment import ( + Environment, + EnvironmentDict, + EnvironmentParam, +) +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import ( + FileArrivalTriggerConfiguration, + FileArrivalTriggerConfigurationDict, + FileArrivalTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.for_each_task import ( + ForEachTask, + ForEachTaskDict, + ForEachTaskParam, +) +from databricks.bundles.jobs._models.gcp_attributes import ( + GcpAttributes, + GcpAttributesDict, + GcpAttributesParam, +) +from databricks.bundles.jobs._models.gcp_availability import ( + GcpAvailability, + GcpAvailabilityParam, +) +from databricks.bundles.jobs._models.gcs_storage_info import ( + GcsStorageInfo, + GcsStorageInfoDict, + GcsStorageInfoParam, +) +from databricks.bundles.jobs._models.gen_ai_compute_task import ( + GenAiComputeTask, + GenAiComputeTaskDict, + GenAiComputeTaskParam, +) from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - +from databricks.bundles.jobs._models.git_source import ( + GitSource, + GitSourceDict, + GitSourceParam, +) +from databricks.bundles.jobs._models.hardware_accelerator_type import ( + HardwareAcceleratorType, + HardwareAcceleratorTypeParam, +) +from databricks.bundles.jobs._models.init_script_info import ( + InitScriptInfo, + InitScriptInfoDict, + InitScriptInfoParam, +) +from databricks.bundles.jobs._models.job import Job, JobDict, JobParam +from databricks.bundles.jobs._models.job_cluster import ( + JobCluster, + JobClusterDict, + JobClusterParam, +) +from databricks.bundles.jobs._models.job_email_notifications import ( + JobEmailNotifications, + JobEmailNotificationsDict, + JobEmailNotificationsParam, +) +from databricks.bundles.jobs._models.job_environment import ( + JobEnvironment, + JobEnvironmentDict, + JobEnvironmentParam, +) +from databricks.bundles.jobs._models.job_environment_variables import ( + JobEnvironmentVariables, + JobEnvironmentVariablesDict, + JobEnvironmentVariablesParam, +) +from databricks.bundles.jobs._models.job_environment_variables_spec import ( + JobEnvironmentVariablesSpec, + JobEnvironmentVariablesSpecDict, + JobEnvironmentVariablesSpecParam, +) +from databricks.bundles.jobs._models.job_notification_settings import ( + JobNotificationSettings, + JobNotificationSettingsDict, + JobNotificationSettingsParam, +) +from databricks.bundles.jobs._models.job_parameter_definition import ( + JobParameterDefinition, + JobParameterDefinitionDict, + JobParameterDefinitionParam, +) +from databricks.bundles.jobs._models.job_permission import ( + JobPermission, + JobPermissionDict, + JobPermissionParam, +) +from databricks.bundles.jobs._models.job_permission_level import ( + JobPermissionLevel, + JobPermissionLevelParam, +) +from databricks.bundles.jobs._models.job_run_as import ( + JobRunAs, + JobRunAsDict, + JobRunAsParam, +) +from databricks.bundles.jobs._models.jobs_health_metric import ( + JobsHealthMetric, + JobsHealthMetricParam, +) +from databricks.bundles.jobs._models.jobs_health_operator import ( + JobsHealthOperator, + JobsHealthOperatorParam, +) +from databricks.bundles.jobs._models.jobs_health_rule import ( + JobsHealthRule, + JobsHealthRuleDict, + JobsHealthRuleParam, +) +from databricks.bundles.jobs._models.jobs_health_rules import ( + JobsHealthRules, + JobsHealthRulesDict, + JobsHealthRulesParam, +) +from databricks.bundles.jobs._models.kind import Kind, KindParam +from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam +from databricks.bundles.jobs._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.jobs._models.local_file_info import ( + LocalFileInfo, + LocalFileInfoDict, + LocalFileInfoParam, +) +from databricks.bundles.jobs._models.log_analytics_info import ( + LogAnalyticsInfo, + LogAnalyticsInfoDict, + LogAnalyticsInfoParam, +) +from databricks.bundles.jobs._models.maintenance_window import ( + MaintenanceWindow, + MaintenanceWindowDict, + MaintenanceWindowParam, +) +from databricks.bundles.jobs._models.maven_library import ( + MavenLibrary, + MavenLibraryDict, + MavenLibraryParam, +) +from databricks.bundles.jobs._models.model_trigger_configuration import ( + ModelTriggerConfiguration, + ModelTriggerConfigurationDict, + ModelTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ( + ModelTriggerConfigurationCondition, + ModelTriggerConfigurationConditionParam, +) +from databricks.bundles.jobs._models.node_type_flexibility import ( + NodeTypeFlexibility, + NodeTypeFlexibilityDict, + NodeTypeFlexibilityParam, +) +from databricks.bundles.jobs._models.notebook_task import ( + NotebookTask, + NotebookTaskDict, + NotebookTaskParam, +) from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - +from databricks.bundles.jobs._models.performance_target import ( + PerformanceTarget, + PerformanceTargetParam, +) +from databricks.bundles.jobs._models.periodic_trigger_configuration import ( + PeriodicTriggerConfiguration, + PeriodicTriggerConfigurationDict, + PeriodicTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import ( + PeriodicTriggerConfigurationTimeUnit, + PeriodicTriggerConfigurationTimeUnitParam, +) +from databricks.bundles.jobs._models.pipeline_params import ( + PipelineParams, + PipelineParamsDict, + PipelineParamsParam, +) +from databricks.bundles.jobs._models.pipeline_task import ( + PipelineTask, + PipelineTaskDict, + PipelineTaskParam, +) +from databricks.bundles.jobs._models.power_bi_model import ( + PowerBiModel, + PowerBiModelDict, + PowerBiModelParam, +) +from databricks.bundles.jobs._models.power_bi_table import ( + PowerBiTable, + PowerBiTableDict, + PowerBiTableParam, +) +from databricks.bundles.jobs._models.power_bi_task import ( + PowerBiTask, + PowerBiTaskDict, + PowerBiTaskParam, +) +from databricks.bundles.jobs._models.python_operator_task import ( + PythonOperatorTask, + PythonOperatorTaskDict, + PythonOperatorTaskParam, +) +from databricks.bundles.jobs._models.python_operator_task_parameter import ( + PythonOperatorTaskParameter, + PythonOperatorTaskParameterDict, + PythonOperatorTaskParameterParam, +) +from databricks.bundles.jobs._models.python_py_pi_library import ( + PythonPyPiLibrary, + PythonPyPiLibraryDict, + PythonPyPiLibraryParam, +) +from databricks.bundles.jobs._models.python_wheel_task import ( + PythonWheelTask, + PythonWheelTaskDict, + PythonWheelTaskParam, +) +from databricks.bundles.jobs._models.queue_settings import ( + QueueSettings, + QueueSettingsDict, + QueueSettingsParam, +) +from databricks.bundles.jobs._models.r_cran_library import ( + RCranLibrary, + RCranLibraryDict, + RCranLibraryParam, +) from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - +from databricks.bundles.jobs._models.run_job_task import ( + RunJobTask, + RunJobTaskDict, + RunJobTaskParam, +) +from databricks.bundles.jobs._models.runtime_engine import ( + RuntimeEngine, + RuntimeEngineParam, +) +from databricks.bundles.jobs._models.s3_storage_info import ( + S3StorageInfo, + S3StorageInfoDict, + S3StorageInfoParam, +) from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - +from databricks.bundles.jobs._models.spark_jar_task import ( + SparkJarTask, + SparkJarTaskDict, + SparkJarTaskParam, +) +from databricks.bundles.jobs._models.spark_python_task import ( + SparkPythonTask, + SparkPythonTaskDict, + SparkPythonTaskParam, +) +from databricks.bundles.jobs._models.spark_submit_task import ( + SparkSubmitTask, + SparkSubmitTaskDict, + SparkSubmitTaskParam, +) +from databricks.bundles.jobs._models.sparse_checkout import ( + SparseCheckout, + SparseCheckoutDict, + SparseCheckoutParam, +) +from databricks.bundles.jobs._models.sql_condition_configuration import ( + SqlConditionConfiguration, + SqlConditionConfigurationDict, + SqlConditionConfigurationParam, +) +from databricks.bundles.jobs._models.sql_condition_trigger_mode import ( + SqlConditionTriggerMode, + SqlConditionTriggerModeParam, +) +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam +from databricks.bundles.jobs._models.sql_task_alert import ( + SqlTaskAlert, + SqlTaskAlertDict, + SqlTaskAlertParam, +) +from databricks.bundles.jobs._models.sql_task_dashboard import ( + SqlTaskDashboard, + SqlTaskDashboardDict, + SqlTaskDashboardParam, +) +from databricks.bundles.jobs._models.sql_task_file import ( + SqlTaskFile, + SqlTaskFileDict, + SqlTaskFileParam, +) +from databricks.bundles.jobs._models.sql_task_query import ( + SqlTaskQuery, + SqlTaskQueryDict, + SqlTaskQueryParam, +) +from databricks.bundles.jobs._models.sql_task_subscription import ( + SqlTaskSubscription, + SqlTaskSubscriptionDict, + SqlTaskSubscriptionParam, +) from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from databricks.bundles.jobs._models.subscription import ( + Subscription, + SubscriptionDict, + SubscriptionParam, +) +from databricks.bundles.jobs._models.subscription_attachment import ( + SubscriptionAttachment, + SubscriptionAttachmentDict, + SubscriptionAttachmentParam, +) +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import ( + SubscriptionAttachmentDataAttachment, + SubscriptionAttachmentDataAttachmentDict, + SubscriptionAttachmentDataAttachmentParam, +) +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import ( + SubscriptionAttachmentDataAttachmentFileFormat, + SubscriptionAttachmentDataAttachmentFileFormatParam, +) +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import ( + SubscriptionAttachmentDataAttachmentWidget, + SubscriptionAttachmentDataAttachmentWidgetDict, + SubscriptionAttachmentDataAttachmentWidgetParam, +) +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import ( + SubscriptionAttachmentPageAttachment, + SubscriptionAttachmentPageAttachmentDict, + SubscriptionAttachmentPageAttachmentParam, +) +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import ( + SubscriptionAttachmentPageAttachmentPageSelection, + SubscriptionAttachmentPageAttachmentPageSelectionParam, +) +from databricks.bundles.jobs._models.subscription_subscriber import ( + SubscriptionSubscriber, + SubscriptionSubscriberDict, + SubscriptionSubscriberParam, +) +from databricks.bundles.jobs._models.table_update_trigger_configuration import ( + TableUpdateTriggerConfiguration, + TableUpdateTriggerConfigurationDict, + TableUpdateTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam +from databricks.bundles.jobs._models.task_dependency import ( + TaskDependency, + TaskDependencyDict, + TaskDependencyParam, +) +from databricks.bundles.jobs._models.task_email_notifications import ( + TaskEmailNotifications, + TaskEmailNotificationsDict, + TaskEmailNotificationsParam, +) +from databricks.bundles.jobs._models.task_notification_settings import ( + TaskNotificationSettings, + TaskNotificationSettingsDict, + TaskNotificationSettingsParam, +) +from databricks.bundles.jobs._models.task_retry_mode import ( + TaskRetryMode, + TaskRetryModeParam, +) +from databricks.bundles.jobs._models.trigger_configuration import ( + TriggerConfiguration, + TriggerConfigurationDict, + TriggerConfigurationParam, +) +from databricks.bundles.jobs._models.trigger_settings import ( + TriggerSettings, + TriggerSettingsDict, + TriggerSettingsParam, +) +from databricks.bundles.jobs._models.volumes_storage_info import ( + VolumesStorageInfo, + VolumesStorageInfoDict, + VolumesStorageInfoParam, +) +from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam +from databricks.bundles.jobs._models.webhook_notifications import ( + WebhookNotifications, + WebhookNotificationsDict, + WebhookNotificationsParam, +) +from databricks.bundles.jobs._models.workload_type import ( + WorkloadType, + WorkloadTypeDict, + WorkloadTypeParam, +) +from databricks.bundles.jobs._models.workspace_storage_info import ( + WorkspaceStorageInfo, + WorkspaceStorageInfoDict, + WorkspaceStorageInfoParam, +) def _resolve_recursive_imports(): @@ -532,4 +935,5 @@ def _resolve_recursive_imports(): globalns={"Task": Task, "VariableOr": VariableOr}, ) + _resolve_recursive_imports() diff --git a/python/databricks/bundles/jobs/_models/adlsgen2_info.py b/python/databricks/bundles/jobs/_models/adlsgen2_info.py index 82d71e373c4..87c883adeb9 100644 --- a/python/databricks/bundles/jobs/_models/adlsgen2_info.py +++ b/python/databricks/bundles/jobs/_models/adlsgen2_info.py @@ -1,194 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Adlsgen2Info: """ A storage location in Adls Gen2 """ + destination: VariableOr[str] """ abfss destination, e.g. `abfss://@.dfs.core.windows.net/`. """ @classmethod - def from_dict(cls, value: 'Adlsgen2InfoDict') -> 'Self': + def from_dict(cls, value: "Adlsgen2InfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'Adlsgen2InfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "Adlsgen2InfoDict": + return _transform_to_json_value(self) # type:ignore class Adlsgen2InfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/ai_runtime_task.py b/python/databricks/bundles/jobs/_models/ai_runtime_task.py index bee1cc5a45c..09144f23c83 100644 --- a/python/databricks/bundles/jobs/_models/ai_runtime_task.py +++ b/python/databricks/bundles/jobs/_models/ai_runtime_task.py @@ -1,189 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.jobs._models.ai_runtime_task_priority_class import ( + AiRuntimeTaskPriorityClass, + AiRuntimeTaskPriorityClassParam, +) +from databricks.bundles.jobs._models.deployment_spec import ( + DeploymentSpec, + DeploymentSpecParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AiRuntimeTask: """ AiRuntimeTask: multi-node GPU compute task definition for Databricks AI Runtime workloads. - + Jobs-framework-level concepts (retries, per-task timeout, idempotency token, usage/budget policy, permissions) live on the surrounding TaskSettings / run-submit request and are intentionally NOT duplicated here. Users compose `ai_runtime_task` with the standard Jobs/DABs task wrapper to get those. """ + experiment: VariableOr[str] """ [Public Preview] MLflow experiment name for this run. If an experiment with this name @@ -268,12 +120,11 @@ class AiRuntimeTask: """ @classmethod - def from_dict(cls, value: 'AiRuntimeTaskDict') -> 'Self': + def from_dict(cls, value: "AiRuntimeTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AiRuntimeTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AiRuntimeTaskDict": + return _transform_to_json_value(self) # type:ignore class AiRuntimeTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/ai_runtime_task_priority_class.py b/python/databricks/bundles/jobs/_models/ai_runtime_task_priority_class.py index 2db44c93491..1dd09afd131 100644 --- a/python/databricks/bundles/jobs/_models/ai_runtime_task_priority_class.py +++ b/python/databricks/bundles/jobs/_models/ai_runtime_task_priority_class.py @@ -1,182 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class AiRuntimeTaskPriorityClass(Enum): """ :meta private: [EXPERIMENTAL] - + Scheduling priority class for a workload — its priority and preemptability when the scheduler ranks pending work. """ + BEST_EFFORT = "BEST_EFFORT" NORMAL = "NORMAL" CRITICAL = "CRITICAL" -AiRuntimeTaskPriorityClassParam = Literal["BEST_EFFORT", "NORMAL", "CRITICAL"] | AiRuntimeTaskPriorityClass + +AiRuntimeTaskPriorityClassParam = ( + Literal["BEST_EFFORT", "NORMAL", "CRITICAL"] | AiRuntimeTaskPriorityClass +) diff --git a/python/databricks/bundles/jobs/_models/alert_task.py b/python/databricks/bundles/jobs/_models/alert_task.py index eb139db7056..4f3b0d97611 100644 --- a/python/databricks/bundles/jobs/_models/alert_task.py +++ b/python/databricks/bundles/jobs/_models/alert_task.py @@ -1,177 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.jobs._models.alert_task_subscriber import ( + AlertTaskSubscriber, + AlertTaskSubscriberParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertTask: """""" @@ -213,12 +60,11 @@ class AlertTask: """ @classmethod - def from_dict(cls, value: 'AlertTaskDict') -> 'Self': + def from_dict(cls, value: "AlertTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertTaskDict": + return _transform_to_json_value(self) # type:ignore class AlertTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/alert_task_subscriber.py b/python/databricks/bundles/jobs/_models/alert_task_subscriber.py index 5fd6c9b9b2a..3805069edcf 100644 --- a/python/databricks/bundles/jobs/_models/alert_task_subscriber.py +++ b/python/databricks/bundles/jobs/_models/alert_task_subscriber.py @@ -1,183 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AlertTaskSubscriber: """ Represents a subscriber that will receive alert notifications. A subscriber can be either a user (via email) or a notification destination (via destination_id). """ + destination_id: VariableOrOptional[str] = None """ [Public Preview] @@ -189,12 +29,11 @@ class AlertTaskSubscriber: """ @classmethod - def from_dict(cls, value: 'AlertTaskSubscriberDict') -> 'Self': + def from_dict(cls, value: "AlertTaskSubscriberDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AlertTaskSubscriberDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AlertTaskSubscriberDict": + return _transform_to_json_value(self) # type:ignore class AlertTaskSubscriberDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/authentication_method.py b/python/databricks/bundles/jobs/_models/authentication_method.py index 55ad8f3e192..ba7b603d8d5 100644 --- a/python/databricks/bundles/jobs/_models/authentication_method.py +++ b/python/databricks/bundles/jobs/_models/authentication_method.py @@ -1,175 +1,12 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class AuthenticationMethod(Enum): OAUTH = "OAUTH" PAT = "PAT" + AuthenticationMethodParam = Literal["OAUTH", "PAT"] | AuthenticationMethod diff --git a/python/databricks/bundles/jobs/_models/auto_scale.py b/python/databricks/bundles/jobs/_models/auto_scale.py index 031836d4fbc..10ca50c26d7 100644 --- a/python/databricks/bundles/jobs/_models/auto_scale.py +++ b/python/databricks/bundles/jobs/_models/auto_scale.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AutoScale: """""" @@ -189,12 +28,11 @@ class AutoScale: """ @classmethod - def from_dict(cls, value: 'AutoScaleDict') -> 'Self': + def from_dict(cls, value: "AutoScaleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AutoScaleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AutoScaleDict": + return _transform_to_json_value(self) # type:ignore class AutoScaleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/aws_attributes.py b/python/databricks/bundles/jobs/_models/aws_attributes.py index 081ac942051..5eaf7acbed8 100644 --- a/python/databricks/bundles/jobs/_models/aws_attributes.py +++ b/python/databricks/bundles/jobs/_models/aws_attributes.py @@ -1,182 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.aws_availability import ( + AwsAvailability, + AwsAvailabilityParam, +) +from databricks.bundles.jobs._models.ebs_volume_type import ( + EbsVolumeType, + EbsVolumeTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AwsAttributes: """ Attributes set during cluster creation which are related to Amazon Web Services. """ + availability: VariableOrOptional[AwsAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -275,12 +123,11 @@ class AwsAttributes: """ @classmethod - def from_dict(cls, value: 'AwsAttributesDict') -> 'Self': + def from_dict(cls, value: "AwsAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AwsAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AwsAttributesDict": + return _transform_to_json_value(self) # type:ignore class AwsAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/aws_availability.py b/python/databricks/bundles/jobs/_models/aws_availability.py index 8188487c483..4c3810f067e 100644 --- a/python/databricks/bundles/jobs/_models/aws_availability.py +++ b/python/databricks/bundles/jobs/_models/aws_availability.py @@ -1,181 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class AwsAvailability(Enum): """ Availability type used for all subsequent nodes past the `first_on_demand` ones. - + Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ + SPOT = "SPOT" ON_DEMAND = "ON_DEMAND" SPOT_WITH_FALLBACK = "SPOT_WITH_FALLBACK" -AwsAvailabilityParam = Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability + +AwsAvailabilityParam = ( + Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability +) diff --git a/python/databricks/bundles/jobs/_models/azure_attributes.py b/python/databricks/bundles/jobs/_models/azure_attributes.py index 969d3abdb89..64509dcd040 100644 --- a/python/databricks/bundles/jobs/_models/azure_attributes.py +++ b/python/databricks/bundles/jobs/_models/azure_attributes.py @@ -1,182 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.azure_availability import ( + AzureAvailability, + AzureAvailabilityParam, +) +from databricks.bundles.jobs._models.log_analytics_info import ( + LogAnalyticsInfo, + LogAnalyticsInfoParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AzureAttributes: """ Attributes set during cluster creation which are related to Microsoft Azure. """ + availability: VariableOrOptional[AzureAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -224,12 +72,11 @@ class AzureAttributes: """ @classmethod - def from_dict(cls, value: 'AzureAttributesDict') -> 'Self': + def from_dict(cls, value: "AzureAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AzureAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AzureAttributesDict": + return _transform_to_json_value(self) # type:ignore class AzureAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/azure_availability.py b/python/databricks/bundles/jobs/_models/azure_availability.py index dfc68e708f0..a03cb1fdddf 100644 --- a/python/databricks/bundles/jobs/_models/azure_availability.py +++ b/python/databricks/bundles/jobs/_models/azure_availability.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class AzureAvailability(Enum): @@ -173,8 +9,13 @@ class AzureAvailability(Enum): Availability type used for all subsequent nodes past the `first_on_demand` ones. Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ + SPOT_AZURE = "SPOT_AZURE" ON_DEMAND_AZURE = "ON_DEMAND_AZURE" SPOT_WITH_FALLBACK_AZURE = "SPOT_WITH_FALLBACK_AZURE" -AzureAvailabilityParam = Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] | AzureAvailability + +AzureAvailabilityParam = ( + Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] + | AzureAvailability +) diff --git a/python/databricks/bundles/jobs/_models/clean_rooms_notebook_task.py b/python/databricks/bundles/jobs/_models/clean_rooms_notebook_task.py index 507f5485c2d..52845928fe1 100644 --- a/python/databricks/bundles/jobs/_models/clean_rooms_notebook_task.py +++ b/python/databricks/bundles/jobs/_models/clean_rooms_notebook_task.py @@ -1,183 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class CleanRoomsNotebookTask: """ Clean Rooms notebook task for V1 Clean Room service (GA). Replaces the deprecated CleanRoomNotebookTask (defined above) which was for V0 service. """ + clean_room_name: VariableOr[str] """ The clean room that the notebook belongs to. @@ -200,12 +44,11 @@ class CleanRoomsNotebookTask: """ @classmethod - def from_dict(cls, value: 'CleanRoomsNotebookTaskDict') -> 'Self': + def from_dict(cls, value: "CleanRoomsNotebookTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CleanRoomsNotebookTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CleanRoomsNotebookTaskDict": + return _transform_to_json_value(self) # type:ignore class CleanRoomsNotebookTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/clients_types.py b/python/databricks/bundles/jobs/_models/clients_types.py index 601bc4a4ba1..87f92446ba5 100644 --- a/python/databricks/bundles/jobs/_models/clients_types.py +++ b/python/databricks/bundles/jobs/_models/clients_types.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ClientsTypes: """""" @@ -187,12 +26,11 @@ class ClientsTypes: """ @classmethod - def from_dict(cls, value: 'ClientsTypesDict') -> 'Self': + def from_dict(cls, value: "ClientsTypesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClientsTypesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClientsTypesDict": + return _transform_to_json_value(self) # type:ignore class ClientsTypesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/cluster_log_conf.py b/python/databricks/bundles/jobs/_models/cluster_log_conf.py index 496995fabd8..a5ff3dd34f8 100644 --- a/python/databricks/bundles/jobs/_models/cluster_log_conf.py +++ b/python/databricks/bundles/jobs/_models/cluster_log_conf.py @@ -1,182 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.dbfs_storage_info import ( + DbfsStorageInfo, + DbfsStorageInfoParam, +) +from databricks.bundles.jobs._models.s3_storage_info import ( + S3StorageInfo, + S3StorageInfoParam, +) +from databricks.bundles.jobs._models.volumes_storage_info import ( + VolumesStorageInfo, + VolumesStorageInfoParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ClusterLogConf: """ Cluster log delivery config """ + dbfs: VariableOrOptional[DbfsStorageInfo] = None """ destination needs to be provided. e.g. @@ -198,12 +50,11 @@ class ClusterLogConf: """ @classmethod - def from_dict(cls, value: 'ClusterLogConfDict') -> 'Self': + def from_dict(cls, value: "ClusterLogConfDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClusterLogConfDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClusterLogConfDict": + return _transform_to_json_value(self) # type:ignore class ClusterLogConfDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/cluster_spec.py b/python/databricks/bundles/jobs/_models/cluster_spec.py index 2ebec63c4c6..2afee23b3ca 100644 --- a/python/databricks/bundles/jobs/_models/cluster_spec.py +++ b/python/databricks/bundles/jobs/_models/cluster_spec.py @@ -1,182 +1,72 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - +from databricks.bundles.core._variable import ( + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.jobs._models.auto_scale import ( + AutoScale, + AutoScaleParam, +) +from databricks.bundles.jobs._models.aws_attributes import ( + AwsAttributes, + AwsAttributesParam, +) +from databricks.bundles.jobs._models.azure_attributes import ( + AzureAttributes, + AzureAttributesParam, +) +from databricks.bundles.jobs._models.cluster_log_conf import ( + ClusterLogConf, + ClusterLogConfParam, +) +from databricks.bundles.jobs._models.data_security_mode import ( + DataSecurityMode, + DataSecurityModeParam, +) +from databricks.bundles.jobs._models.dependency_mode import ( + DependencyMode, + DependencyModeParam, +) +from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageParam +from databricks.bundles.jobs._models.gcp_attributes import ( + GcpAttributes, + GcpAttributesParam, +) +from databricks.bundles.jobs._models.init_script_info import ( + InitScriptInfo, + InitScriptInfoParam, +) from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.jobs._models.node_type_flexibility import ( + NodeTypeFlexibility, + NodeTypeFlexibilityParam, +) +from databricks.bundles.jobs._models.runtime_engine import ( + RuntimeEngine, + RuntimeEngineParam, +) +from databricks.bundles.jobs._models.workload_type import ( + WorkloadType, + WorkloadTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ClusterSpec: """ Contains a snapshot of the latest user specified settings that were used to create/edit the cluster. """ + apply_policy_default_values: VariableOrOptional[bool] = None """ When set to true, fixed and default values from the policy will be used for fields that are omitted. When set to false, only fixed values from the policy will be applied. @@ -442,12 +332,11 @@ class ClusterSpec: """ @classmethod - def from_dict(cls, value: 'ClusterSpecDict') -> 'Self': + def from_dict(cls, value: "ClusterSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClusterSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClusterSpecDict": + return _transform_to_json_value(self) # type:ignore class ClusterSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/compute.py b/python/databricks/bundles/jobs/_models/compute.py index 8fe082b5b51..51cfac4566c 100644 --- a/python/databricks/bundles/jobs/_models/compute.py +++ b/python/databricks/bundles/jobs/_models/compute.py @@ -1,177 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.hardware_accelerator_type import ( + HardwareAcceleratorType, + HardwareAcceleratorTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Compute: """""" @@ -182,12 +25,11 @@ class Compute: """ @classmethod - def from_dict(cls, value: 'ComputeDict') -> 'Self': + def from_dict(cls, value: "ComputeDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ComputeDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ComputeDict": + return _transform_to_json_value(self) # type:ignore class ComputeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/compute_config.py b/python/databricks/bundles/jobs/_models/compute_config.py index b3e9e45f235..cac29578ce0 100644 --- a/python/databricks/bundles/jobs/_models/compute_config.py +++ b/python/databricks/bundles/jobs/_models/compute_config.py @@ -1,182 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ComputeConfig: """ :meta private: [EXPERIMENTAL] """ + num_gpus: VariableOr[int] """ :meta private: [EXPERIMENTAL] @@ -199,12 +39,11 @@ class ComputeConfig: """ @classmethod - def from_dict(cls, value: 'ComputeConfigDict') -> 'Self': + def from_dict(cls, value: "ComputeConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ComputeConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ComputeConfigDict": + return _transform_to_json_value(self) # type:ignore class ComputeConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/compute_spec.py b/python/databricks/bundles/jobs/_models/compute_spec.py index 2786c8f7f20..404c4eb6b37 100644 --- a/python/databricks/bundles/jobs/_models/compute_spec.py +++ b/python/databricks/bundles/jobs/_models/compute_spec.py @@ -1,183 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.jobs._models.compute_spec_accelerator_type import ( + ComputeSpecAcceleratorType, + ComputeSpecAcceleratorTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ComputeSpec: """ ComputeSpec: compute configuration — accelerator type and total accelerator count across all nodes. """ + accelerator_count: VariableOr[int] """ [Public Preview] Total number of accelerators across all nodes. Must be a positive @@ -204,12 +48,11 @@ class ComputeSpec: """ @classmethod - def from_dict(cls, value: 'ComputeSpecDict') -> 'Self': + def from_dict(cls, value: "ComputeSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ComputeSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ComputeSpecDict": + return _transform_to_json_value(self) # type:ignore class ComputeSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/compute_spec_accelerator_type.py b/python/databricks/bundles/jobs/_models/compute_spec_accelerator_type.py index 5abf5de3d41..218cce4766f 100644 --- a/python/databricks/bundles/jobs/_models/compute_spec_accelerator_type.py +++ b/python/databricks/bundles/jobs/_models/compute_spec_accelerator_type.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class ComputeSpecAcceleratorType(Enum): @@ -174,9 +10,14 @@ class ComputeSpecAcceleratorType(Enum): accelerator count is encoded in the value name (e.g. `GPU_8xH100` means 8 H100s per node). """ + GPU_1X_A10 = "GPU_1xA10" GPU_1X_H100 = "GPU_1xH100" GPU_8X_H100 = "GPU_8xH100" GPU_8X_B300 = "GPU_8xB300" -ComputeSpecAcceleratorTypeParam = Literal["GPU_1xA10", "GPU_1xH100", "GPU_8xH100", "GPU_8xB300"] | ComputeSpecAcceleratorType + +ComputeSpecAcceleratorTypeParam = ( + Literal["GPU_1xA10", "GPU_1xH100", "GPU_8xH100", "GPU_8xB300"] + | ComputeSpecAcceleratorType +) diff --git a/python/databricks/bundles/jobs/_models/condition.py b/python/databricks/bundles/jobs/_models/condition.py index 195d285f9d1..7c242ae9dbf 100644 --- a/python/databricks/bundles/jobs/_models/condition.py +++ b/python/databricks/bundles/jobs/_models/condition.py @@ -1,175 +1,12 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class Condition(Enum): ANY_UPDATED = "ANY_UPDATED" ALL_UPDATED = "ALL_UPDATED" + ConditionParam = Literal["ANY_UPDATED", "ALL_UPDATED"] | Condition diff --git a/python/databricks/bundles/jobs/_models/condition_task.py b/python/databricks/bundles/jobs/_models/condition_task.py index 7dd85d12143..478ca88de11 100644 --- a/python/databricks/bundles/jobs/_models/condition_task.py +++ b/python/databricks/bundles/jobs/_models/condition_task.py @@ -1,177 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr +from databricks.bundles.jobs._models.condition_task_op import ( + ConditionTaskOp, + ConditionTaskOpParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ConditionTask: """""" @@ -195,12 +38,11 @@ class ConditionTask: """ @classmethod - def from_dict(cls, value: 'ConditionTaskDict') -> 'Self': + def from_dict(cls, value: "ConditionTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ConditionTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ConditionTaskDict": + return _transform_to_json_value(self) # type:ignore class ConditionTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/condition_task_op.py b/python/databricks/bundles/jobs/_models/condition_task_op.py index e77ecd5d7b2..615a7ea995c 100644 --- a/python/databricks/bundles/jobs/_models/condition_task_op.py +++ b/python/databricks/bundles/jobs/_models/condition_task_op.py @@ -1,180 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class ConditionTaskOp(Enum): """ * `EQUAL_TO`, `NOT_EQUAL` operators perform string comparison of their operands. This means that `“12.0” == “12”` will evaluate to `false`. * `GREATER_THAN`, `GREATER_THAN_OR_EQUAL`, `LESS_THAN`, `LESS_THAN_OR_EQUAL` operators perform numeric comparison of their operands. `“12.0” >= “12”` will evaluate to `true`, `“10.0” >= “12”` will evaluate to `false`. - + The boolean comparison to task values can be implemented with operators `EQUAL_TO`, `NOT_EQUAL`. If a task value was set to a boolean value, it will be serialized to `“true”` or `“false”` for the comparison. """ + EQUAL_TO = "EQUAL_TO" GREATER_THAN = "GREATER_THAN" GREATER_THAN_OR_EQUAL = "GREATER_THAN_OR_EQUAL" @@ -182,4 +19,15 @@ class ConditionTaskOp(Enum): LESS_THAN_OR_EQUAL = "LESS_THAN_OR_EQUAL" NOT_EQUAL = "NOT_EQUAL" -ConditionTaskOpParam = Literal["EQUAL_TO", "GREATER_THAN", "GREATER_THAN_OR_EQUAL", "LESS_THAN", "LESS_THAN_OR_EQUAL", "NOT_EQUAL"] | ConditionTaskOp + +ConditionTaskOpParam = ( + Literal[ + "EQUAL_TO", + "GREATER_THAN", + "GREATER_THAN_OR_EQUAL", + "LESS_THAN", + "LESS_THAN_OR_EQUAL", + "NOT_EQUAL", + ] + | ConditionTaskOp +) diff --git a/python/databricks/bundles/jobs/_models/confidential_compute_type.py b/python/databricks/bundles/jobs/_models/confidential_compute_type.py index bee78ddb3aa..e881295bd47 100644 --- a/python/databricks/bundles/jobs/_models/confidential_compute_type.py +++ b/python/databricks/bundles/jobs/_models/confidential_compute_type.py @@ -1,183 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class ConfidentialComputeType(Enum): """ :meta private: [EXPERIMENTAL] - + Confidential computing technology for GCP instances. Aligns with gcloud's --confidential-compute-type flag and the REST API's confidentialInstanceConfig.confidentialInstanceType field. See: https://cloud.google.com/confidential-computing/confidential-vm/docs/create-a-confidential-vm-instance """ + CONFIDENTIAL_COMPUTE_TYPE_NONE = "CONFIDENTIAL_COMPUTE_TYPE_NONE" SEV_SNP = "SEV_SNP" -ConfidentialComputeTypeParam = Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType + +ConfidentialComputeTypeParam = ( + Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType +) diff --git a/python/databricks/bundles/jobs/_models/continuous.py b/python/databricks/bundles/jobs/_models/continuous.py index 72ef2fb5357..e707bb377f7 100644 --- a/python/databricks/bundles/jobs/_models/continuous.py +++ b/python/databricks/bundles/jobs/_models/continuous.py @@ -1,177 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.maintenance_window import ( + MaintenanceWindow, + MaintenanceWindowParam, +) from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.jobs._models.task_retry_mode import ( + TaskRetryMode, + TaskRetryModeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Continuous: """""" @@ -192,12 +40,11 @@ class Continuous: """ @classmethod - def from_dict(cls, value: 'ContinuousDict') -> 'Self': + def from_dict(cls, value: "ContinuousDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ContinuousDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ContinuousDict": + return _transform_to_json_value(self) # type:ignore class ContinuousDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/continuous_trigger_configuration.py b/python/databricks/bundles/jobs/_models/continuous_trigger_configuration.py index 3b500e591c5..421e630a548 100644 --- a/python/databricks/bundles/jobs/_models/continuous_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/continuous_trigger_configuration.py @@ -1,183 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.maintenance_window import ( + MaintenanceWindow, + MaintenanceWindowParam, +) +from databricks.bundles.jobs._models.task_retry_mode import ( + TaskRetryMode, + TaskRetryModeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ContinuousTriggerConfiguration: """ Continuous trigger. Stripped-down counterpart to `ContinuousSettings`: `pause_status` is owned by the enclosing `TriggerConfiguration` and intentionally omitted here. """ + maintenance_window: VariableOrOptional[MaintenanceWindow] = None """ :meta private: [EXPERIMENTAL] @@ -194,12 +42,11 @@ class ContinuousTriggerConfiguration: """ @classmethod - def from_dict(cls, value: 'ContinuousTriggerConfigurationDict') -> 'Self': + def from_dict(cls, value: "ContinuousTriggerConfigurationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ContinuousTriggerConfigurationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ContinuousTriggerConfigurationDict": + return _transform_to_json_value(self) # type:ignore class ContinuousTriggerConfigurationDict(TypedDict, total=False): @@ -221,4 +68,6 @@ class ContinuousTriggerConfigurationDict(TypedDict, total=False): """ -ContinuousTriggerConfigurationParam = ContinuousTriggerConfigurationDict | ContinuousTriggerConfiguration +ContinuousTriggerConfigurationParam = ( + ContinuousTriggerConfigurationDict | ContinuousTriggerConfiguration +) diff --git a/python/databricks/bundles/jobs/_models/cron_schedule.py b/python/databricks/bundles/jobs/_models/cron_schedule.py index 570e42beb07..3c8a37992af 100644 --- a/python/databricks/bundles/jobs/_models/cron_schedule.py +++ b/python/databricks/bundles/jobs/_models/cron_schedule.py @@ -1,177 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.jobs._models.sql_condition_configuration import ( + SqlConditionConfiguration, + SqlConditionConfigurationParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class CronSchedule: """""" @@ -200,12 +44,11 @@ class CronSchedule: """ @classmethod - def from_dict(cls, value: 'CronScheduleDict') -> 'Self': + def from_dict(cls, value: "CronScheduleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CronScheduleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CronScheduleDict": + return _transform_to_json_value(self) # type:ignore class CronScheduleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/cron_trigger_configuration.py b/python/databricks/bundles/jobs/_models/cron_trigger_configuration.py index 98cd9ab6dd8..9e083514b15 100644 --- a/python/databricks/bundles/jobs/_models/cron_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/cron_trigger_configuration.py @@ -1,183 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class CronTriggerConfiguration: """ Cron schedule trigger. Stripped-down counterpart to `CronSchedule`: `pause_status` and `sql_condition` are owned by the enclosing `TriggerConfiguration` and intentionally omitted here. """ + quartz_cron_expression: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -195,12 +35,11 @@ class CronTriggerConfiguration: """ @classmethod - def from_dict(cls, value: 'CronTriggerConfigurationDict') -> 'Self': + def from_dict(cls, value: "CronTriggerConfigurationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CronTriggerConfigurationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CronTriggerConfigurationDict": + return _transform_to_json_value(self) # type:ignore class CronTriggerConfigurationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/dashboard_task.py b/python/databricks/bundles/jobs/_models/dashboard_task.py index ccb28da7381..7d07ea50175 100644 --- a/python/databricks/bundles/jobs/_models/dashboard_task.py +++ b/python/databricks/bundles/jobs/_models/dashboard_task.py @@ -1,182 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrDict, VariableOrOptional +from databricks.bundles.jobs._models.subscription import ( + Subscription, + SubscriptionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DashboardTask: """ Configures the Lakeview Dashboard job task type. """ + dashboard_id: VariableOrOptional[str] = None """ The identifier of the dashboard to refresh. @@ -204,12 +48,11 @@ class DashboardTask: """ @classmethod - def from_dict(cls, value: 'DashboardTaskDict') -> 'Self': + def from_dict(cls, value: "DashboardTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DashboardTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DashboardTaskDict": + return _transform_to_json_value(self) # type:ignore class DashboardTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/data_security_mode.py b/python/databricks/bundles/jobs/_models/data_security_mode.py index b91c86dccd6..ae1aafcf085 100644 --- a/python/databricks/bundles/jobs/_models/data_security_mode.py +++ b/python/databricks/bundles/jobs/_models/data_security_mode.py @@ -1,195 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class DataSecurityMode(Enum): """ Data security mode decides what data governance model to use when accessing data from a cluster. - + * `DATA_SECURITY_MODE_AUTO`: Databricks will choose the most appropriate access mode depending on your compute configuration. * `DATA_SECURITY_MODE_STANDARD`: A secure cluster that can be shared by multiple users. Cluster users are fully isolated so that they cannot see each other’s data and credentials. Most data governance features are supported in this mode. But programming languages and cluster features might be limited. * `DATA_SECURITY_MODE_DEDICATED`: A secure cluster that can only be exclusively used by a single user specified in `single_user_name`. Most programming languages, cluster features and data governance features are available in this mode. - + The following modes are legacy aliases for the above modes: - + * `USER_ISOLATION`: Legacy alias for `DATA_SECURITY_MODE_STANDARD`. * `SINGLE_USER`: Legacy alias for `DATA_SECURITY_MODE_DEDICATED`. - + The following modes are deprecated starting with Databricks Runtime 15.0 and will be removed for future Databricks Runtime versions: - + * `LEGACY_TABLE_ACL`: This mode is for users migrating from legacy Table ACL clusters. * `LEGACY_PASSTHROUGH`: This mode is for users migrating from legacy Passthrough on high concurrency clusters. * `LEGACY_SINGLE_USER`: This mode is for users migrating from legacy Passthrough on standard clusters. * `LEGACY_SINGLE_USER_STANDARD`: This mode provides a way that doesn’t have UC nor passthrough enabled. """ + NONE = "NONE" SINGLE_USER = "SINGLE_USER" USER_ISOLATION = "USER_ISOLATION" @@ -201,4 +38,19 @@ class DataSecurityMode(Enum): DATA_SECURITY_MODE_DEDICATED = "DATA_SECURITY_MODE_DEDICATED" DATA_SECURITY_MODE_AUTO = "DATA_SECURITY_MODE_AUTO" -DataSecurityModeParam = Literal["NONE", "SINGLE_USER", "USER_ISOLATION", "LEGACY_TABLE_ACL", "LEGACY_PASSTHROUGH", "LEGACY_SINGLE_USER", "LEGACY_SINGLE_USER_STANDARD", "DATA_SECURITY_MODE_STANDARD", "DATA_SECURITY_MODE_DEDICATED", "DATA_SECURITY_MODE_AUTO"] | DataSecurityMode + +DataSecurityModeParam = ( + Literal[ + "NONE", + "SINGLE_USER", + "USER_ISOLATION", + "LEGACY_TABLE_ACL", + "LEGACY_PASSTHROUGH", + "LEGACY_SINGLE_USER", + "LEGACY_SINGLE_USER_STANDARD", + "DATA_SECURITY_MODE_STANDARD", + "DATA_SECURITY_MODE_DEDICATED", + "DATA_SECURITY_MODE_AUTO", + ] + | DataSecurityMode +) diff --git a/python/databricks/bundles/jobs/_models/day_of_week.py b/python/databricks/bundles/jobs/_models/day_of_week.py index 39e0e7263bb..e5874108dd6 100644 --- a/python/databricks/bundles/jobs/_models/day_of_week.py +++ b/python/databricks/bundles/jobs/_models/day_of_week.py @@ -1,177 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class DayOfWeek(Enum): """ Days of week that can be referenced by Jobs scheduling settings. """ + MONDAY = "MONDAY" TUESDAY = "TUESDAY" WEDNESDAY = "WEDNESDAY" @@ -180,4 +17,10 @@ class DayOfWeek(Enum): SATURDAY = "SATURDAY" SUNDAY = "SUNDAY" -DayOfWeekParam = Literal["MONDAY", "TUESDAY", "WEDNESDAY", "THURSDAY", "FRIDAY", "SATURDAY", "SUNDAY"] | DayOfWeek + +DayOfWeekParam = ( + Literal[ + "MONDAY", "TUESDAY", "WEDNESDAY", "THURSDAY", "FRIDAY", "SATURDAY", "SUNDAY" + ] + | DayOfWeek +) diff --git a/python/databricks/bundles/jobs/_models/dbfs_storage_info.py b/python/databricks/bundles/jobs/_models/dbfs_storage_info.py index 02ea147094a..29744acf26a 100644 --- a/python/databricks/bundles/jobs/_models/dbfs_storage_info.py +++ b/python/databricks/bundles/jobs/_models/dbfs_storage_info.py @@ -1,194 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DbfsStorageInfo: """ A storage location in DBFS """ + destination: VariableOr[str] """ dbfs destination, e.g. `dbfs:/my/path` """ @classmethod - def from_dict(cls, value: 'DbfsStorageInfoDict') -> 'Self': + def from_dict(cls, value: "DbfsStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DbfsStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DbfsStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class DbfsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/dbt_platform_task.py b/python/databricks/bundles/jobs/_models/dbt_platform_task.py index 1c32f08969f..61128b71634 100644 --- a/python/databricks/bundles/jobs/_models/dbt_platform_task.py +++ b/python/databricks/bundles/jobs/_models/dbt_platform_task.py @@ -1,182 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DbtPlatformTask: """ :meta private: [EXPERIMENTAL] """ + connection_resource_name: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -192,12 +32,11 @@ class DbtPlatformTask: """ @classmethod - def from_dict(cls, value: 'DbtPlatformTaskDict') -> 'Self': + def from_dict(cls, value: "DbtPlatformTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DbtPlatformTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DbtPlatformTaskDict": + return _transform_to_json_value(self) # type:ignore class DbtPlatformTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/dbt_task.py b/python/databricks/bundles/jobs/_models/dbt_task.py index 10ef2fd70f7..d8c92391e30 100644 --- a/python/databricks/bundles/jobs/_models/dbt_task.py +++ b/python/databricks/bundles/jobs/_models/dbt_task.py @@ -1,177 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional from databricks.bundles.jobs._models.source import Source, SourceParam -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DbtTask: """""" @@ -218,12 +58,11 @@ class DbtTask: """ @classmethod - def from_dict(cls, value: 'DbtTaskDict') -> 'Self': + def from_dict(cls, value: "DbtTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DbtTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DbtTaskDict": + return _transform_to_json_value(self) # type:ignore class DbtTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/dependency_mode.py b/python/databricks/bundles/jobs/_models/dependency_mode.py index 90de274cfd6..1fdc22cfa4c 100644 --- a/python/databricks/bundles/jobs/_models/dependency_mode.py +++ b/python/databricks/bundles/jobs/_models/dependency_mode.py @@ -1,183 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class DependencyMode(Enum): """ Controls dependency configuration for the cluster. - + * `DEPENDENCY_MODE_AUTO`: Databricks will choose the most appropriate dependency mode based on your compute configuration. * `DEPENDENCY_MODE_ENVIRONMENTS`: Enables a unified dependency management experience across classic and serverless, resulting in increased stability and performance. Supported only on DBR 19+ in Standard access mode. * `DEPENDENCY_MODE_CLUSTER_LIBRARIES`: Legacy mode: dependencies come from cluster libraries and init scripts. """ + DEPENDENCY_MODE_ENVIRONMENTS = "DEPENDENCY_MODE_ENVIRONMENTS" DEPENDENCY_MODE_CLUSTER_LIBRARIES = "DEPENDENCY_MODE_CLUSTER_LIBRARIES" DEPENDENCY_MODE_AUTO = "DEPENDENCY_MODE_AUTO" -DependencyModeParam = Literal["DEPENDENCY_MODE_ENVIRONMENTS", "DEPENDENCY_MODE_CLUSTER_LIBRARIES", "DEPENDENCY_MODE_AUTO"] | DependencyMode + +DependencyModeParam = ( + Literal[ + "DEPENDENCY_MODE_ENVIRONMENTS", + "DEPENDENCY_MODE_CLUSTER_LIBRARIES", + "DEPENDENCY_MODE_AUTO", + ] + | DependencyMode +) diff --git a/python/databricks/bundles/jobs/_models/deployment_spec.py b/python/databricks/bundles/jobs/_models/deployment_spec.py index 48a01c16135..3c0d8d77a6f 100644 --- a/python/databricks/bundles/jobs/_models/deployment_spec.py +++ b/python/databricks/bundles/jobs/_models/deployment_spec.py @@ -1,177 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.jobs._models.compute_spec import ( + ComputeSpec, + ComputeSpecParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DeploymentSpec: """ @@ -182,6 +25,7 @@ class DeploymentSpec: role-split workloads (driver + worker, parameter server, separate eval node, etc.) use multiple entries. """ + compute: VariableOr[ComputeSpec] """ [Public Preview] Compute resources allocated to each node in this deployment. @@ -198,12 +42,11 @@ class DeploymentSpec: """ @classmethod - def from_dict(cls, value: 'DeploymentSpecDict') -> 'Self': + def from_dict(cls, value: "DeploymentSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DeploymentSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DeploymentSpecDict": + return _transform_to_json_value(self) # type:ignore class DeploymentSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/docker_basic_auth.py b/python/databricks/bundles/jobs/_models/docker_basic_auth.py index 93f10ecfa99..552ea90a83b 100644 --- a/python/databricks/bundles/jobs/_models/docker_basic_auth.py +++ b/python/databricks/bundles/jobs/_models/docker_basic_auth.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DockerBasicAuth: """""" @@ -187,12 +26,11 @@ class DockerBasicAuth: """ @classmethod - def from_dict(cls, value: 'DockerBasicAuthDict') -> 'Self': + def from_dict(cls, value: "DockerBasicAuthDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DockerBasicAuthDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DockerBasicAuthDict": + return _transform_to_json_value(self) # type:ignore class DockerBasicAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/docker_image.py b/python/databricks/bundles/jobs/_models/docker_image.py index 5f542b48795..9dca00df1e2 100644 --- a/python/databricks/bundles/jobs/_models/docker_image.py +++ b/python/databricks/bundles/jobs/_models/docker_image.py @@ -1,177 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.docker_basic_auth import ( + DockerBasicAuth, + DockerBasicAuthParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DockerImage: """""" @@ -187,12 +30,11 @@ class DockerImage: """ @classmethod - def from_dict(cls, value: 'DockerImageDict') -> 'Self': + def from_dict(cls, value: "DockerImageDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DockerImageDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DockerImageDict": + return _transform_to_json_value(self) # type:ignore class DockerImageDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/ebs_volume_type.py b/python/databricks/bundles/jobs/_models/ebs_volume_type.py index 93277238be3..d8f3ef13f89 100644 --- a/python/databricks/bundles/jobs/_models/ebs_volume_type.py +++ b/python/databricks/bundles/jobs/_models/ebs_volume_type.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class EbsVolumeType(Enum): @@ -173,7 +9,11 @@ class EbsVolumeType(Enum): All EBS volume types that Databricks supports. See https://aws.amazon.com/ebs/details/ for details. """ + GENERAL_PURPOSE_SSD = "GENERAL_PURPOSE_SSD" THROUGHPUT_OPTIMIZED_HDD = "THROUGHPUT_OPTIMIZED_HDD" -EbsVolumeTypeParam = Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType + +EbsVolumeTypeParam = ( + Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType +) diff --git a/python/databricks/bundles/jobs/_models/environment.py b/python/databricks/bundles/jobs/_models/environment.py index 0b1e2077a18..850a1cf3098 100644 --- a/python/databricks/bundles/jobs/_models/environment.py +++ b/python/databricks/bundles/jobs/_models/environment.py @@ -1,183 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Environment: """ The environment entity used to preserve serverless environment side panel, jobs' environment for non-notebook task, and SDP's environment for classic and serverless pipelines. In this minimal environment spec, only pip and java dependencies are supported. """ + base_environment: VariableOrOptional[str] = None """ The base environment this environment is built on top of. A base environment defines the environment version and a @@ -221,12 +61,11 @@ class Environment: """ @classmethod - def from_dict(cls, value: 'EnvironmentDict') -> 'Self': + def from_dict(cls, value: "EnvironmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EnvironmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EnvironmentDict": + return _transform_to_json_value(self) # type:ignore class EnvironmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/file_arrival_trigger_configuration.py b/python/databricks/bundles/jobs/_models/file_arrival_trigger_configuration.py index aba1a82fab5..7e31a4b6946 100644 --- a/python/databricks/bundles/jobs/_models/file_arrival_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/file_arrival_trigger_configuration.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FileArrivalTriggerConfiguration: """""" @@ -195,12 +34,11 @@ class FileArrivalTriggerConfiguration: """ @classmethod - def from_dict(cls, value: 'FileArrivalTriggerConfigurationDict') -> 'Self': + def from_dict(cls, value: "FileArrivalTriggerConfigurationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FileArrivalTriggerConfigurationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FileArrivalTriggerConfigurationDict": + return _transform_to_json_value(self) # type:ignore class FileArrivalTriggerConfigurationDict(TypedDict, total=False): @@ -225,4 +63,6 @@ class FileArrivalTriggerConfigurationDict(TypedDict, total=False): """ -FileArrivalTriggerConfigurationParam = FileArrivalTriggerConfigurationDict | FileArrivalTriggerConfiguration +FileArrivalTriggerConfigurationParam = ( + FileArrivalTriggerConfigurationDict | FileArrivalTriggerConfiguration +) diff --git a/python/databricks/bundles/jobs/_models/for_each_task.py b/python/databricks/bundles/jobs/_models/for_each_task.py index b52421df3d8..c8ac1625ff6 100644 --- a/python/databricks/bundles/jobs/_models/for_each_task.py +++ b/python/databricks/bundles/jobs/_models/for_each_task.py @@ -1,177 +1,18 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: - from databricks.bundles.jobs._models.task import Task, TaskParam from typing_extensions import Self + from databricks.bundles.jobs._models.task import Task, TaskParam + + @dataclass(kw_only=True) class ForEachTask: """""" @@ -194,12 +35,11 @@ class ForEachTask: """ @classmethod - def from_dict(cls, value: 'ForEachTaskDict') -> 'Self': + def from_dict(cls, value: "ForEachTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ForEachTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ForEachTaskDict": + return _transform_to_json_value(self) # type:ignore class ForEachTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/gcp_attributes.py b/python/databricks/bundles/jobs/_models/gcp_attributes.py index 1e4e47efb17..2874f9da8e4 100644 --- a/python/databricks/bundles/jobs/_models/gcp_attributes.py +++ b/python/databricks/bundles/jobs/_models/gcp_attributes.py @@ -1,182 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.confidential_compute_type import ( + ConfidentialComputeType, + ConfidentialComputeTypeParam, +) +from databricks.bundles.jobs._models.gcp_availability import ( + GcpAvailability, + GcpAvailabilityParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GcpAttributes: """ Attributes set during cluster creation which are related to GCP. """ + availability: VariableOrOptional[GcpAvailability] = None """ This field determines whether the spark executors will be scheduled to run on preemptible @@ -243,12 +91,11 @@ class GcpAttributes: """ @classmethod - def from_dict(cls, value: 'GcpAttributesDict') -> 'Self': + def from_dict(cls, value: "GcpAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GcpAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GcpAttributesDict": + return _transform_to_json_value(self) # type:ignore class GcpAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/gcp_availability.py b/python/databricks/bundles/jobs/_models/gcp_availability.py index 2f1cdb6dea9..0d391b87fe3 100644 --- a/python/databricks/bundles/jobs/_models/gcp_availability.py +++ b/python/databricks/bundles/jobs/_models/gcp_availability.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class GcpAvailability(Enum): @@ -173,8 +9,13 @@ class GcpAvailability(Enum): This field determines whether the instance pool will contain preemptible VMs, on-demand VMs, or preemptible VMs with a fallback to on-demand VMs if the former is unavailable. """ + PREEMPTIBLE_GCP = "PREEMPTIBLE_GCP" ON_DEMAND_GCP = "ON_DEMAND_GCP" PREEMPTIBLE_WITH_FALLBACK_GCP = "PREEMPTIBLE_WITH_FALLBACK_GCP" -GcpAvailabilityParam = Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] | GcpAvailability + +GcpAvailabilityParam = ( + Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] + | GcpAvailability +) diff --git a/python/databricks/bundles/jobs/_models/gcs_storage_info.py b/python/databricks/bundles/jobs/_models/gcs_storage_info.py index 7dc8748f2fa..e8f1059373b 100644 --- a/python/databricks/bundles/jobs/_models/gcs_storage_info.py +++ b/python/databricks/bundles/jobs/_models/gcs_storage_info.py @@ -1,194 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GcsStorageInfo: """ A storage location in Google Cloud Platform's GCS """ + destination: VariableOr[str] """ GCS destination/URI, e.g. `gs://my-bucket/some-prefix` """ @classmethod - def from_dict(cls, value: 'GcsStorageInfoDict') -> 'Self': + def from_dict(cls, value: "GcsStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GcsStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GcsStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class GcsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/gen_ai_compute_task.py b/python/databricks/bundles/jobs/_models/gen_ai_compute_task.py index 4787305d82b..4e42c50f81d 100644 --- a/python/databricks/bundles/jobs/_models/gen_ai_compute_task.py +++ b/python/databricks/bundles/jobs/_models/gen_ai_compute_task.py @@ -1,182 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.jobs._models.compute_config import ( + ComputeConfig, + ComputeConfigParam, +) from databricks.bundles.jobs._models.source import Source, SourceParam -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GenAiComputeTask: """ :meta private: [EXPERIMENTAL] - + DEPRECATED — use `AiRuntimeTask` for all new BYOT multi-node GPU workloads (see ai_runtime_task.proto). `AiRuntimeTask` is the only supported BYOT task type for new workloads; this proto is retained only @@ -184,6 +28,7 @@ class GenAiComputeTask: removed once the pywheel → databricks-cli migration completes (post- PuPr). """ + dl_runtime_image: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -246,12 +91,11 @@ class GenAiComputeTask: """ @classmethod - def from_dict(cls, value: 'GenAiComputeTaskDict') -> 'Self': + def from_dict(cls, value: "GenAiComputeTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GenAiComputeTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GenAiComputeTaskDict": + return _transform_to_json_value(self) # type:ignore class GenAiComputeTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/git_provider.py b/python/databricks/bundles/jobs/_models/git_provider.py index d69bfbdfe9d..165f77de5cb 100644 --- a/python/databricks/bundles/jobs/_models/git_provider.py +++ b/python/databricks/bundles/jobs/_models/git_provider.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class GitProvider(Enum): @@ -178,4 +14,17 @@ class GitProvider(Enum): GIT_LAB_ENTERPRISE_EDITION = "gitLabEnterpriseEdition" AWS_CODE_COMMIT = "awsCodeCommit" -GitProviderParam = Literal["gitHub", "bitbucketCloud", "azureDevOpsServices", "gitHubEnterprise", "bitbucketServer", "gitLab", "gitLabEnterpriseEdition", "awsCodeCommit"] | GitProvider + +GitProviderParam = ( + Literal[ + "gitHub", + "bitbucketCloud", + "azureDevOpsServices", + "gitHubEnterprise", + "bitbucketServer", + "gitLab", + "gitLabEnterpriseEdition", + "awsCodeCommit", + ] + | GitProvider +) diff --git a/python/databricks/bundles/jobs/_models/git_source.py b/python/databricks/bundles/jobs/_models/git_source.py index bee4198df05..4ddd1d83764 100644 --- a/python/databricks/bundles/jobs/_models/git_source.py +++ b/python/databricks/bundles/jobs/_models/git_source.py @@ -1,186 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.jobs._models.sparse_checkout import ( + SparseCheckout, + SparseCheckoutParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GitSource: """ An optional specification for a remote Git repository containing the source code used by tasks. Version-controlled source code is supported by notebook, dbt, Python script, and SQL File tasks. - + If `git_source` is set, these tasks retrieve the file from the remote repository by default. However, this behavior can be overridden by setting `source` to `WORKSPACE` on the task. - + Note: dbt and SQL File tasks support only version-controlled sources. If dbt or SQL File tasks are used, `git_source` must be defined on the job. """ + git_provider: VariableOr[GitProvider] """ Unique identifier of the service used to host the Git repository. The value is case insensitive. @@ -209,12 +54,11 @@ class GitSource: sparse_checkout: VariableOrOptional[SparseCheckout] = None @classmethod - def from_dict(cls, value: 'GitSourceDict') -> 'Self': + def from_dict(cls, value: "GitSourceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GitSourceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GitSourceDict": + return _transform_to_json_value(self) # type:ignore class GitSourceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/hardware_accelerator_type.py b/python/databricks/bundles/jobs/_models/hardware_accelerator_type.py index 6aa2466d4ad..3d77c62b727 100644 --- a/python/databricks/bundles/jobs/_models/hardware_accelerator_type.py +++ b/python/databricks/bundles/jobs/_models/hardware_accelerator_type.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class HardwareAcceleratorType(Enum): @@ -174,9 +10,14 @@ class HardwareAcceleratorType(Enum): NOTE: This enum is referenced and is intended to be used by other Databricks services that need to specify hardware accelerator requirements for AI compute workloads. """ + GPU_1X_A10 = "GPU_1xA10" GPU_8X_H100 = "GPU_8xH100" GPU_1X_H100 = "GPU_1xH100" GPU_8X_B300 = "GPU_8xB300" -HardwareAcceleratorTypeParam = Literal["GPU_1xA10", "GPU_8xH100", "GPU_1xH100", "GPU_8xB300"] | HardwareAcceleratorType + +HardwareAcceleratorTypeParam = ( + Literal["GPU_1xA10", "GPU_8xH100", "GPU_1xH100", "GPU_8xB300"] + | HardwareAcceleratorType +) diff --git a/python/databricks/bundles/jobs/_models/init_script_info.py b/python/databricks/bundles/jobs/_models/init_script_info.py index 854d00dc6f8..a7c5800a594 100644 --- a/python/databricks/bundles/jobs/_models/init_script_info.py +++ b/python/databricks/bundles/jobs/_models/init_script_info.py @@ -1,182 +1,50 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.adlsgen2_info import ( + Adlsgen2Info, + Adlsgen2InfoParam, +) +from databricks.bundles.jobs._models.dbfs_storage_info import ( + DbfsStorageInfo, + DbfsStorageInfoParam, +) +from databricks.bundles.jobs._models.gcs_storage_info import ( + GcsStorageInfo, + GcsStorageInfoParam, +) +from databricks.bundles.jobs._models.local_file_info import ( + LocalFileInfo, + LocalFileInfoParam, +) +from databricks.bundles.jobs._models.s3_storage_info import ( + S3StorageInfo, + S3StorageInfoParam, +) +from databricks.bundles.jobs._models.volumes_storage_info import ( + VolumesStorageInfo, + VolumesStorageInfoParam, +) +from databricks.bundles.jobs._models.workspace_storage_info import ( + WorkspaceStorageInfo, + WorkspaceStorageInfoParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InitScriptInfo: """ Config for an individual init script """ + abfss: VariableOrOptional[Adlsgen2Info] = None """ Contains the Azure Data Lake Storage destination path @@ -221,12 +89,11 @@ class InitScriptInfo: """ @classmethod - def from_dict(cls, value: 'InitScriptInfoDict') -> 'Self': + def from_dict(cls, value: "InitScriptInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InitScriptInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InitScriptInfoDict": + return _transform_to_json_value(self) # type:ignore class InitScriptInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job.py b/python/databricks/bundles/jobs/_models/job.py index e3df4756349..e38305c7d02 100644 --- a/python/databricks/bundles/jobs/_models/job.py +++ b/python/databricks/bundles/jobs/_models/job.py @@ -1,177 +1,82 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.jobs._models.continuous import ( + Continuous, + ContinuousParam, +) +from databricks.bundles.jobs._models.cron_schedule import ( + CronSchedule, + CronScheduleParam, +) +from databricks.bundles.jobs._models.git_source import GitSource, GitSourceParam +from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterParam +from databricks.bundles.jobs._models.job_email_notifications import ( + JobEmailNotifications, + JobEmailNotificationsParam, +) +from databricks.bundles.jobs._models.job_environment import ( + JobEnvironment, + JobEnvironmentParam, +) +from databricks.bundles.jobs._models.job_environment_variables import ( + JobEnvironmentVariables, + JobEnvironmentVariablesParam, +) +from databricks.bundles.jobs._models.job_notification_settings import ( + JobNotificationSettings, + JobNotificationSettingsParam, +) +from databricks.bundles.jobs._models.job_parameter_definition import ( + JobParameterDefinition, + JobParameterDefinitionParam, +) +from databricks.bundles.jobs._models.job_permission import ( + JobPermission, + JobPermissionParam, +) +from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsParam +from databricks.bundles.jobs._models.jobs_health_rules import ( + JobsHealthRules, + JobsHealthRulesParam, +) +from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.jobs._models.performance_target import ( + PerformanceTarget, + PerformanceTargetParam, +) +from databricks.bundles.jobs._models.queue_settings import ( + QueueSettings, + QueueSettingsParam, +) +from databricks.bundles.jobs._models.task import Task, TaskParam +from databricks.bundles.jobs._models.trigger_configuration import ( + TriggerConfiguration, + TriggerConfigurationParam, +) +from databricks.bundles.jobs._models.trigger_settings import ( + TriggerSettings, + TriggerSettingsParam, +) +from databricks.bundles.jobs._models.webhook_notifications import ( + WebhookNotifications, + WebhookNotificationsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Job(Resource): """""" @@ -210,7 +115,9 @@ class Job(Resource): An optional set of email addresses that is notified when runs of this job begin or complete as well as when this job is deleted. """ - environment_variables: VariableOrList[JobEnvironmentVariables] = field(default_factory=list) + environment_variables: VariableOrList[JobEnvironmentVariables] = field( + default_factory=list + ) """ :meta private: [EXPERIMENTAL] @@ -361,12 +268,11 @@ class Job(Resource): """ @classmethod - def from_dict(cls, value: 'JobDict') -> 'Self': + def from_dict(cls, value: "JobDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobDict": + return _transform_to_json_value(self) # type:ignore class JobDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_cluster.py b/python/databricks/bundles/jobs/_models/job_cluster.py index 57cffd54587..8c3390db857 100644 --- a/python/databricks/bundles/jobs/_models/job_cluster.py +++ b/python/databricks/bundles/jobs/_models/job_cluster.py @@ -1,177 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecParam if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobCluster: """""" @@ -197,12 +37,11 @@ class JobCluster: """ @classmethod - def from_dict(cls, value: 'JobClusterDict') -> 'Self': + def from_dict(cls, value: "JobClusterDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobClusterDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobClusterDict": + return _transform_to_json_value(self) # type:ignore class JobClusterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_email_notifications.py b/python/databricks/bundles/jobs/_models/job_email_notifications.py index 89d7472177e..f6bb6fdd548 100644 --- a/python/databricks/bundles/jobs/_models/job_email_notifications.py +++ b/python/databricks/bundles/jobs/_models/job_email_notifications.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobEmailNotifications: """""" @@ -182,7 +21,9 @@ class JobEmailNotifications: This field is `deprecated`. Please use the `notification_settings.no_alert_for_skipped_runs` field. """ - on_duration_warning_threshold_exceeded: VariableOrList[str] = field(default_factory=list) + on_duration_warning_threshold_exceeded: VariableOrList[str] = field( + default_factory=list + ) """ A list of email addresses to be notified when the duration of a run exceeds the threshold specified for the `RUN_DURATION_SECONDS` metric in the `health` field. If no rule for the `RUN_DURATION_SECONDS` metric is specified in the `health` field for the job, notifications are not sent. """ @@ -220,12 +61,11 @@ class JobEmailNotifications: """ @classmethod - def from_dict(cls, value: 'JobEmailNotificationsDict') -> 'Self': + def from_dict(cls, value: "JobEmailNotificationsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobEmailNotificationsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobEmailNotificationsDict": + return _transform_to_json_value(self) # type:ignore class JobEmailNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_environment.py b/python/databricks/bundles/jobs/_models/job_environment.py index a698e361c5d..c71ee45dbd6 100644 --- a/python/databricks/bundles/jobs/_models/job_environment.py +++ b/python/databricks/bundles/jobs/_models/job_environment.py @@ -1,177 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.jobs._models.environment import Environment, EnvironmentParam if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobEnvironment: """""" @@ -188,12 +28,11 @@ class JobEnvironment: """ @classmethod - def from_dict(cls, value: 'JobEnvironmentDict') -> 'Self': + def from_dict(cls, value: "JobEnvironmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobEnvironmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobEnvironmentDict": + return _transform_to_json_value(self) # type:ignore class JobEnvironmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_environment_variables.py b/python/databricks/bundles/jobs/_models/job_environment_variables.py index f4bf0a58cb4..aca6fd61d7d 100644 --- a/python/databricks/bundles/jobs/_models/job_environment_variables.py +++ b/python/databricks/bundles/jobs/_models/job_environment_variables.py @@ -1,177 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.jobs._models.job_environment_variables_spec import ( + JobEnvironmentVariablesSpec, + JobEnvironmentVariablesSpecParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobEnvironmentVariables: """ @@ -180,6 +23,7 @@ class JobEnvironmentVariables: `JobSettings.environment_variables`, and tasks select one via `TaskSettings.environment_variables_key`. """ + environment_variables_key: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -197,12 +41,11 @@ class JobEnvironmentVariables: """ @classmethod - def from_dict(cls, value: 'JobEnvironmentVariablesDict') -> 'Self': + def from_dict(cls, value: "JobEnvironmentVariablesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobEnvironmentVariablesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobEnvironmentVariablesDict": + return _transform_to_json_value(self) # type:ignore class JobEnvironmentVariablesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_environment_variables_spec.py b/python/databricks/bundles/jobs/_models/job_environment_variables_spec.py index 1cbdfb3c906..907d2914d57 100644 --- a/python/databricks/bundles/jobs/_models/job_environment_variables_spec.py +++ b/python/databricks/bundles/jobs/_models/job_environment_variables_spec.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrDict, VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobEnvironmentVariablesSpec: """ @@ -179,6 +18,7 @@ class JobEnvironmentVariablesSpec: variable entry. Runtime environment variables override inline `variables`, which override values from `files`, on duplicate keys. """ + files: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -218,12 +58,11 @@ class JobEnvironmentVariablesSpec: """ @classmethod - def from_dict(cls, value: 'JobEnvironmentVariablesSpecDict') -> 'Self': + def from_dict(cls, value: "JobEnvironmentVariablesSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobEnvironmentVariablesSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobEnvironmentVariablesSpecDict": + return _transform_to_json_value(self) # type:ignore class JobEnvironmentVariablesSpecDict(TypedDict, total=False): @@ -268,4 +107,6 @@ class JobEnvironmentVariablesSpecDict(TypedDict, total=False): """ -JobEnvironmentVariablesSpecParam = JobEnvironmentVariablesSpecDict | JobEnvironmentVariablesSpec +JobEnvironmentVariablesSpecParam = ( + JobEnvironmentVariablesSpecDict | JobEnvironmentVariablesSpec +) diff --git a/python/databricks/bundles/jobs/_models/job_notification_settings.py b/python/databricks/bundles/jobs/_models/job_notification_settings.py index a54982135d3..eb6ca454186 100644 --- a/python/databricks/bundles/jobs/_models/job_notification_settings.py +++ b/python/databricks/bundles/jobs/_models/job_notification_settings.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobNotificationSettings: """""" @@ -187,12 +26,11 @@ class JobNotificationSettings: """ @classmethod - def from_dict(cls, value: 'JobNotificationSettingsDict') -> 'Self': + def from_dict(cls, value: "JobNotificationSettingsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobNotificationSettingsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobNotificationSettingsDict": + return _transform_to_json_value(self) # type:ignore class JobNotificationSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_parameter_definition.py b/python/databricks/bundles/jobs/_models/job_parameter_definition.py index 211b052d70a..3e81a538218 100644 --- a/python/databricks/bundles/jobs/_models/job_parameter_definition.py +++ b/python/databricks/bundles/jobs/_models/job_parameter_definition.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobParameterDefinition: """""" @@ -187,12 +26,11 @@ class JobParameterDefinition: """ @classmethod - def from_dict(cls, value: 'JobParameterDefinitionDict') -> 'Self': + def from_dict(cls, value: "JobParameterDefinitionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobParameterDefinitionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobParameterDefinitionDict": + return _transform_to_json_value(self) # type:ignore class JobParameterDefinitionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_permission.py b/python/databricks/bundles/jobs/_models/job_permission.py index b5e1a1b8e35..6864bf99e23 100644 --- a/python/databricks/bundles/jobs/_models/job_permission.py +++ b/python/databricks/bundles/jobs/_models/job_permission.py @@ -1,177 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.jobs._models.job_permission_level import ( + JobPermissionLevel, + JobPermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobPermission: """""" @@ -197,12 +40,11 @@ class JobPermission: """ @classmethod - def from_dict(cls, value: 'JobPermissionDict') -> 'Self': + def from_dict(cls, value: "JobPermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobPermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobPermissionDict": + return _transform_to_json_value(self) # type:ignore class JobPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/job_permission_level.py b/python/databricks/bundles/jobs/_models/job_permission_level.py index 88a4481b22b..e56c96b09d2 100644 --- a/python/databricks/bundles/jobs/_models/job_permission_level.py +++ b/python/databricks/bundles/jobs/_models/job_permission_level.py @@ -1,180 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class JobPermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" IS_OWNER = "IS_OWNER" CAN_MANAGE_RUN = "CAN_MANAGE_RUN" CAN_VIEW = "CAN_VIEW" -JobPermissionLevelParam = Literal["CAN_MANAGE", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW"] | JobPermissionLevel + +JobPermissionLevelParam = ( + Literal["CAN_MANAGE", "IS_OWNER", "CAN_MANAGE_RUN", "CAN_VIEW"] | JobPermissionLevel +) diff --git a/python/databricks/bundles/jobs/_models/job_run_as.py b/python/databricks/bundles/jobs/_models/job_run_as.py index b1df2878fba..6242eafad59 100644 --- a/python/databricks/bundles/jobs/_models/job_run_as.py +++ b/python/databricks/bundles/jobs/_models/job_run_as.py @@ -1,184 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobRunAs: """ Write-only setting. Specifies the user, service principal, or group that the job runs as. If not specified, the job runs as the user who created the job. - + One of `user_name`, `service_principal_name`, or `group_name` should be specified. If not, an error is thrown. """ + group_name: VariableOrOptional[str] = None """ Group name of an account group assigned to the workspace. When set, all tasks run as the group and the group's permissions are used for data access. Setting this field requires being a member of the group, or having the `Assume` permission on the group. @@ -195,12 +35,11 @@ class JobRunAs: """ @classmethod - def from_dict(cls, value: 'JobRunAsDict') -> 'Self': + def from_dict(cls, value: "JobRunAsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobRunAsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobRunAsDict": + return _transform_to_json_value(self) # type:ignore class JobRunAsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/jobs_health_metric.py b/python/databricks/bundles/jobs/_models/jobs_health_metric.py index 3adecf417cf..b662439d3ef 100644 --- a/python/databricks/bundles/jobs/_models/jobs_health_metric.py +++ b/python/databricks/bundles/jobs/_models/jobs_health_metric.py @@ -1,187 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class JobsHealthMetric(Enum): """ Specifies the health metric that is being evaluated for a particular health rule. - + * `RUN_DURATION_SECONDS`: Expected total time for a run in seconds. * `STREAMING_BACKLOG_BYTES`: An estimate of the maximum bytes of data waiting to be consumed across all streams. This metric is in Public Preview. * `STREAMING_BACKLOG_RECORDS`: An estimate of the maximum offset lag across all streams. This metric is in Public Preview. * `STREAMING_BACKLOG_SECONDS`: An estimate of the maximum consumer delay across all streams. This metric is in Public Preview. * `STREAMING_BACKLOG_FILES`: An estimate of the maximum number of outstanding files across all streams. This metric is in Public Preview. """ + RUN_DURATION_SECONDS = "RUN_DURATION_SECONDS" STREAMING_BACKLOG_BYTES = "STREAMING_BACKLOG_BYTES" STREAMING_BACKLOG_RECORDS = "STREAMING_BACKLOG_RECORDS" STREAMING_BACKLOG_SECONDS = "STREAMING_BACKLOG_SECONDS" STREAMING_BACKLOG_FILES = "STREAMING_BACKLOG_FILES" -JobsHealthMetricParam = Literal["RUN_DURATION_SECONDS", "STREAMING_BACKLOG_BYTES", "STREAMING_BACKLOG_RECORDS", "STREAMING_BACKLOG_SECONDS", "STREAMING_BACKLOG_FILES"] | JobsHealthMetric + +JobsHealthMetricParam = ( + Literal[ + "RUN_DURATION_SECONDS", + "STREAMING_BACKLOG_BYTES", + "STREAMING_BACKLOG_RECORDS", + "STREAMING_BACKLOG_SECONDS", + "STREAMING_BACKLOG_FILES", + ] + | JobsHealthMetric +) diff --git a/python/databricks/bundles/jobs/_models/jobs_health_operator.py b/python/databricks/bundles/jobs/_models/jobs_health_operator.py index 478e48bb561..3de343a8b6a 100644 --- a/python/databricks/bundles/jobs/_models/jobs_health_operator.py +++ b/python/databricks/bundles/jobs/_models/jobs_health_operator.py @@ -1,177 +1,15 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class JobsHealthOperator(Enum): """ Specifies the operator used to compare the health metric value with the specified threshold. """ + GREATER_THAN = "GREATER_THAN" + JobsHealthOperatorParam = Literal["GREATER_THAN"] | JobsHealthOperator diff --git a/python/databricks/bundles/jobs/_models/jobs_health_rule.py b/python/databricks/bundles/jobs/_models/jobs_health_rule.py index ed9f08952c2..59447874eda 100644 --- a/python/databricks/bundles/jobs/_models/jobs_health_rule.py +++ b/python/databricks/bundles/jobs/_models/jobs_health_rule.py @@ -1,177 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr +from databricks.bundles.jobs._models.jobs_health_metric import ( + JobsHealthMetric, + JobsHealthMetricParam, +) +from databricks.bundles.jobs._models.jobs_health_operator import ( + JobsHealthOperator, + JobsHealthOperatorParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobsHealthRule: """""" @@ -198,12 +45,11 @@ class JobsHealthRule: """ @classmethod - def from_dict(cls, value: 'JobsHealthRuleDict') -> 'Self': + def from_dict(cls, value: "JobsHealthRuleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobsHealthRuleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobsHealthRuleDict": + return _transform_to_json_value(self) # type:ignore class JobsHealthRuleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/jobs_health_rules.py b/python/databricks/bundles/jobs/_models/jobs_health_rules.py index 0754e1d108c..daf613406b5 100644 --- a/python/databricks/bundles/jobs/_models/jobs_health_rules.py +++ b/python/databricks/bundles/jobs/_models/jobs_health_rules.py @@ -1,191 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.jobs._models.jobs_health_rule import ( + JobsHealthRule, + JobsHealthRuleParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JobsHealthRules: """ An optional set of health rules that can be defined for this job. """ + rules: VariableOrList[JobsHealthRule] = field(default_factory=list) @classmethod - def from_dict(cls, value: 'JobsHealthRulesDict') -> 'Self': + def from_dict(cls, value: "JobsHealthRulesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JobsHealthRulesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JobsHealthRulesDict": + return _transform_to_json_value(self) # type:ignore class JobsHealthRulesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/kind.py b/python/databricks/bundles/jobs/_models/kind.py index 15e60a1c5d9..8614e25dd52 100644 --- a/python/databricks/bundles/jobs/_models/kind.py +++ b/python/databricks/bundles/jobs/_models/kind.py @@ -1,185 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class Kind(Enum): """ The kind of compute described by this compute specification. - + Depending on `kind`, different validations and default values will be applied. - + Clusters with `kind = CLASSIC_PREVIEW` support the following fields, whereas clusters with no specified `kind` do not. * [is_single_node](/api/workspace/clusters/create#is_single_node) * [use_ml_runtime](/api/workspace/clusters/create#use_ml_runtime) - + By using the [simple form](https://docs.databricks.com/compute/simple-form.html), your clusters are automatically using `kind = CLASSIC_PREVIEW`. """ + CLASSIC_PREVIEW = "CLASSIC_PREVIEW" + KindParam = Literal["CLASSIC_PREVIEW"] | Kind diff --git a/python/databricks/bundles/jobs/_models/library.py b/python/databricks/bundles/jobs/_models/library.py index a7158011d72..29d4c5b1fdd 100644 --- a/python/databricks/bundles/jobs/_models/library.py +++ b/python/databricks/bundles/jobs/_models/library.py @@ -1,177 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.maven_library import ( + MavenLibrary, + MavenLibraryParam, +) +from databricks.bundles.jobs._models.python_py_pi_library import ( + PythonPyPiLibrary, + PythonPyPiLibraryParam, +) +from databricks.bundles.jobs._models.r_cran_library import ( + RCranLibrary, + RCranLibraryParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Library: """""" @@ -223,12 +74,11 @@ class Library: """ @classmethod - def from_dict(cls, value: 'LibraryDict') -> 'Self': + def from_dict(cls, value: "LibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LibraryDict": + return _transform_to_json_value(self) # type:ignore class LibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/lifecycle.py b/python/databricks/bundles/jobs/_models/lifecycle.py index f21f05311e9..697776a198e 100644 --- a/python/databricks/bundles/jobs/_models/lifecycle.py +++ b/python/databricks/bundles/jobs/_models/lifecycle.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -182,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/local_file_info.py b/python/databricks/bundles/jobs/_models/local_file_info.py index 1ed6df8e8b9..875f2fe8353 100644 --- a/python/databricks/bundles/jobs/_models/local_file_info.py +++ b/python/databricks/bundles/jobs/_models/local_file_info.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LocalFileInfo: """""" @@ -182,12 +21,11 @@ class LocalFileInfo: """ @classmethod - def from_dict(cls, value: 'LocalFileInfoDict') -> 'Self': + def from_dict(cls, value: "LocalFileInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LocalFileInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LocalFileInfoDict": + return _transform_to_json_value(self) # type:ignore class LocalFileInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/log_analytics_info.py b/python/databricks/bundles/jobs/_models/log_analytics_info.py index 1a4646a2154..67cdf85b2b9 100644 --- a/python/databricks/bundles/jobs/_models/log_analytics_info.py +++ b/python/databricks/bundles/jobs/_models/log_analytics_info.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LogAnalyticsInfo: """""" @@ -187,12 +26,11 @@ class LogAnalyticsInfo: """ @classmethod - def from_dict(cls, value: 'LogAnalyticsInfoDict') -> 'Self': + def from_dict(cls, value: "LogAnalyticsInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LogAnalyticsInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LogAnalyticsInfoDict": + return _transform_to_json_value(self) # type:ignore class LogAnalyticsInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/maintenance_window.py b/python/databricks/bundles/jobs/_models/maintenance_window.py index 0ac9f824faa..180c65e75cc 100644 --- a/python/databricks/bundles/jobs/_models/maintenance_window.py +++ b/python/databricks/bundles/jobs/_models/maintenance_window.py @@ -1,183 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - +from databricks.bundles.core._variable import VariableOr from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MaintenanceWindow: """ A recurring weekly time window during which platform-initiated maintenance is allowed to run for a continuous job. """ + day_of_week: VariableOr[DayOfWeek] """ The day of week on which maintenance is allowed to happen. This field is required. @@ -196,12 +37,11 @@ class MaintenanceWindow: """ @classmethod - def from_dict(cls, value: 'MaintenanceWindowDict') -> 'Self': + def from_dict(cls, value: "MaintenanceWindowDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MaintenanceWindowDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MaintenanceWindowDict": + return _transform_to_json_value(self) # type:ignore class MaintenanceWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/maven_library.py b/python/databricks/bundles/jobs/_models/maven_library.py index 681257d6b53..25be1599c28 100644 --- a/python/databricks/bundles/jobs/_models/maven_library.py +++ b/python/databricks/bundles/jobs/_models/maven_library.py @@ -1,177 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MavenLibrary: """""" @@ -196,12 +39,11 @@ class MavenLibrary: """ @classmethod - def from_dict(cls, value: 'MavenLibraryDict') -> 'Self': + def from_dict(cls, value: "MavenLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MavenLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MavenLibraryDict": + return _transform_to_json_value(self) # type:ignore class MavenLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/model_trigger_configuration.py b/python/databricks/bundles/jobs/_models/model_trigger_configuration.py index 389371f378d..29ba7ed1479 100644 --- a/python/databricks/bundles/jobs/_models/model_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/model_trigger_configuration.py @@ -1,182 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.jobs._models.model_trigger_configuration_condition import ( + ModelTriggerConfigurationCondition, + ModelTriggerConfigurationConditionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelTriggerConfiguration: """ :meta private: [EXPERIMENTAL] """ + condition: VariableOr[ModelTriggerConfigurationCondition] """ :meta private: [EXPERIMENTAL] @@ -217,12 +65,11 @@ class ModelTriggerConfiguration: """ @classmethod - def from_dict(cls, value: 'ModelTriggerConfigurationDict') -> 'Self': + def from_dict(cls, value: "ModelTriggerConfigurationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelTriggerConfigurationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelTriggerConfigurationDict": + return _transform_to_json_value(self) # type:ignore class ModelTriggerConfigurationDict(TypedDict, total=False): @@ -268,4 +115,6 @@ class ModelTriggerConfigurationDict(TypedDict, total=False): """ -ModelTriggerConfigurationParam = ModelTriggerConfigurationDict | ModelTriggerConfiguration +ModelTriggerConfigurationParam = ( + ModelTriggerConfigurationDict | ModelTriggerConfiguration +) diff --git a/python/databricks/bundles/jobs/_models/model_trigger_configuration_condition.py b/python/databricks/bundles/jobs/_models/model_trigger_configuration_condition.py index 99adeadbbed..013f413303b 100644 --- a/python/databricks/bundles/jobs/_models/model_trigger_configuration_condition.py +++ b/python/databricks/bundles/jobs/_models/model_trigger_configuration_condition.py @@ -1,179 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class ModelTriggerConfigurationCondition(Enum): """ :meta private: [EXPERIMENTAL] """ + MODEL_CREATED = "MODEL_CREATED" MODEL_VERSION_READY = "MODEL_VERSION_READY" MODEL_ALIAS_SET = "MODEL_ALIAS_SET" -ModelTriggerConfigurationConditionParam = Literal["MODEL_CREATED", "MODEL_VERSION_READY", "MODEL_ALIAS_SET"] | ModelTriggerConfigurationCondition + +ModelTriggerConfigurationConditionParam = ( + Literal["MODEL_CREATED", "MODEL_VERSION_READY", "MODEL_ALIAS_SET"] + | ModelTriggerConfigurationCondition +) diff --git a/python/databricks/bundles/jobs/_models/node_type_flexibility.py b/python/databricks/bundles/jobs/_models/node_type_flexibility.py index 1eb9aef0a90..418ea9337bd 100644 --- a/python/databricks/bundles/jobs/_models/node_type_flexibility.py +++ b/python/databricks/bundles/jobs/_models/node_type_flexibility.py @@ -1,182 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class NodeTypeFlexibility: """ Configuration for flexible node types, allowing fallback to alternate node types during cluster launch and upscale. """ + alternate_node_type_ids: VariableOrList[str] = field(default_factory=list) """ A list of node type IDs to use as fallbacks when the primary node type is unavailable. @@ -189,12 +29,11 @@ class NodeTypeFlexibility: """ @classmethod - def from_dict(cls, value: 'NodeTypeFlexibilityDict') -> 'Self': + def from_dict(cls, value: "NodeTypeFlexibilityDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'NodeTypeFlexibilityDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "NodeTypeFlexibilityDict": + return _transform_to_json_value(self) # type:ignore class NodeTypeFlexibilityDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/notebook_task.py b/python/databricks/bundles/jobs/_models/notebook_task.py index f35946725a3..7b4329392db 100644 --- a/python/databricks/bundles/jobs/_models/notebook_task.py +++ b/python/databricks/bundles/jobs/_models/notebook_task.py @@ -1,177 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrOptional, +) from databricks.bundles.jobs._models.source import Source, SourceParam -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class NotebookTask: """""" @@ -214,12 +58,11 @@ class NotebookTask: """ @classmethod - def from_dict(cls, value: 'NotebookTaskDict') -> 'Self': + def from_dict(cls, value: "NotebookTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'NotebookTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "NotebookTaskDict": + return _transform_to_json_value(self) # type:ignore class NotebookTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/pause_status.py b/python/databricks/bundles/jobs/_models/pause_status.py index 19bb45bf6ac..9e5044fb880 100644 --- a/python/databricks/bundles/jobs/_models/pause_status.py +++ b/python/databricks/bundles/jobs/_models/pause_status.py @@ -1,175 +1,12 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class PauseStatus(Enum): UNPAUSED = "UNPAUSED" PAUSED = "PAUSED" + PauseStatusParam = Literal["UNPAUSED", "PAUSED"] | PauseStatus diff --git a/python/databricks/bundles/jobs/_models/performance_target.py b/python/databricks/bundles/jobs/_models/performance_target.py index b26e2ed6d46..8dbe7e4a435 100644 --- a/python/databricks/bundles/jobs/_models/performance_target.py +++ b/python/databricks/bundles/jobs/_models/performance_target.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class PerformanceTarget(Enum): @@ -174,7 +10,11 @@ class PerformanceTarget(Enum): The performance mode on the job or pipeline should map to a performance setting that is passed to Cluster Manager (see cluster-common PerformanceTarget). """ + PERFORMANCE_OPTIMIZED = "PERFORMANCE_OPTIMIZED" STANDARD = "STANDARD" -PerformanceTargetParam = Literal["PERFORMANCE_OPTIMIZED", "STANDARD"] | PerformanceTarget + +PerformanceTargetParam = ( + Literal["PERFORMANCE_OPTIMIZED", "STANDARD"] | PerformanceTarget +) diff --git a/python/databricks/bundles/jobs/_models/periodic_trigger_configuration.py b/python/databricks/bundles/jobs/_models/periodic_trigger_configuration.py index 288f44df438..2e7215aad06 100644 --- a/python/databricks/bundles/jobs/_models/periodic_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/periodic_trigger_configuration.py @@ -1,177 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr +from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import ( + PeriodicTriggerConfigurationTimeUnit, + PeriodicTriggerConfigurationTimeUnitParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PeriodicTriggerConfiguration: """""" @@ -187,12 +30,11 @@ class PeriodicTriggerConfiguration: """ @classmethod - def from_dict(cls, value: 'PeriodicTriggerConfigurationDict') -> 'Self': + def from_dict(cls, value: "PeriodicTriggerConfigurationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PeriodicTriggerConfigurationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PeriodicTriggerConfigurationDict": + return _transform_to_json_value(self) # type:ignore class PeriodicTriggerConfigurationDict(TypedDict, total=False): @@ -209,4 +51,6 @@ class PeriodicTriggerConfigurationDict(TypedDict, total=False): """ -PeriodicTriggerConfigurationParam = PeriodicTriggerConfigurationDict | PeriodicTriggerConfiguration +PeriodicTriggerConfigurationParam = ( + PeriodicTriggerConfigurationDict | PeriodicTriggerConfiguration +) diff --git a/python/databricks/bundles/jobs/_models/periodic_trigger_configuration_time_unit.py b/python/databricks/bundles/jobs/_models/periodic_trigger_configuration_time_unit.py index 58680def8f0..e035958b12c 100644 --- a/python/databricks/bundles/jobs/_models/periodic_trigger_configuration_time_unit.py +++ b/python/databricks/bundles/jobs/_models/periodic_trigger_configuration_time_unit.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class PeriodicTriggerConfigurationTimeUnit(Enum): @@ -174,4 +10,7 @@ class PeriodicTriggerConfigurationTimeUnit(Enum): WEEKS = "WEEKS" MINUTES = "MINUTES" -PeriodicTriggerConfigurationTimeUnitParam = Literal["HOURS", "DAYS", "WEEKS", "MINUTES"] | PeriodicTriggerConfigurationTimeUnit + +PeriodicTriggerConfigurationTimeUnitParam = ( + Literal["HOURS", "DAYS", "WEEKS", "MINUTES"] | PeriodicTriggerConfigurationTimeUnit +) diff --git a/python/databricks/bundles/jobs/_models/pipeline_params.py b/python/databricks/bundles/jobs/_models/pipeline_params.py index c554bdfa76b..98550f271fb 100644 --- a/python/databricks/bundles/jobs/_models/pipeline_params.py +++ b/python/databricks/bundles/jobs/_models/pipeline_params.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PipelineParams: """""" @@ -203,12 +42,11 @@ class PipelineParams: """ @classmethod - def from_dict(cls, value: 'PipelineParamsDict') -> 'Self': + def from_dict(cls, value: "PipelineParamsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PipelineParamsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PipelineParamsDict": + return _transform_to_json_value(self) # type:ignore class PipelineParamsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/pipeline_task.py b/python/databricks/bundles/jobs/_models/pipeline_task.py index 4af5e8a39d6..87d720fcd69 100644 --- a/python/databricks/bundles/jobs/_models/pipeline_task.py +++ b/python/databricks/bundles/jobs/_models/pipeline_task.py @@ -1,177 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PipelineTask: """""" @@ -216,12 +60,11 @@ class PipelineTask: """ @classmethod - def from_dict(cls, value: 'PipelineTaskDict') -> 'Self': + def from_dict(cls, value: "PipelineTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PipelineTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PipelineTaskDict": + return _transform_to_json_value(self) # type:ignore class PipelineTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/power_bi_model.py b/python/databricks/bundles/jobs/_models/power_bi_model.py index d30147df0be..f791b1901a8 100644 --- a/python/databricks/bundles/jobs/_models/power_bi_model.py +++ b/python/databricks/bundles/jobs/_models/power_bi_model.py @@ -1,177 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.authentication_method import ( + AuthenticationMethod, + AuthenticationMethodParam, +) from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PowerBiModel: """""" @@ -202,12 +46,11 @@ class PowerBiModel: """ @classmethod - def from_dict(cls, value: 'PowerBiModelDict') -> 'Self': + def from_dict(cls, value: "PowerBiModelDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PowerBiModelDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PowerBiModelDict": + return _transform_to_json_value(self) # type:ignore class PowerBiModelDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/power_bi_table.py b/python/databricks/bundles/jobs/_models/power_bi_table.py index e7df4faff75..368f432c45d 100644 --- a/python/databricks/bundles/jobs/_models/power_bi_table.py +++ b/python/databricks/bundles/jobs/_models/power_bi_table.py @@ -1,177 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - +from databricks.bundles.core._variable import VariableOrOptional from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PowerBiTable: """""" @@ -197,12 +37,11 @@ class PowerBiTable: """ @classmethod - def from_dict(cls, value: 'PowerBiTableDict') -> 'Self': + def from_dict(cls, value: "PowerBiTableDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PowerBiTableDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PowerBiTableDict": + return _transform_to_json_value(self) # type:ignore class PowerBiTableDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/power_bi_task.py b/python/databricks/bundles/jobs/_models/power_bi_task.py index d2fa7428816..cd6246cbca8 100644 --- a/python/databricks/bundles/jobs/_models/power_bi_task.py +++ b/python/databricks/bundles/jobs/_models/power_bi_task.py @@ -1,177 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.jobs._models.power_bi_model import ( + PowerBiModel, + PowerBiModelParam, +) +from databricks.bundles.jobs._models.power_bi_table import ( + PowerBiTable, + PowerBiTableParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PowerBiTask: """""" @@ -202,12 +49,11 @@ class PowerBiTask: """ @classmethod - def from_dict(cls, value: 'PowerBiTaskDict') -> 'Self': + def from_dict(cls, value: "PowerBiTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PowerBiTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PowerBiTaskDict": + return _transform_to_json_value(self) # type:ignore class PowerBiTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/python_operator_task.py b/python/databricks/bundles/jobs/_models/python_operator_task.py index b49a402a573..a2ca2be4c3b 100644 --- a/python/databricks/bundles/jobs/_models/python_operator_task.py +++ b/python/databricks/bundles/jobs/_models/python_operator_task.py @@ -1,182 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.jobs._models.python_operator_task_parameter import ( + PythonOperatorTaskParameter, + PythonOperatorTaskParameterParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PythonOperatorTask: """ :meta private: [EXPERIMENTAL] """ + main: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -185,7 +29,9 @@ class PythonOperatorTask: For example, `my_project.my_function` or `my_project.MyOperator`. """ - parameters: VariableOrList[PythonOperatorTaskParameter] = field(default_factory=list) + parameters: VariableOrList[PythonOperatorTaskParameter] = field( + default_factory=list + ) """ :meta private: [EXPERIMENTAL] @@ -194,12 +40,11 @@ class PythonOperatorTask: """ @classmethod - def from_dict(cls, value: 'PythonOperatorTaskDict') -> 'Self': + def from_dict(cls, value: "PythonOperatorTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PythonOperatorTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PythonOperatorTaskDict": + return _transform_to_json_value(self) # type:ignore class PythonOperatorTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/python_operator_task_parameter.py b/python/databricks/bundles/jobs/_models/python_operator_task_parameter.py index 865c01a0867..867dff0db64 100644 --- a/python/databricks/bundles/jobs/_models/python_operator_task_parameter.py +++ b/python/databricks/bundles/jobs/_models/python_operator_task_parameter.py @@ -1,182 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PythonOperatorTaskParameter: """ :meta private: [EXPERIMENTAL] """ + name: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -192,12 +32,11 @@ class PythonOperatorTaskParameter: """ @classmethod - def from_dict(cls, value: 'PythonOperatorTaskParameterDict') -> 'Self': + def from_dict(cls, value: "PythonOperatorTaskParameterDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PythonOperatorTaskParameterDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PythonOperatorTaskParameterDict": + return _transform_to_json_value(self) # type:ignore class PythonOperatorTaskParameterDict(TypedDict, total=False): @@ -218,4 +57,6 @@ class PythonOperatorTaskParameterDict(TypedDict, total=False): """ -PythonOperatorTaskParameterParam = PythonOperatorTaskParameterDict | PythonOperatorTaskParameter +PythonOperatorTaskParameterParam = ( + PythonOperatorTaskParameterDict | PythonOperatorTaskParameter +) diff --git a/python/databricks/bundles/jobs/_models/python_py_pi_library.py b/python/databricks/bundles/jobs/_models/python_py_pi_library.py index 782396b06b4..037c2029f50 100644 --- a/python/databricks/bundles/jobs/_models/python_py_pi_library.py +++ b/python/databricks/bundles/jobs/_models/python_py_pi_library.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PythonPyPiLibrary: """""" @@ -189,12 +28,11 @@ class PythonPyPiLibrary: """ @classmethod - def from_dict(cls, value: 'PythonPyPiLibraryDict') -> 'Self': + def from_dict(cls, value: "PythonPyPiLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PythonPyPiLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PythonPyPiLibraryDict": + return _transform_to_json_value(self) # type:ignore class PythonPyPiLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/python_wheel_task.py b/python/databricks/bundles/jobs/_models/python_wheel_task.py index 3417ad3e222..ce2dea3ea15 100644 --- a/python/databricks/bundles/jobs/_models/python_wheel_task.py +++ b/python/databricks/bundles/jobs/_models/python_wheel_task.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrDict, VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PythonWheelTask: """""" @@ -197,12 +36,11 @@ class PythonWheelTask: """ @classmethod - def from_dict(cls, value: 'PythonWheelTaskDict') -> 'Self': + def from_dict(cls, value: "PythonWheelTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PythonWheelTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PythonWheelTaskDict": + return _transform_to_json_value(self) # type:ignore class PythonWheelTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/queue_settings.py b/python/databricks/bundles/jobs/_models/queue_settings.py index 2f201600a6e..a72921aed96 100644 --- a/python/databricks/bundles/jobs/_models/queue_settings.py +++ b/python/databricks/bundles/jobs/_models/queue_settings.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class QueueSettings: """""" @@ -182,12 +21,11 @@ class QueueSettings: """ @classmethod - def from_dict(cls, value: 'QueueSettingsDict') -> 'Self': + def from_dict(cls, value: "QueueSettingsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'QueueSettingsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "QueueSettingsDict": + return _transform_to_json_value(self) # type:ignore class QueueSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/r_cran_library.py b/python/databricks/bundles/jobs/_models/r_cran_library.py index e13423f6fbe..0bd2c4d5296 100644 --- a/python/databricks/bundles/jobs/_models/r_cran_library.py +++ b/python/databricks/bundles/jobs/_models/r_cran_library.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RCranLibrary: """""" @@ -187,12 +26,11 @@ class RCranLibrary: """ @classmethod - def from_dict(cls, value: 'RCranLibraryDict') -> 'Self': + def from_dict(cls, value: "RCranLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RCranLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RCranLibraryDict": + return _transform_to_json_value(self) # type:ignore class RCranLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/run_if.py b/python/databricks/bundles/jobs/_models/run_if.py index 4e34c30dd61..fdbf95b1bc2 100644 --- a/python/databricks/bundles/jobs/_models/run_if.py +++ b/python/databricks/bundles/jobs/_models/run_if.py @@ -1,177 +1,13 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class RunIf(Enum): """ An optional value indicating the condition that determines whether the task should be run once its dependencies have been completed. When omitted, defaults to `ALL_SUCCESS`. - + Possible values are: * `ALL_SUCCESS`: All dependencies have executed and succeeded * `AT_LEAST_ONE_SUCCESS`: At least one dependency has succeeded @@ -180,6 +16,7 @@ class RunIf(Enum): * `AT_LEAST_ONE_FAILED`: At least one dependency failed * `ALL_FAILED`: ALl dependencies have failed """ + ALL_SUCCESS = "ALL_SUCCESS" ALL_DONE = "ALL_DONE" NONE_FAILED = "NONE_FAILED" @@ -187,4 +24,15 @@ class RunIf(Enum): ALL_FAILED = "ALL_FAILED" AT_LEAST_ONE_FAILED = "AT_LEAST_ONE_FAILED" -RunIfParam = Literal["ALL_SUCCESS", "ALL_DONE", "NONE_FAILED", "AT_LEAST_ONE_SUCCESS", "ALL_FAILED", "AT_LEAST_ONE_FAILED"] | RunIf + +RunIfParam = ( + Literal[ + "ALL_SUCCESS", + "ALL_DONE", + "NONE_FAILED", + "AT_LEAST_ONE_SUCCESS", + "ALL_FAILED", + "AT_LEAST_ONE_FAILED", + ] + | RunIf +) diff --git a/python/databricks/bundles/jobs/_models/run_job_task.py b/python/databricks/bundles/jobs/_models/run_job_task.py index 78877f5fcce..24d0246a8a7 100644 --- a/python/databricks/bundles/jobs/_models/run_job_task.py +++ b/python/databricks/bundles/jobs/_models/run_job_task.py @@ -1,177 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrOptional, +) +from databricks.bundles.jobs._models.pipeline_params import ( + PipelineParams, + PipelineParamsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RunJobTask: """""" @@ -192,12 +39,11 @@ class RunJobTask: """ @classmethod - def from_dict(cls, value: 'RunJobTaskDict') -> 'Self': + def from_dict(cls, value: "RunJobTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RunJobTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RunJobTaskDict": + return _transform_to_json_value(self) # type:ignore class RunJobTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/runtime_engine.py b/python/databricks/bundles/jobs/_models/runtime_engine.py index 5138fc821cc..2e1559fa801 100644 --- a/python/databricks/bundles/jobs/_models/runtime_engine.py +++ b/python/databricks/bundles/jobs/_models/runtime_engine.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class RuntimeEngine(Enum): @@ -173,4 +9,5 @@ class RuntimeEngine(Enum): STANDARD = "STANDARD" PHOTON = "PHOTON" + RuntimeEngineParam = Literal["NULL", "STANDARD", "PHOTON"] | RuntimeEngine diff --git a/python/databricks/bundles/jobs/_models/s3_storage_info.py b/python/databricks/bundles/jobs/_models/s3_storage_info.py index 4818f09c213..071abd5b6f3 100644 --- a/python/databricks/bundles/jobs/_models/s3_storage_info.py +++ b/python/databricks/bundles/jobs/_models/s3_storage_info.py @@ -1,182 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class S3StorageInfo: """ A storage location in Amazon S3 """ + destination: VariableOr[str] """ S3 destination, e.g. `s3://my-bucket/some-prefix` Note that logs will be delivered using @@ -224,12 +64,11 @@ class S3StorageInfo: """ @classmethod - def from_dict(cls, value: 'S3StorageInfoDict') -> 'Self': + def from_dict(cls, value: "S3StorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'S3StorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "S3StorageInfoDict": + return _transform_to_json_value(self) # type:ignore class S3StorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/source.py b/python/databricks/bundles/jobs/_models/source.py index 4393bf79e24..380ab711022 100644 --- a/python/databricks/bundles/jobs/_models/source.py +++ b/python/databricks/bundles/jobs/_models/source.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class Source(Enum): @@ -177,7 +13,9 @@ class Source(Enum): * `WORKSPACE`: SQL file is located in Databricks workspace. * `GIT`: SQL file is located in cloud Git provider. """ + WORKSPACE = "WORKSPACE" GIT = "GIT" + SourceParam = Literal["WORKSPACE", "GIT"] | Source diff --git a/python/databricks/bundles/jobs/_models/spark_jar_task.py b/python/databricks/bundles/jobs/_models/spark_jar_task.py index 8522ffed962..d782da867f3 100644 --- a/python/databricks/bundles/jobs/_models/spark_jar_task.py +++ b/python/databricks/bundles/jobs/_models/spark_jar_task.py @@ -1,177 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SparkJarTask: """""" @@ -203,12 +46,11 @@ class SparkJarTask: """ @classmethod - def from_dict(cls, value: 'SparkJarTaskDict') -> 'Self': + def from_dict(cls, value: "SparkJarTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SparkJarTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SparkJarTaskDict": + return _transform_to_json_value(self) # type:ignore class SparkJarTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/spark_python_task.py b/python/databricks/bundles/jobs/_models/spark_python_task.py index efa1c04d656..001d9287b1c 100644 --- a/python/databricks/bundles/jobs/_models/spark_python_task.py +++ b/python/databricks/bundles/jobs/_models/spark_python_task.py @@ -1,177 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) from databricks.bundles.jobs._models.source import Source, SourceParam -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SparkPythonTask: """""" @@ -199,12 +43,11 @@ class SparkPythonTask: """ @classmethod - def from_dict(cls, value: 'SparkPythonTaskDict') -> 'Self': + def from_dict(cls, value: "SparkPythonTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SparkPythonTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SparkPythonTaskDict": + return _transform_to_json_value(self) # type:ignore class SparkPythonTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/spark_submit_task.py b/python/databricks/bundles/jobs/_models/spark_submit_task.py index 358a7248f7d..2acc389b91b 100644 --- a/python/databricks/bundles/jobs/_models/spark_submit_task.py +++ b/python/databricks/bundles/jobs/_models/spark_submit_task.py @@ -1,182 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SparkSubmitTask: """ - [DEPRECATED] + [DEPRECATED] """ + parameters: VariableOrList[str] = field(default_factory=list) """ Command-line parameters passed to spark submit. @@ -185,12 +25,11 @@ class SparkSubmitTask: """ @classmethod - def from_dict(cls, value: 'SparkSubmitTaskDict') -> 'Self': + def from_dict(cls, value: "SparkSubmitTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SparkSubmitTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SparkSubmitTaskDict": + return _transform_to_json_value(self) # type:ignore class SparkSubmitTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sparse_checkout.py b/python/databricks/bundles/jobs/_models/sparse_checkout.py index 232518a268c..a42a78e49f1 100644 --- a/python/databricks/bundles/jobs/_models/sparse_checkout.py +++ b/python/databricks/bundles/jobs/_models/sparse_checkout.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SparseCheckout: """""" @@ -182,12 +21,11 @@ class SparseCheckout: """ @classmethod - def from_dict(cls, value: 'SparseCheckoutDict') -> 'Self': + def from_dict(cls, value: "SparseCheckoutDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SparseCheckoutDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SparseCheckoutDict": + return _transform_to_json_value(self) # type:ignore class SparseCheckoutDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_condition_configuration.py b/python/databricks/bundles/jobs/_models/sql_condition_configuration.py index 028bb176dbe..f6b2da80685 100644 --- a/python/databricks/bundles/jobs/_models/sql_condition_configuration.py +++ b/python/databricks/bundles/jobs/_models/sql_condition_configuration.py @@ -1,182 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.jobs._models.sql_condition_trigger_mode import ( + SqlConditionTriggerMode, + SqlConditionTriggerModeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SqlConditionConfiguration: """ :meta private: [EXPERIMENTAL] """ + sql_query_id: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -203,12 +47,11 @@ class SqlConditionConfiguration: """ @classmethod - def from_dict(cls, value: 'SqlConditionConfigurationDict') -> 'Self': + def from_dict(cls, value: "SqlConditionConfigurationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SqlConditionConfigurationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SqlConditionConfigurationDict": + return _transform_to_json_value(self) # type:ignore class SqlConditionConfigurationDict(TypedDict, total=False): @@ -240,4 +83,6 @@ class SqlConditionConfigurationDict(TypedDict, total=False): """ -SqlConditionConfigurationParam = SqlConditionConfigurationDict | SqlConditionConfiguration +SqlConditionConfigurationParam = ( + SqlConditionConfigurationDict | SqlConditionConfiguration +) diff --git a/python/databricks/bundles/jobs/_models/sql_condition_trigger_mode.py b/python/databricks/bundles/jobs/_models/sql_condition_trigger_mode.py index 5548ee01618..c2b8472faea 100644 --- a/python/databricks/bundles/jobs/_models/sql_condition_trigger_mode.py +++ b/python/databricks/bundles/jobs/_models/sql_condition_trigger_mode.py @@ -1,179 +1,15 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class SqlConditionTriggerMode(Enum): """ :meta private: [EXPERIMENTAL] - + The strategy used to evaluate a SQL condition trigger against a query result set. - + * `SQL_CONDITION_TRIGGER_MODE_UNSPECIFIED`: Sentinel zero-value. Not a valid input — the validator rejects this when sent explicitly. Internally treated as `QUERY_RETURNS_ROWS` when reading legacy data that predates this field. @@ -183,7 +19,11 @@ class SqlConditionTriggerMode(Enum): previous evaluation. The first evaluation always fires. Queries must return exactly one cell (one row, one column). """ + QUERY_RETURNS_ROWS = "QUERY_RETURNS_ROWS" RESULT_VALUE_CHANGES = "RESULT_VALUE_CHANGES" -SqlConditionTriggerModeParam = Literal["QUERY_RETURNS_ROWS", "RESULT_VALUE_CHANGES"] | SqlConditionTriggerMode + +SqlConditionTriggerModeParam = ( + Literal["QUERY_RETURNS_ROWS", "RESULT_VALUE_CHANGES"] | SqlConditionTriggerMode +) diff --git a/python/databricks/bundles/jobs/_models/sql_task.py b/python/databricks/bundles/jobs/_models/sql_task.py index ea97314e1d7..dd731928af0 100644 --- a/python/databricks/bundles/jobs/_models/sql_task.py +++ b/python/databricks/bundles/jobs/_models/sql_task.py @@ -1,177 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrOptional, +) +from databricks.bundles.jobs._models.sql_task_alert import ( + SqlTaskAlert, + SqlTaskAlertParam, +) +from databricks.bundles.jobs._models.sql_task_dashboard import ( + SqlTaskDashboard, + SqlTaskDashboardParam, +) +from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileParam +from databricks.bundles.jobs._models.sql_task_query import ( + SqlTaskQuery, + SqlTaskQueryParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SqlTask: """""" @@ -207,12 +63,11 @@ class SqlTask: """ @classmethod - def from_dict(cls, value: 'SqlTaskDict') -> 'Self': + def from_dict(cls, value: "SqlTaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SqlTaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SqlTaskDict": + return _transform_to_json_value(self) # type:ignore class SqlTaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_alert.py b/python/databricks/bundles/jobs/_models/sql_task_alert.py index 0fa6e6141e7..6db18d1c876 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_alert.py +++ b/python/databricks/bundles/jobs/_models/sql_task_alert.py @@ -1,177 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.jobs._models.sql_task_subscription import ( + SqlTaskSubscription, + SqlTaskSubscriptionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SqlTaskAlert: """""" @@ -192,12 +39,11 @@ class SqlTaskAlert: """ @classmethod - def from_dict(cls, value: 'SqlTaskAlertDict') -> 'Self': + def from_dict(cls, value: "SqlTaskAlertDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SqlTaskAlertDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SqlTaskAlertDict": + return _transform_to_json_value(self) # type:ignore class SqlTaskAlertDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_dashboard.py b/python/databricks/bundles/jobs/_models/sql_task_dashboard.py index 5f4970e2a37..98e544e5aa0 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_dashboard.py +++ b/python/databricks/bundles/jobs/_models/sql_task_dashboard.py @@ -1,177 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.jobs._models.sql_task_subscription import ( + SqlTaskSubscription, + SqlTaskSubscriptionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SqlTaskDashboard: """""" @@ -197,12 +44,11 @@ class SqlTaskDashboard: """ @classmethod - def from_dict(cls, value: 'SqlTaskDashboardDict') -> 'Self': + def from_dict(cls, value: "SqlTaskDashboardDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SqlTaskDashboardDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SqlTaskDashboardDict": + return _transform_to_json_value(self) # type:ignore class SqlTaskDashboardDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_file.py b/python/databricks/bundles/jobs/_models/sql_task_file.py index b9be171537f..8a49f54c7b7 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_file.py +++ b/python/databricks/bundles/jobs/_models/sql_task_file.py @@ -1,177 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional from databricks.bundles.jobs._models.source import Source, SourceParam -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SqlTaskFile: """""" @@ -192,12 +32,11 @@ class SqlTaskFile: """ @classmethod - def from_dict(cls, value: 'SqlTaskFileDict') -> 'Self': + def from_dict(cls, value: "SqlTaskFileDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SqlTaskFileDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SqlTaskFileDict": + return _transform_to_json_value(self) # type:ignore class SqlTaskFileDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_query.py b/python/databricks/bundles/jobs/_models/sql_task_query.py index 4ef7aed0ea0..2682c425112 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_query.py +++ b/python/databricks/bundles/jobs/_models/sql_task_query.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SqlTaskQuery: """""" @@ -182,12 +21,11 @@ class SqlTaskQuery: """ @classmethod - def from_dict(cls, value: 'SqlTaskQueryDict') -> 'Self': + def from_dict(cls, value: "SqlTaskQueryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SqlTaskQueryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SqlTaskQueryDict": + return _transform_to_json_value(self) # type:ignore class SqlTaskQueryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/sql_task_subscription.py b/python/databricks/bundles/jobs/_models/sql_task_subscription.py index 340bbceff70..d5f2a65ddf5 100644 --- a/python/databricks/bundles/jobs/_models/sql_task_subscription.py +++ b/python/databricks/bundles/jobs/_models/sql_task_subscription.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SqlTaskSubscription: """""" @@ -187,12 +26,11 @@ class SqlTaskSubscription: """ @classmethod - def from_dict(cls, value: 'SqlTaskSubscriptionDict') -> 'Self': + def from_dict(cls, value: "SqlTaskSubscriptionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SqlTaskSubscriptionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SqlTaskSubscriptionDict": + return _transform_to_json_value(self) # type:ignore class SqlTaskSubscriptionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/storage_mode.py b/python/databricks/bundles/jobs/_models/storage_mode.py index c084b16175b..f15a94689d6 100644 --- a/python/databricks/bundles/jobs/_models/storage_mode.py +++ b/python/databricks/bundles/jobs/_models/storage_mode.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class StorageMode(Enum): @@ -173,4 +9,5 @@ class StorageMode(Enum): IMPORT = "IMPORT" DUAL = "DUAL" + StorageModeParam = Literal["DIRECT_QUERY", "IMPORT", "DUAL"] | StorageMode diff --git a/python/databricks/bundles/jobs/_models/subscription.py b/python/databricks/bundles/jobs/_models/subscription.py index eceeafecebd..55de6336d7b 100644 --- a/python/databricks/bundles/jobs/_models/subscription.py +++ b/python/databricks/bundles/jobs/_models/subscription.py @@ -1,177 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.jobs._models.subscription_attachment import ( + SubscriptionAttachment, + SubscriptionAttachmentParam, +) +from databricks.bundles.jobs._models.subscription_subscriber import ( + SubscriptionSubscriber, + SubscriptionSubscriberParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Subscription: """""" @@ -203,12 +50,11 @@ class Subscription: """ @classmethod - def from_dict(cls, value: 'SubscriptionDict') -> 'Self': + def from_dict(cls, value: "SubscriptionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SubscriptionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SubscriptionDict": + return _transform_to_json_value(self) # type:ignore class SubscriptionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment.py b/python/databricks/bundles/jobs/_models/subscription_attachment.py index f5e2c3b5252..0bc23c65229 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment.py @@ -1,177 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.subscription_attachment_data_attachment import ( + SubscriptionAttachmentDataAttachment, + SubscriptionAttachmentDataAttachmentParam, +) +from databricks.bundles.jobs._models.subscription_attachment_page_attachment import ( + SubscriptionAttachmentPageAttachment, + SubscriptionAttachmentPageAttachmentParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SubscriptionAttachment: """ @@ -180,6 +27,7 @@ class SubscriptionAttachment: as data files, and `page_attachment` describes which dashboard pages are included in the attached PDF. """ + data_attachment: VariableOrOptional[SubscriptionAttachmentDataAttachment] = None """ :meta private: [EXPERIMENTAL] @@ -197,12 +45,11 @@ class SubscriptionAttachment: """ @classmethod - def from_dict(cls, value: 'SubscriptionAttachmentDict') -> 'Self': + def from_dict(cls, value: "SubscriptionAttachmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SubscriptionAttachmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SubscriptionAttachmentDict": + return _transform_to_json_value(self) # type:ignore class SubscriptionAttachmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment.py b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment.py index 6e7f899176c..d998d07d7b6 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment.py @@ -1,177 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import ( + SubscriptionAttachmentDataAttachmentFileFormat, + SubscriptionAttachmentDataAttachmentFileFormatParam, +) +from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import ( + SubscriptionAttachmentDataAttachmentWidget, + SubscriptionAttachmentDataAttachmentWidgetParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SubscriptionAttachmentDataAttachment: """ @@ -179,6 +30,7 @@ class SubscriptionAttachmentDataAttachment: lists the widgets whose data is exported and `file_format` is the output format; each referenced widget's data is exported as a separate file. """ + file_format: VariableOr[SubscriptionAttachmentDataAttachmentFileFormat] """ :meta private: [EXPERIMENTAL] @@ -194,7 +46,9 @@ class SubscriptionAttachmentDataAttachment: an additional attached file. """ - widgets: VariableOrList[SubscriptionAttachmentDataAttachmentWidget] = field(default_factory=list) + widgets: VariableOrList[SubscriptionAttachmentDataAttachmentWidget] = field( + default_factory=list + ) """ :meta private: [EXPERIMENTAL] @@ -202,12 +56,11 @@ class SubscriptionAttachmentDataAttachment: """ @classmethod - def from_dict(cls, value: 'SubscriptionAttachmentDataAttachmentDict') -> 'Self': + def from_dict(cls, value: "SubscriptionAttachmentDataAttachmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SubscriptionAttachmentDataAttachmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SubscriptionAttachmentDataAttachmentDict": + return _transform_to_json_value(self) # type:ignore class SubscriptionAttachmentDataAttachmentDict(TypedDict, total=False): @@ -236,4 +89,6 @@ class SubscriptionAttachmentDataAttachmentDict(TypedDict, total=False): """ -SubscriptionAttachmentDataAttachmentParam = SubscriptionAttachmentDataAttachmentDict | SubscriptionAttachmentDataAttachment +SubscriptionAttachmentDataAttachmentParam = ( + SubscriptionAttachmentDataAttachmentDict | SubscriptionAttachmentDataAttachment +) diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_file_format.py b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_file_format.py index 7d618540f87..9c9e7a2c632 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_file_format.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_file_format.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class SubscriptionAttachmentDataAttachmentFileFormat(Enum): @@ -173,8 +9,13 @@ class SubscriptionAttachmentDataAttachmentFileFormat(Enum): The output format of an exported tabular data file. Must be set to a value other than `FILE_FORMAT_UNSPECIFIED`. """ + FILE_FORMAT_CSV = "FILE_FORMAT_CSV" FILE_FORMAT_TSV = "FILE_FORMAT_TSV" FILE_FORMAT_EXCEL = "FILE_FORMAT_EXCEL" -SubscriptionAttachmentDataAttachmentFileFormatParam = Literal["FILE_FORMAT_CSV", "FILE_FORMAT_TSV", "FILE_FORMAT_EXCEL"] | SubscriptionAttachmentDataAttachmentFileFormat + +SubscriptionAttachmentDataAttachmentFileFormatParam = ( + Literal["FILE_FORMAT_CSV", "FILE_FORMAT_TSV", "FILE_FORMAT_EXCEL"] + | SubscriptionAttachmentDataAttachmentFileFormat +) diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_widget.py b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_widget.py index ce56890dd9d..be802c0e07e 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_widget.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_data_attachment_widget.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SubscriptionAttachmentDataAttachmentWidget: """ @@ -179,6 +18,7 @@ class SubscriptionAttachmentDataAttachmentWidget: the URL identifiers from the dashboard's serialized definition returned by the Get dashboard API (`serialized_dashboard`), and together they uniquely identify the widget. """ + page_id: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -198,12 +38,13 @@ class SubscriptionAttachmentDataAttachmentWidget: """ @classmethod - def from_dict(cls, value: 'SubscriptionAttachmentDataAttachmentWidgetDict') -> 'Self': + def from_dict( + cls, value: "SubscriptionAttachmentDataAttachmentWidgetDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SubscriptionAttachmentDataAttachmentWidgetDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SubscriptionAttachmentDataAttachmentWidgetDict": + return _transform_to_json_value(self) # type:ignore class SubscriptionAttachmentDataAttachmentWidgetDict(TypedDict, total=False): @@ -228,4 +69,7 @@ class SubscriptionAttachmentDataAttachmentWidgetDict(TypedDict, total=False): """ -SubscriptionAttachmentDataAttachmentWidgetParam = SubscriptionAttachmentDataAttachmentWidgetDict | SubscriptionAttachmentDataAttachmentWidget +SubscriptionAttachmentDataAttachmentWidgetParam = ( + SubscriptionAttachmentDataAttachmentWidgetDict + | SubscriptionAttachmentDataAttachmentWidget +) diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment.py b/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment.py index 7860b18867a..087c872fad4 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment.py @@ -1,183 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import ( + SubscriptionAttachmentPageAttachmentPageSelection, + SubscriptionAttachmentPageAttachmentPageSelectionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SubscriptionAttachmentPageAttachment: """ The selection of dashboard pages included in the attached PDF. `page_selection` chooses between all pages, the first page only, or an explicit subset listed in `page_ids`. """ + page_ids: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -189,7 +33,9 @@ class SubscriptionAttachmentPageAttachment: Format: alphanumeric, hyphen, and underscore characters allowed, max length 63. """ - page_selection: VariableOrOptional[SubscriptionAttachmentPageAttachmentPageSelection] = None + page_selection: VariableOrOptional[ + SubscriptionAttachmentPageAttachmentPageSelection + ] = None """ :meta private: [EXPERIMENTAL] @@ -198,12 +44,11 @@ class SubscriptionAttachmentPageAttachment: """ @classmethod - def from_dict(cls, value: 'SubscriptionAttachmentPageAttachmentDict') -> 'Self': + def from_dict(cls, value: "SubscriptionAttachmentPageAttachmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SubscriptionAttachmentPageAttachmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SubscriptionAttachmentPageAttachmentDict": + return _transform_to_json_value(self) # type:ignore class SubscriptionAttachmentPageAttachmentDict(TypedDict, total=False): @@ -220,7 +65,9 @@ class SubscriptionAttachmentPageAttachmentDict(TypedDict, total=False): Format: alphanumeric, hyphen, and underscore characters allowed, max length 63. """ - page_selection: VariableOrOptional[SubscriptionAttachmentPageAttachmentPageSelectionParam] + page_selection: VariableOrOptional[ + SubscriptionAttachmentPageAttachmentPageSelectionParam + ] """ :meta private: [EXPERIMENTAL] @@ -229,4 +76,6 @@ class SubscriptionAttachmentPageAttachmentDict(TypedDict, total=False): """ -SubscriptionAttachmentPageAttachmentParam = SubscriptionAttachmentPageAttachmentDict | SubscriptionAttachmentPageAttachment +SubscriptionAttachmentPageAttachmentParam = ( + SubscriptionAttachmentPageAttachmentDict | SubscriptionAttachmentPageAttachment +) diff --git a/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment_page_selection.py b/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment_page_selection.py index bbbf72ae20b..b24c75c4695 100644 --- a/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment_page_selection.py +++ b/python/databricks/bundles/jobs/_models/subscription_attachment_page_attachment_page_selection.py @@ -1,178 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam +from typing import Literal class SubscriptionAttachmentPageAttachmentPageSelection(Enum): """ Determines which dashboard pages are attached to the notification. """ + PAGE_SELECTION_ALL = "PAGE_SELECTION_ALL" PAGE_SELECTION_SUBSET = "PAGE_SELECTION_SUBSET" -SubscriptionAttachmentPageAttachmentPageSelectionParam = Literal["PAGE_SELECTION_ALL", "PAGE_SELECTION_SUBSET"] | SubscriptionAttachmentPageAttachmentPageSelection + +SubscriptionAttachmentPageAttachmentPageSelectionParam = ( + Literal["PAGE_SELECTION_ALL", "PAGE_SELECTION_SUBSET"] + | SubscriptionAttachmentPageAttachmentPageSelection +) diff --git a/python/databricks/bundles/jobs/_models/subscription_subscriber.py b/python/databricks/bundles/jobs/_models/subscription_subscriber.py index 0e62012edd4..d03875f1ab2 100644 --- a/python/databricks/bundles/jobs/_models/subscription_subscriber.py +++ b/python/databricks/bundles/jobs/_models/subscription_subscriber.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SubscriptionSubscriber: """""" @@ -187,12 +26,11 @@ class SubscriptionSubscriber: """ @classmethod - def from_dict(cls, value: 'SubscriptionSubscriberDict') -> 'Self': + def from_dict(cls, value: "SubscriptionSubscriberDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SubscriptionSubscriberDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SubscriptionSubscriberDict": + return _transform_to_json_value(self) # type:ignore class SubscriptionSubscriberDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/table_update_trigger_configuration.py b/python/databricks/bundles/jobs/_models/table_update_trigger_configuration.py index ba99e1b5c9a..24cf1301154 100644 --- a/python/databricks/bundles/jobs/_models/table_update_trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/table_update_trigger_configuration.py @@ -1,177 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional from databricks.bundles.jobs._models.condition import Condition, ConditionParam -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TableUpdateTriggerConfiguration: """""" @@ -200,12 +40,11 @@ class TableUpdateTriggerConfiguration: """ @classmethod - def from_dict(cls, value: 'TableUpdateTriggerConfigurationDict') -> 'Self': + def from_dict(cls, value: "TableUpdateTriggerConfigurationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TableUpdateTriggerConfigurationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TableUpdateTriggerConfigurationDict": + return _transform_to_json_value(self) # type:ignore class TableUpdateTriggerConfigurationDict(TypedDict, total=False): @@ -235,4 +74,6 @@ class TableUpdateTriggerConfigurationDict(TypedDict, total=False): """ -TableUpdateTriggerConfigurationParam = TableUpdateTriggerConfigurationDict | TableUpdateTriggerConfiguration +TableUpdateTriggerConfigurationParam = ( + TableUpdateTriggerConfigurationDict | TableUpdateTriggerConfiguration +) diff --git a/python/databricks/bundles/jobs/_models/task.py b/python/databricks/bundles/jobs/_models/task.py index c1529538a1e..c459192ba2b 100644 --- a/python/databricks/bundles/jobs/_models/task.py +++ b/python/databricks/bundles/jobs/_models/task.py @@ -1,177 +1,105 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.jobs._models.ai_runtime_task import ( + AiRuntimeTask, + AiRuntimeTaskParam, +) +from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskParam +from databricks.bundles.jobs._models.clean_rooms_notebook_task import ( + CleanRoomsNotebookTask, + CleanRoomsNotebookTaskParam, +) +from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecParam +from databricks.bundles.jobs._models.compute import Compute, ComputeParam +from databricks.bundles.jobs._models.condition_task import ( + ConditionTask, + ConditionTaskParam, +) +from databricks.bundles.jobs._models.dashboard_task import ( + DashboardTask, + DashboardTaskParam, +) +from databricks.bundles.jobs._models.dbt_platform_task import ( + DbtPlatformTask, + DbtPlatformTaskParam, +) +from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskParam +from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskParam +from databricks.bundles.jobs._models.gen_ai_compute_task import ( + GenAiComputeTask, + GenAiComputeTaskParam, +) +from databricks.bundles.jobs._models.jobs_health_rules import ( + JobsHealthRules, + JobsHealthRulesParam, +) +from databricks.bundles.jobs._models.library import Library, LibraryParam +from databricks.bundles.jobs._models.notebook_task import ( + NotebookTask, + NotebookTaskParam, +) +from databricks.bundles.jobs._models.pipeline_task import ( + PipelineTask, + PipelineTaskParam, +) +from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskParam +from databricks.bundles.jobs._models.python_operator_task import ( + PythonOperatorTask, + PythonOperatorTaskParam, +) +from databricks.bundles.jobs._models.python_wheel_task import ( + PythonWheelTask, + PythonWheelTaskParam, +) from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.jobs._models.run_job_task import ( + RunJobTask, + RunJobTaskParam, +) +from databricks.bundles.jobs._models.spark_jar_task import ( + SparkJarTask, + SparkJarTaskParam, +) +from databricks.bundles.jobs._models.spark_python_task import ( + SparkPythonTask, + SparkPythonTaskParam, +) +from databricks.bundles.jobs._models.spark_submit_task import ( + SparkSubmitTask, + SparkSubmitTaskParam, +) +from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskParam +from databricks.bundles.jobs._models.task_dependency import ( + TaskDependency, + TaskDependencyParam, +) +from databricks.bundles.jobs._models.task_email_notifications import ( + TaskEmailNotifications, + TaskEmailNotificationsParam, +) +from databricks.bundles.jobs._models.task_notification_settings import ( + TaskNotificationSettings, + TaskNotificationSettingsParam, +) +from databricks.bundles.jobs._models.webhook_notifications import ( + WebhookNotifications, + WebhookNotificationsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Task: """""" @@ -413,12 +341,11 @@ class Task: """ @classmethod - def from_dict(cls, value: 'TaskDict') -> 'Self': + def from_dict(cls, value: "TaskDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TaskDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TaskDict": + return _transform_to_json_value(self) # type:ignore class TaskDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/task_dependency.py b/python/databricks/bundles/jobs/_models/task_dependency.py index dcd597778e1..a1e18f0bb59 100644 --- a/python/databricks/bundles/jobs/_models/task_dependency.py +++ b/python/databricks/bundles/jobs/_models/task_dependency.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TaskDependency: """""" @@ -187,12 +26,11 @@ class TaskDependency: """ @classmethod - def from_dict(cls, value: 'TaskDependencyDict') -> 'Self': + def from_dict(cls, value: "TaskDependencyDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TaskDependencyDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TaskDependencyDict": + return _transform_to_json_value(self) # type:ignore class TaskDependencyDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/task_email_notifications.py b/python/databricks/bundles/jobs/_models/task_email_notifications.py index 4ab3ccc97d0..cdf0dee5fdf 100644 --- a/python/databricks/bundles/jobs/_models/task_email_notifications.py +++ b/python/databricks/bundles/jobs/_models/task_email_notifications.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TaskEmailNotifications: """""" @@ -182,7 +21,9 @@ class TaskEmailNotifications: This field is `deprecated`. Please use the `notification_settings.no_alert_for_skipped_runs` field. """ - on_duration_warning_threshold_exceeded: VariableOrList[str] = field(default_factory=list) + on_duration_warning_threshold_exceeded: VariableOrList[str] = field( + default_factory=list + ) """ A list of email addresses to be notified when the duration of a run exceeds the threshold specified for the `RUN_DURATION_SECONDS` metric in the `health` field. If no rule for the `RUN_DURATION_SECONDS` metric is specified in the `health` field for the job, notifications are not sent. """ @@ -220,12 +61,11 @@ class TaskEmailNotifications: """ @classmethod - def from_dict(cls, value: 'TaskEmailNotificationsDict') -> 'Self': + def from_dict(cls, value: "TaskEmailNotificationsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TaskEmailNotificationsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TaskEmailNotificationsDict": + return _transform_to_json_value(self) # type:ignore class TaskEmailNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/task_notification_settings.py b/python/databricks/bundles/jobs/_models/task_notification_settings.py index 274dd11acff..346f4463b9a 100644 --- a/python/databricks/bundles/jobs/_models/task_notification_settings.py +++ b/python/databricks/bundles/jobs/_models/task_notification_settings.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TaskNotificationSettings: """""" @@ -192,12 +31,11 @@ class TaskNotificationSettings: """ @classmethod - def from_dict(cls, value: 'TaskNotificationSettingsDict') -> 'Self': + def from_dict(cls, value: "TaskNotificationSettingsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TaskNotificationSettingsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TaskNotificationSettingsDict": + return _transform_to_json_value(self) # type:ignore class TaskNotificationSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/task_retry_mode.py b/python/databricks/bundles/jobs/_models/task_retry_mode.py index 20a8d9fdd81..f0642b54f2c 100644 --- a/python/databricks/bundles/jobs/_models/task_retry_mode.py +++ b/python/databricks/bundles/jobs/_models/task_retry_mode.py @@ -1,171 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam +from typing import Literal class TaskRetryMode(Enum): @@ -175,7 +11,9 @@ class TaskRetryMode(Enum): * ON_FAILURE: Retry a failed task if at least one other task in the job is still running its first attempt. When this condition is no longer met or the retry limit is reached, the job run is cancelled and a new run is started. """ + NEVER = "NEVER" ON_FAILURE = "ON_FAILURE" + TaskRetryModeParam = Literal["NEVER", "ON_FAILURE"] | TaskRetryMode diff --git a/python/databricks/bundles/jobs/_models/trigger_configuration.py b/python/databricks/bundles/jobs/_models/trigger_configuration.py index bbf33d9cdf3..b9663effddc 100644 --- a/python/databricks/bundles/jobs/_models/trigger_configuration.py +++ b/python/databricks/bundles/jobs/_models/trigger_configuration.py @@ -1,177 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.continuous_trigger_configuration import ( + ContinuousTriggerConfiguration, + ContinuousTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.cron_trigger_configuration import ( + CronTriggerConfiguration, + CronTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import ( + FileArrivalTriggerConfiguration, + FileArrivalTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.model_trigger_configuration import ( + ModelTriggerConfiguration, + ModelTriggerConfigurationParam, +) from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.jobs._models.periodic_trigger_configuration import ( + PeriodicTriggerConfiguration, + PeriodicTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.sql_condition_configuration import ( + SqlConditionConfiguration, + SqlConditionConfigurationParam, +) +from databricks.bundles.jobs._models.table_update_trigger_configuration import ( + TableUpdateTriggerConfiguration, + TableUpdateTriggerConfigurationParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TriggerConfiguration: """ @@ -180,6 +48,7 @@ class TriggerConfiguration: mutual exclusivity is enforced in the API handler rather than via `oneof` so that codegen, validation, and JSON serialization across SDKs and Terraform behave consistently. """ + continuous: VariableOrOptional[ContinuousTriggerConfiguration] = None """ :meta private: [EXPERIMENTAL] @@ -238,12 +107,11 @@ class TriggerConfiguration: """ @classmethod - def from_dict(cls, value: 'TriggerConfigurationDict') -> 'Self': + def from_dict(cls, value: "TriggerConfigurationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TriggerConfigurationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TriggerConfigurationDict": + return _transform_to_json_value(self) # type:ignore class TriggerConfigurationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/trigger_settings.py b/python/databricks/bundles/jobs/_models/trigger_settings.py index c2c968fca6c..878f45732fa 100644 --- a/python/databricks/bundles/jobs/_models/trigger_settings.py +++ b/python/databricks/bundles/jobs/_models/trigger_settings.py @@ -1,177 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.jobs._models.file_arrival_trigger_configuration import ( + FileArrivalTriggerConfiguration, + FileArrivalTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.model_trigger_configuration import ( + ModelTriggerConfiguration, + ModelTriggerConfigurationParam, +) from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.jobs._models.periodic_trigger_configuration import ( + PeriodicTriggerConfiguration, + PeriodicTriggerConfigurationParam, +) +from databricks.bundles.jobs._models.sql_condition_configuration import ( + SqlConditionConfiguration, + SqlConditionConfigurationParam, +) +from databricks.bundles.jobs._models.table_update_trigger_configuration import ( + TableUpdateTriggerConfiguration, + TableUpdateTriggerConfigurationParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TriggerSettings: """""" @@ -209,12 +69,11 @@ class TriggerSettings: table_update: VariableOrOptional[TableUpdateTriggerConfiguration] = None @classmethod - def from_dict(cls, value: 'TriggerSettingsDict') -> 'Self': + def from_dict(cls, value: "TriggerSettingsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TriggerSettingsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TriggerSettingsDict": + return _transform_to_json_value(self) # type:ignore class TriggerSettingsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/volumes_storage_info.py b/python/databricks/bundles/jobs/_models/volumes_storage_info.py index da6897b5c29..1eb4fff7bc2 100644 --- a/python/databricks/bundles/jobs/_models/volumes_storage_info.py +++ b/python/databricks/bundles/jobs/_models/volumes_storage_info.py @@ -1,182 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class VolumesStorageInfo: """ A storage location back by UC Volumes. """ + destination: VariableOr[str] """ UC Volumes destination, e.g. `/Volumes/catalog/schema/vol1/init-scripts/setup-datadog.sh` @@ -184,12 +24,11 @@ class VolumesStorageInfo: """ @classmethod - def from_dict(cls, value: 'VolumesStorageInfoDict') -> 'Self': + def from_dict(cls, value: "VolumesStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'VolumesStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "VolumesStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class VolumesStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/webhook.py b/python/databricks/bundles/jobs/_models/webhook.py index 5fdaf88e2b4..17a391ac2c2 100644 --- a/python/databricks/bundles/jobs/_models/webhook.py +++ b/python/databricks/bundles/jobs/_models/webhook.py @@ -1,177 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Webhook: """""" @@ -179,12 +18,11 @@ class Webhook: id: VariableOr[str] @classmethod - def from_dict(cls, value: 'WebhookDict') -> 'Self': + def from_dict(cls, value: "WebhookDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'WebhookDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "WebhookDict": + return _transform_to_json_value(self) # type:ignore class WebhookDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/webhook_notifications.py b/python/databricks/bundles/jobs/_models/webhook_notifications.py index edf13270be2..0eed809f68b 100644 --- a/python/databricks/bundles/jobs/_models/webhook_notifications.py +++ b/python/databricks/bundles/jobs/_models/webhook_notifications.py @@ -1,182 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.jobs._models.webhook import Webhook, WebhookParam if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class WebhookNotifications: """""" - on_duration_warning_threshold_exceeded: VariableOrList[Webhook] = field(default_factory=list) + on_duration_warning_threshold_exceeded: VariableOrList[Webhook] = field( + default_factory=list + ) """ An optional list of system notification IDs to call when the duration of a run exceeds the threshold specified for the `RUN_DURATION_SECONDS` metric in the `health` field. A maximum of 3 destinations can be specified for the `on_duration_warning_threshold_exceeded` property. """ @@ -217,12 +59,11 @@ class WebhookNotifications: """ @classmethod - def from_dict(cls, value: 'WebhookNotificationsDict') -> 'Self': + def from_dict(cls, value: "WebhookNotificationsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'WebhookNotificationsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "WebhookNotificationsDict": + return _transform_to_json_value(self) # type:ignore class WebhookNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/workload_type.py b/python/databricks/bundles/jobs/_models/workload_type.py index 9d21b749537..033590c3949 100644 --- a/python/databricks/bundles/jobs/_models/workload_type.py +++ b/python/databricks/bundles/jobs/_models/workload_type.py @@ -1,194 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr +from databricks.bundles.jobs._models.clients_types import ( + ClientsTypes, + ClientsTypesParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class WorkloadType: """ Cluster Attributes showing for clusters workload types. """ + clients: VariableOr[ClientsTypes] """ defined what type of clients can use the cluster. E.g. Notebooks, Jobs """ @classmethod - def from_dict(cls, value: 'WorkloadTypeDict') -> 'Self': + def from_dict(cls, value: "WorkloadTypeDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'WorkloadTypeDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "WorkloadTypeDict": + return _transform_to_json_value(self) # type:ignore class WorkloadTypeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/jobs/_models/workspace_storage_info.py b/python/databricks/bundles/jobs/_models/workspace_storage_info.py index 37f44aadfbe..49aef9256c8 100644 --- a/python/databricks/bundles/jobs/_models/workspace_storage_info.py +++ b/python/databricks/bundles/jobs/_models/workspace_storage_info.py @@ -1,194 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.jobs._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.jobs._models.auto_scale import AutoScale, AutoScaleDict, AutoScaleParam -from databricks.bundles.jobs._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.jobs._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.jobs._models.clients_types import ClientsTypes, ClientsTypesDict, ClientsTypesParam -from databricks.bundles.jobs._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.jobs._models.cluster_spec import ClusterSpec, ClusterSpecDict, ClusterSpecParam -from databricks.bundles.jobs._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.jobs._models.docker_basic_auth import DockerBasicAuth, DockerBasicAuthDict, DockerBasicAuthParam -from databricks.bundles.jobs._models.docker_image import DockerImage, DockerImageDict, DockerImageParam -from databricks.bundles.jobs._models.environment import Environment, EnvironmentDict, EnvironmentParam -from databricks.bundles.jobs._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.jobs._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.jobs._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.jobs._models.library import Library, LibraryDict, LibraryParam -from databricks.bundles.jobs._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.jobs._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.jobs._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.jobs._models.node_type_flexibility import NodeTypeFlexibility, NodeTypeFlexibilityDict, NodeTypeFlexibilityParam -from databricks.bundles.jobs._models.python_py_pi_library import PythonPyPiLibrary, PythonPyPiLibraryDict, PythonPyPiLibraryParam -from databricks.bundles.jobs._models.r_cran_library import RCranLibrary, RCranLibraryDict, RCranLibraryParam -from databricks.bundles.jobs._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.jobs._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.jobs._models.workload_type import WorkloadType, WorkloadTypeDict, WorkloadTypeParam -from databricks.bundles.jobs._models.ai_runtime_task import AiRuntimeTask, AiRuntimeTaskDict, AiRuntimeTaskParam -from databricks.bundles.jobs._models.alert_task import AlertTask, AlertTaskDict, AlertTaskParam -from databricks.bundles.jobs._models.alert_task_subscriber import AlertTaskSubscriber, AlertTaskSubscriberDict, AlertTaskSubscriberParam -from databricks.bundles.jobs._models.clean_rooms_notebook_task import CleanRoomsNotebookTask, CleanRoomsNotebookTaskDict, CleanRoomsNotebookTaskParam -from databricks.bundles.jobs._models.compute import Compute, ComputeDict, ComputeParam -from databricks.bundles.jobs._models.compute_config import ComputeConfig, ComputeConfigDict, ComputeConfigParam -from databricks.bundles.jobs._models.compute_spec import ComputeSpec, ComputeSpecDict, ComputeSpecParam -from databricks.bundles.jobs._models.condition_task import ConditionTask, ConditionTaskDict, ConditionTaskParam -from databricks.bundles.jobs._models.continuous import Continuous, ContinuousDict, ContinuousParam -from databricks.bundles.jobs._models.continuous_trigger_configuration import ContinuousTriggerConfiguration, ContinuousTriggerConfigurationDict, ContinuousTriggerConfigurationParam -from databricks.bundles.jobs._models.cron_schedule import CronSchedule, CronScheduleDict, CronScheduleParam -from databricks.bundles.jobs._models.cron_trigger_configuration import CronTriggerConfiguration, CronTriggerConfigurationDict, CronTriggerConfigurationParam -from databricks.bundles.jobs._models.dashboard_task import DashboardTask, DashboardTaskDict, DashboardTaskParam -from databricks.bundles.jobs._models.dbt_platform_task import DbtPlatformTask, DbtPlatformTaskDict, DbtPlatformTaskParam -from databricks.bundles.jobs._models.dbt_task import DbtTask, DbtTaskDict, DbtTaskParam -from databricks.bundles.jobs._models.deployment_spec import DeploymentSpec, DeploymentSpecDict, DeploymentSpecParam -from databricks.bundles.jobs._models.file_arrival_trigger_configuration import FileArrivalTriggerConfiguration, FileArrivalTriggerConfigurationDict, FileArrivalTriggerConfigurationParam -from databricks.bundles.jobs._models.for_each_task import ForEachTask, ForEachTaskDict, ForEachTaskParam -from databricks.bundles.jobs._models.gen_ai_compute_task import GenAiComputeTask, GenAiComputeTaskDict, GenAiComputeTaskParam -from databricks.bundles.jobs._models.git_source import GitSource, GitSourceDict, GitSourceParam -from databricks.bundles.jobs._models.job_cluster import JobCluster, JobClusterDict, JobClusterParam -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications, JobEmailNotificationsDict, JobEmailNotificationsParam -from databricks.bundles.jobs._models.job_environment import JobEnvironment, JobEnvironmentDict, JobEnvironmentParam -from databricks.bundles.jobs._models.job_environment_variables import JobEnvironmentVariables, JobEnvironmentVariablesDict, JobEnvironmentVariablesParam -from databricks.bundles.jobs._models.job_environment_variables_spec import JobEnvironmentVariablesSpec, JobEnvironmentVariablesSpecDict, JobEnvironmentVariablesSpecParam -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings, JobNotificationSettingsDict, JobNotificationSettingsParam -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition, JobParameterDefinitionDict, JobParameterDefinitionParam -from databricks.bundles.jobs._models.job_run_as import JobRunAs, JobRunAsDict, JobRunAsParam -from databricks.bundles.jobs._models.jobs_health_rule import JobsHealthRule, JobsHealthRuleDict, JobsHealthRuleParam -from databricks.bundles.jobs._models.jobs_health_rules import JobsHealthRules, JobsHealthRulesDict, JobsHealthRulesParam -from databricks.bundles.jobs._models.maintenance_window import MaintenanceWindow, MaintenanceWindowDict, MaintenanceWindowParam -from databricks.bundles.jobs._models.model_trigger_configuration import ModelTriggerConfiguration, ModelTriggerConfigurationDict, ModelTriggerConfigurationParam -from databricks.bundles.jobs._models.notebook_task import NotebookTask, NotebookTaskDict, NotebookTaskParam -from databricks.bundles.jobs._models.periodic_trigger_configuration import PeriodicTriggerConfiguration, PeriodicTriggerConfigurationDict, PeriodicTriggerConfigurationParam -from databricks.bundles.jobs._models.pipeline_params import PipelineParams, PipelineParamsDict, PipelineParamsParam -from databricks.bundles.jobs._models.pipeline_task import PipelineTask, PipelineTaskDict, PipelineTaskParam -from databricks.bundles.jobs._models.power_bi_model import PowerBiModel, PowerBiModelDict, PowerBiModelParam -from databricks.bundles.jobs._models.power_bi_table import PowerBiTable, PowerBiTableDict, PowerBiTableParam -from databricks.bundles.jobs._models.power_bi_task import PowerBiTask, PowerBiTaskDict, PowerBiTaskParam -from databricks.bundles.jobs._models.python_operator_task import PythonOperatorTask, PythonOperatorTaskDict, PythonOperatorTaskParam -from databricks.bundles.jobs._models.python_operator_task_parameter import PythonOperatorTaskParameter, PythonOperatorTaskParameterDict, PythonOperatorTaskParameterParam -from databricks.bundles.jobs._models.python_wheel_task import PythonWheelTask, PythonWheelTaskDict, PythonWheelTaskParam -from databricks.bundles.jobs._models.queue_settings import QueueSettings, QueueSettingsDict, QueueSettingsParam -from databricks.bundles.jobs._models.run_job_task import RunJobTask, RunJobTaskDict, RunJobTaskParam -from databricks.bundles.jobs._models.spark_jar_task import SparkJarTask, SparkJarTaskDict, SparkJarTaskParam -from databricks.bundles.jobs._models.spark_python_task import SparkPythonTask, SparkPythonTaskDict, SparkPythonTaskParam -from databricks.bundles.jobs._models.spark_submit_task import SparkSubmitTask, SparkSubmitTaskDict, SparkSubmitTaskParam -from databricks.bundles.jobs._models.sparse_checkout import SparseCheckout, SparseCheckoutDict, SparseCheckoutParam -from databricks.bundles.jobs._models.sql_condition_configuration import SqlConditionConfiguration, SqlConditionConfigurationDict, SqlConditionConfigurationParam -from databricks.bundles.jobs._models.sql_task import SqlTask, SqlTaskDict, SqlTaskParam -from databricks.bundles.jobs._models.sql_task_alert import SqlTaskAlert, SqlTaskAlertDict, SqlTaskAlertParam -from databricks.bundles.jobs._models.sql_task_dashboard import SqlTaskDashboard, SqlTaskDashboardDict, SqlTaskDashboardParam -from databricks.bundles.jobs._models.sql_task_file import SqlTaskFile, SqlTaskFileDict, SqlTaskFileParam -from databricks.bundles.jobs._models.sql_task_query import SqlTaskQuery, SqlTaskQueryDict, SqlTaskQueryParam -from databricks.bundles.jobs._models.sql_task_subscription import SqlTaskSubscription, SqlTaskSubscriptionDict, SqlTaskSubscriptionParam -from databricks.bundles.jobs._models.subscription import Subscription, SubscriptionDict, SubscriptionParam -from databricks.bundles.jobs._models.subscription_attachment import SubscriptionAttachment, SubscriptionAttachmentDict, SubscriptionAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment import SubscriptionAttachmentDataAttachment, SubscriptionAttachmentDataAttachmentDict, SubscriptionAttachmentDataAttachmentParam -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_widget import SubscriptionAttachmentDataAttachmentWidget, SubscriptionAttachmentDataAttachmentWidgetDict, SubscriptionAttachmentDataAttachmentWidgetParam -from databricks.bundles.jobs._models.subscription_attachment_page_attachment import SubscriptionAttachmentPageAttachment, SubscriptionAttachmentPageAttachmentDict, SubscriptionAttachmentPageAttachmentParam -from databricks.bundles.jobs._models.subscription_subscriber import SubscriptionSubscriber, SubscriptionSubscriberDict, SubscriptionSubscriberParam -from databricks.bundles.jobs._models.table_update_trigger_configuration import TableUpdateTriggerConfiguration, TableUpdateTriggerConfigurationDict, TableUpdateTriggerConfigurationParam -from databricks.bundles.jobs._models.task import Task, TaskDict, TaskParam -from databricks.bundles.jobs._models.task_dependency import TaskDependency, TaskDependencyDict, TaskDependencyParam -from databricks.bundles.jobs._models.task_email_notifications import TaskEmailNotifications, TaskEmailNotificationsDict, TaskEmailNotificationsParam -from databricks.bundles.jobs._models.task_notification_settings import TaskNotificationSettings, TaskNotificationSettingsDict, TaskNotificationSettingsParam -from databricks.bundles.jobs._models.trigger_configuration import TriggerConfiguration, TriggerConfigurationDict, TriggerConfigurationParam -from databricks.bundles.jobs._models.trigger_settings import TriggerSettings, TriggerSettingsDict, TriggerSettingsParam -from databricks.bundles.jobs._models.webhook import Webhook, WebhookDict, WebhookParam -from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications, WebhookNotificationsDict, WebhookNotificationsParam -from databricks.bundles.jobs._models.job import Job, JobDict, JobParam -from databricks.bundles.jobs._models.job_permission import JobPermission, JobPermissionDict, JobPermissionParam -from databricks.bundles.jobs._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.jobs._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.jobs._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.jobs._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.jobs._models.data_security_mode import DataSecurityMode, DataSecurityModeParam - -from databricks.bundles.jobs._models.dependency_mode import DependencyMode, DependencyModeParam - -from databricks.bundles.jobs._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.jobs._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.jobs._models.hardware_accelerator_type import HardwareAcceleratorType, HardwareAcceleratorTypeParam - -from databricks.bundles.jobs._models.kind import Kind, KindParam - -from databricks.bundles.jobs._models.runtime_engine import RuntimeEngine, RuntimeEngineParam - -from databricks.bundles.jobs._models.ai_runtime_task_priority_class import AiRuntimeTaskPriorityClass, AiRuntimeTaskPriorityClassParam - -from databricks.bundles.jobs._models.authentication_method import AuthenticationMethod, AuthenticationMethodParam - -from databricks.bundles.jobs._models.compute_spec_accelerator_type import ComputeSpecAcceleratorType, ComputeSpecAcceleratorTypeParam - -from databricks.bundles.jobs._models.condition import Condition, ConditionParam - -from databricks.bundles.jobs._models.condition_task_op import ConditionTaskOp, ConditionTaskOpParam - -from databricks.bundles.jobs._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.jobs._models.git_provider import GitProvider, GitProviderParam - -from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel, JobPermissionLevelParam - -from databricks.bundles.jobs._models.jobs_health_metric import JobsHealthMetric, JobsHealthMetricParam - -from databricks.bundles.jobs._models.jobs_health_operator import JobsHealthOperator, JobsHealthOperatorParam - -from databricks.bundles.jobs._models.model_trigger_configuration_condition import ModelTriggerConfigurationCondition, ModelTriggerConfigurationConditionParam - -from databricks.bundles.jobs._models.pause_status import PauseStatus, PauseStatusParam - -from databricks.bundles.jobs._models.performance_target import PerformanceTarget, PerformanceTargetParam - -from databricks.bundles.jobs._models.periodic_trigger_configuration_time_unit import PeriodicTriggerConfigurationTimeUnit, PeriodicTriggerConfigurationTimeUnitParam - -from databricks.bundles.jobs._models.run_if import RunIf, RunIfParam - -from databricks.bundles.jobs._models.source import Source, SourceParam - -from databricks.bundles.jobs._models.sql_condition_trigger_mode import SqlConditionTriggerMode, SqlConditionTriggerModeParam - -from databricks.bundles.jobs._models.storage_mode import StorageMode, StorageModeParam - -from databricks.bundles.jobs._models.subscription_attachment_data_attachment_file_format import SubscriptionAttachmentDataAttachmentFileFormat, SubscriptionAttachmentDataAttachmentFileFormatParam - -from databricks.bundles.jobs._models.subscription_attachment_page_attachment_page_selection import SubscriptionAttachmentPageAttachmentPageSelection, SubscriptionAttachmentPageAttachmentPageSelectionParam - -from databricks.bundles.jobs._models.task_retry_mode import TaskRetryMode, TaskRetryModeParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class WorkspaceStorageInfo: """ A storage location in Workspace Filesystem (WSFS) """ + destination: VariableOr[str] """ wsfs destination, e.g. `workspace:/cluster-init-scripts/setup-datadog.sh` """ @classmethod - def from_dict(cls, value: 'WorkspaceStorageInfoDict') -> 'Self': + def from_dict(cls, value: "WorkspaceStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'WorkspaceStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "WorkspaceStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class WorkspaceStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/__init__.py b/python/databricks/bundles/mcp_services/__init__.py index 00f577ad846..a4fe01b5199 100644 --- a/python/databricks/bundles/mcp_services/__init__.py +++ b/python/databricks/bundles/mcp_services/__init__.py @@ -1,42 +1,69 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'McpService', - 'McpServiceConfig', - 'McpServiceConfigDict', - 'McpServiceConfigParam', - 'McpServiceConfigSourceConnection', - 'McpServiceConfigSourceConnectionDict', - 'McpServiceConfigSourceConnectionParam', - 'McpServiceDict', - 'McpServiceParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'RateLimit', - 'RateLimitDict', - 'RateLimitParam', - 'RateLimitRateLimitKey', - 'RateLimitRateLimitKeyParam', - 'RateLimitRateLimitRenewalPeriod', - 'RateLimitRateLimitRenewalPeriodParam', + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "McpService", + "McpServiceConfig", + "McpServiceConfigDict", + "McpServiceConfigParam", + "McpServiceConfigSourceConnection", + "McpServiceConfigSourceConnectionDict", + "McpServiceConfigSourceConnectionParam", + "McpServiceDict", + "McpServiceParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", + "RateLimit", + "RateLimitDict", + "RateLimitParam", + "RateLimitRateLimitKey", + "RateLimitRateLimitKeyParam", + "RateLimitRateLimitRenewalPeriod", + "RateLimitRateLimitRenewalPeriodParam", ] -from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam +from databricks.bundles.mcp_services._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.mcp_services._models.mcp_service import ( + McpService, + McpServiceDict, + McpServiceParam, +) +from databricks.bundles.mcp_services._models.mcp_service_config import ( + McpServiceConfig, + McpServiceConfigDict, + McpServiceConfigParam, +) +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import ( + McpServiceConfigSourceConnection, + McpServiceConfigSourceConnectionDict, + McpServiceConfigSourceConnectionParam, +) from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.mcp_services._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) +from databricks.bundles.mcp_services._models.rate_limit import ( + RateLimit, + RateLimitDict, + RateLimitParam, +) +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import ( + RateLimitRateLimitKey, + RateLimitRateLimitKeyParam, +) +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import ( + RateLimitRateLimitRenewalPeriod, + RateLimitRateLimitRenewalPeriodParam, +) diff --git a/python/databricks/bundles/mcp_services/_models/lifecycle.py b/python/databricks/bundles/mcp_services/_models/lifecycle.py index 02ab7b03257..697776a198e 100644 --- a/python/databricks/bundles/mcp_services/_models/lifecycle.py +++ b/python/databricks/bundles/mcp_services/_models/lifecycle.py @@ -1,29 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam -from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -34,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/mcp_service.py b/python/databricks/bundles/mcp_services/_models/mcp_service.py index 63dd67b86de..1050e5153f4 100644 --- a/python/databricks/bundles/mcp_services/_models/mcp_service.py +++ b/python/databricks/bundles/mcp_services/_models/mcp_service.py @@ -1,29 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.mcp_services._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.mcp_services._models.mcp_service_config import ( + McpServiceConfig, + McpServiceConfigParam, +) +from databricks.bundles.mcp_services._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class McpService(Resource): """""" @@ -41,12 +45,11 @@ class McpService(Resource): lifecycle: VariableOrOptional[Lifecycle] = None @classmethod - def from_dict(cls, value: 'McpServiceDict') -> 'Self': + def from_dict(cls, value: "McpServiceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'McpServiceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "McpServiceDict": + return _transform_to_json_value(self) # type:ignore class McpServiceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/mcp_service_config.py b/python/databricks/bundles/mcp_services/_models/mcp_service_config.py index 36b70fd4bce..5a9a984bf01 100644 --- a/python/databricks/bundles/mcp_services/_models/mcp_service_config.py +++ b/python/databricks/bundles/mcp_services/_models/mcp_service_config.py @@ -1,29 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam -from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import ( + McpServiceConfigSourceConnection, + McpServiceConfigSourceConnectionParam, +) +from databricks.bundles.mcp_services._models.rate_limit import ( + RateLimit, + RateLimitParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class McpServiceConfig: """ @@ -31,6 +26,7 @@ class McpServiceConfig: tool selectors, and rate limits -- the fields that configure how the MCP service behaves at invocation time. """ + include_tool_selectors: VariableOrList[str] = field(default_factory=list) """ Tool names or prefix patterns to expose from the MCP server. Use exact @@ -52,12 +48,11 @@ class McpServiceConfig: """ @classmethod - def from_dict(cls, value: 'McpServiceConfigDict') -> 'Self': + def from_dict(cls, value: "McpServiceConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'McpServiceConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "McpServiceConfigDict": + return _transform_to_json_value(self) # type:ignore class McpServiceConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/mcp_service_config_source_connection.py b/python/databricks/bundles/mcp_services/_models/mcp_service_config_source_connection.py index 837f6f2a2f9..4da924fcd38 100644 --- a/python/databricks/bundles/mcp_services/_models/mcp_service_config_source_connection.py +++ b/python/databricks/bundles/mcp_services/_models/mcp_service_config_source_connection.py @@ -1,29 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam -from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class McpServiceConfigSourceConnection: """ @@ -33,6 +20,7 @@ class McpServiceConfigSourceConnection: populates the resolved connection metadata. If the connection is deleted, its reference remains visible so you can identify the broken dependency. """ + name: VariableOr[str] """ Resource name of the Unity Catalog connection used to access the MCP @@ -40,12 +28,11 @@ class McpServiceConfigSourceConnection: """ @classmethod - def from_dict(cls, value: 'McpServiceConfigSourceConnectionDict') -> 'Self': + def from_dict(cls, value: "McpServiceConfigSourceConnectionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'McpServiceConfigSourceConnectionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "McpServiceConfigSourceConnectionDict": + return _transform_to_json_value(self) # type:ignore class McpServiceConfigSourceConnectionDict(TypedDict, total=False): @@ -58,4 +45,6 @@ class McpServiceConfigSourceConnectionDict(TypedDict, total=False): """ -McpServiceConfigSourceConnectionParam = McpServiceConfigSourceConnectionDict | McpServiceConfigSourceConnection +McpServiceConfigSourceConnectionParam = ( + McpServiceConfigSourceConnectionDict | McpServiceConfigSourceConnection +) diff --git a/python/databricks/bundles/mcp_services/_models/privilege.py b/python/databricks/bundles/mcp_services/_models/privilege.py index f8ad08d6ec1..df45211a76a 100644 --- a/python/databricks/bundles/mcp_services/_models/privilege.py +++ b/python/databricks/bundles/mcp_services/_models/privilege.py @@ -1,23 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam +from typing import Literal class Privilege(Enum): @@ -74,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/mcp_services/_models/privilege_assignment.py b/python/databricks/bundles/mcp_services/_models/privilege_assignment.py index c95110a7933..cdaba80f76a 100644 --- a/python/databricks/bundles/mcp_services/_models/privilege_assignment.py +++ b/python/databricks/bundles/mcp_services/_models/privilege_assignment.py @@ -1,29 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam -from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -40,12 +28,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/rate_limit.py b/python/databricks/bundles/mcp_services/_models/rate_limit.py index 5de5d60246a..cba7fed7056 100644 --- a/python/databricks/bundles/mcp_services/_models/rate_limit.py +++ b/python/databricks/bundles/mcp_services/_models/rate_limit.py @@ -1,29 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam -from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import ( + RateLimitRateLimitKey, + RateLimitRateLimitKeyParam, +) +from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import ( + RateLimitRateLimitRenewalPeriod, + RateLimitRateLimitRenewalPeriodParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RateLimit: """ @@ -31,6 +26,7 @@ class RateLimit: unset to impose no limit on that dimension; set a value to cap that dimension within the renewal period. """ + key: VariableOr[RateLimitRateLimitKey] """ Scope of the rate limit. Depending on this value, the limit applies to a @@ -62,12 +58,11 @@ class RateLimit: """ @classmethod - def from_dict(cls, value: 'RateLimitDict') -> 'Self': + def from_dict(cls, value: "RateLimitDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RateLimitDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RateLimitDict": + return _transform_to_json_value(self) # type:ignore class RateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_key.py b/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_key.py index 8f5958333b3..fb907da810f 100644 --- a/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_key.py +++ b/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_key.py @@ -1,33 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam -from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam +from typing import Literal class RateLimitRateLimitKey(Enum): """ Scope key for a rate limit. """ + RATE_LIMIT_KEY_USER = "RATE_LIMIT_KEY_USER" RATE_LIMIT_KEY_USER_GROUP = "RATE_LIMIT_KEY_USER_GROUP" RATE_LIMIT_KEY_SERVICE_PRINCIPAL = "RATE_LIMIT_KEY_SERVICE_PRINCIPAL" RATE_LIMIT_KEY_SERVICE = "RATE_LIMIT_KEY_SERVICE" RATE_LIMIT_KEY_USER_DEFAULT = "RATE_LIMIT_KEY_USER_DEFAULT" -RateLimitRateLimitKeyParam = Literal["RATE_LIMIT_KEY_USER", "RATE_LIMIT_KEY_USER_GROUP", "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", "RATE_LIMIT_KEY_SERVICE", "RATE_LIMIT_KEY_USER_DEFAULT"] | RateLimitRateLimitKey + +RateLimitRateLimitKeyParam = ( + Literal[ + "RATE_LIMIT_KEY_USER", + "RATE_LIMIT_KEY_USER_GROUP", + "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", + "RATE_LIMIT_KEY_SERVICE", + "RATE_LIMIT_KEY_USER_DEFAULT", + ] + | RateLimitRateLimitKey +) diff --git a/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_renewal_period.py b/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_renewal_period.py index c37c8d583a4..58636c784c0 100644 --- a/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_renewal_period.py +++ b/python/databricks/bundles/mcp_services/_models/rate_limit_rate_limit_renewal_period.py @@ -1,30 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig, McpServiceConfigDict, McpServiceConfigParam -from databricks.bundles.mcp_services._models.mcp_service_config_source_connection import McpServiceConfigSourceConnection, McpServiceConfigSourceConnectionDict, McpServiceConfigSourceConnectionParam -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.mcp_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.mcp_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.mcp_services._models.mcp_service import McpService, McpServiceDict, McpServiceParam -from databricks.bundles.mcp_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.mcp_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam +from typing import Literal class RateLimitRateLimitRenewalPeriod(Enum): """ Renewal period for a rate limit. """ + RATE_LIMIT_RENEWAL_PERIOD_MINUTE = "RATE_LIMIT_RENEWAL_PERIOD_MINUTE" RATE_LIMIT_RENEWAL_PERIOD_HOUR = "RATE_LIMIT_RENEWAL_PERIOD_HOUR" -RateLimitRateLimitRenewalPeriodParam = Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] | RateLimitRateLimitRenewalPeriod + +RateLimitRateLimitRenewalPeriodParam = ( + Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] + | RateLimitRateLimitRenewalPeriod +) diff --git a/python/databricks/bundles/model_provider_services/__init__.py b/python/databricks/bundles/model_provider_services/__init__.py index 14f8af90796..e7b2a5b13f7 100644 --- a/python/databricks/bundles/model_provider_services/__init__.py +++ b/python/databricks/bundles/model_provider_services/__init__.py @@ -1,138 +1,262 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'InferenceTableConfig', - 'InferenceTableConfigDict', - 'InferenceTableConfigParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'ModelProviderService', - 'ModelProviderServiceConfig', - 'ModelProviderServiceConfigAmazonBedrockProviderConfig', - 'ModelProviderServiceConfigAmazonBedrockProviderConfigDict', - 'ModelProviderServiceConfigAmazonBedrockProviderConfigParam', - 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfig', - 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict', - 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam', - 'ModelProviderServiceConfigAnthropicProviderConfig', - 'ModelProviderServiceConfigAnthropicProviderConfigDict', - 'ModelProviderServiceConfigAnthropicProviderConfigParam', - 'ModelProviderServiceConfigAnthropicProviderDirectConfig', - 'ModelProviderServiceConfigAnthropicProviderDirectConfigDict', - 'ModelProviderServiceConfigAnthropicProviderDirectConfigParam', - 'ModelProviderServiceConfigAnthropicProviderRelayedConfig', - 'ModelProviderServiceConfigAnthropicProviderRelayedConfigDict', - 'ModelProviderServiceConfigAnthropicProviderRelayedConfigParam', - 'ModelProviderServiceConfigAwsAccessKey', - 'ModelProviderServiceConfigAwsAccessKeyDict', - 'ModelProviderServiceConfigAwsAccessKeyParam', - 'ModelProviderServiceConfigAzureOpenAiProviderConfig', - 'ModelProviderServiceConfigAzureOpenAiProviderConfigDict', - 'ModelProviderServiceConfigAzureOpenAiProviderConfigParam', - 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfig', - 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict', - 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam', - 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth', - 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict', - 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam', - 'ModelProviderServiceConfigCustomProviderConfig', - 'ModelProviderServiceConfigCustomProviderConfigDict', - 'ModelProviderServiceConfigCustomProviderConfigParam', - 'ModelProviderServiceConfigCustomProviderDirectConfig', - 'ModelProviderServiceConfigCustomProviderDirectConfigDict', - 'ModelProviderServiceConfigCustomProviderDirectConfigParam', - 'ModelProviderServiceConfigDict', - 'ModelProviderServiceConfigEntraServicePrincipal', - 'ModelProviderServiceConfigEntraServicePrincipalDict', - 'ModelProviderServiceConfigEntraServicePrincipalParam', - 'ModelProviderServiceConfigExternalModelProviderType', - 'ModelProviderServiceConfigExternalModelProviderTypeParam', - 'ModelProviderServiceConfigGeminiEnterpriseProviderConfig', - 'ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict', - 'ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam', - 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig', - 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict', - 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam', - 'ModelProviderServiceConfigMicrosoftFoundryProviderConfig', - 'ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict', - 'ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam', - 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig', - 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict', - 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam', - 'ModelProviderServiceConfigModelTargetConfig', - 'ModelProviderServiceConfigModelTargetConfigDict', - 'ModelProviderServiceConfigModelTargetConfigParam', - 'ModelProviderServiceConfigOpenAiProviderConfig', - 'ModelProviderServiceConfigOpenAiProviderConfigDict', - 'ModelProviderServiceConfigOpenAiProviderConfigParam', - 'ModelProviderServiceConfigOpenAiProviderDirectConfig', - 'ModelProviderServiceConfigOpenAiProviderDirectConfigDict', - 'ModelProviderServiceConfigOpenAiProviderDirectConfigParam', - 'ModelProviderServiceConfigParam', - 'ModelProviderServiceConfigProviderPricingConfig', - 'ModelProviderServiceConfigProviderPricingConfigDict', - 'ModelProviderServiceConfigProviderPricingConfigParam', - 'ModelProviderServiceConfigProviderSecret', - 'ModelProviderServiceConfigProviderSecretDict', - 'ModelProviderServiceConfigProviderSecretParam', - 'ModelProviderServiceConfigSecretReference', - 'ModelProviderServiceConfigSecretReferenceDict', - 'ModelProviderServiceConfigSecretReferenceParam', - 'ModelProviderServiceConfigServiceCredential', - 'ModelProviderServiceConfigServiceCredentialDict', - 'ModelProviderServiceConfigServiceCredentialParam', - 'ModelProviderServiceDict', - 'ModelProviderServiceParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'RateLimit', - 'RateLimitDict', - 'RateLimitParam', - 'RateLimitRateLimitKey', - 'RateLimitRateLimitKeyParam', - 'RateLimitRateLimitRenewalPeriod', - 'RateLimitRateLimitRenewalPeriodParam', + "InferenceTableConfig", + "InferenceTableConfigDict", + "InferenceTableConfigParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "ModelProviderService", + "ModelProviderServiceConfig", + "ModelProviderServiceConfigAmazonBedrockProviderConfig", + "ModelProviderServiceConfigAmazonBedrockProviderConfigDict", + "ModelProviderServiceConfigAmazonBedrockProviderConfigParam", + "ModelProviderServiceConfigAmazonBedrockProviderDirectConfig", + "ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict", + "ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam", + "ModelProviderServiceConfigAnthropicProviderConfig", + "ModelProviderServiceConfigAnthropicProviderConfigDict", + "ModelProviderServiceConfigAnthropicProviderConfigParam", + "ModelProviderServiceConfigAnthropicProviderDirectConfig", + "ModelProviderServiceConfigAnthropicProviderDirectConfigDict", + "ModelProviderServiceConfigAnthropicProviderDirectConfigParam", + "ModelProviderServiceConfigAnthropicProviderRelayedConfig", + "ModelProviderServiceConfigAnthropicProviderRelayedConfigDict", + "ModelProviderServiceConfigAnthropicProviderRelayedConfigParam", + "ModelProviderServiceConfigAwsAccessKey", + "ModelProviderServiceConfigAwsAccessKeyDict", + "ModelProviderServiceConfigAwsAccessKeyParam", + "ModelProviderServiceConfigAzureOpenAiProviderConfig", + "ModelProviderServiceConfigAzureOpenAiProviderConfigDict", + "ModelProviderServiceConfigAzureOpenAiProviderConfigParam", + "ModelProviderServiceConfigAzureOpenAiProviderDirectConfig", + "ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict", + "ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam", + "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth", + "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict", + "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam", + "ModelProviderServiceConfigCustomProviderConfig", + "ModelProviderServiceConfigCustomProviderConfigDict", + "ModelProviderServiceConfigCustomProviderConfigParam", + "ModelProviderServiceConfigCustomProviderDirectConfig", + "ModelProviderServiceConfigCustomProviderDirectConfigDict", + "ModelProviderServiceConfigCustomProviderDirectConfigParam", + "ModelProviderServiceConfigDict", + "ModelProviderServiceConfigEntraServicePrincipal", + "ModelProviderServiceConfigEntraServicePrincipalDict", + "ModelProviderServiceConfigEntraServicePrincipalParam", + "ModelProviderServiceConfigExternalModelProviderType", + "ModelProviderServiceConfigExternalModelProviderTypeParam", + "ModelProviderServiceConfigGeminiEnterpriseProviderConfig", + "ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict", + "ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam", + "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig", + "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict", + "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam", + "ModelProviderServiceConfigMicrosoftFoundryProviderConfig", + "ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict", + "ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam", + "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig", + "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict", + "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam", + "ModelProviderServiceConfigModelTargetConfig", + "ModelProviderServiceConfigModelTargetConfigDict", + "ModelProviderServiceConfigModelTargetConfigParam", + "ModelProviderServiceConfigOpenAiProviderConfig", + "ModelProviderServiceConfigOpenAiProviderConfigDict", + "ModelProviderServiceConfigOpenAiProviderConfigParam", + "ModelProviderServiceConfigOpenAiProviderDirectConfig", + "ModelProviderServiceConfigOpenAiProviderDirectConfigDict", + "ModelProviderServiceConfigOpenAiProviderDirectConfigParam", + "ModelProviderServiceConfigParam", + "ModelProviderServiceConfigProviderPricingConfig", + "ModelProviderServiceConfigProviderPricingConfigDict", + "ModelProviderServiceConfigProviderPricingConfigParam", + "ModelProviderServiceConfigProviderSecret", + "ModelProviderServiceConfigProviderSecretDict", + "ModelProviderServiceConfigProviderSecretParam", + "ModelProviderServiceConfigSecretReference", + "ModelProviderServiceConfigSecretReferenceDict", + "ModelProviderServiceConfigSecretReferenceParam", + "ModelProviderServiceConfigServiceCredential", + "ModelProviderServiceConfigServiceCredentialDict", + "ModelProviderServiceConfigServiceCredentialParam", + "ModelProviderServiceDict", + "ModelProviderServiceParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", + "RateLimit", + "RateLimitDict", + "RateLimitParam", + "RateLimitRateLimitKey", + "RateLimitRateLimitKeyParam", + "RateLimitRateLimitRenewalPeriod", + "RateLimitRateLimitRenewalPeriodParam", ] -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.model_provider_services._models.inference_table_config import ( + InferenceTableConfig, + InferenceTableConfigDict, + InferenceTableConfigParam, +) +from databricks.bundles.model_provider_services._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service import ( + ModelProviderService, + ModelProviderServiceDict, + ModelProviderServiceParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config import ( + ModelProviderServiceConfig, + ModelProviderServiceConfigDict, + ModelProviderServiceConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ( + ModelProviderServiceConfigAmazonBedrockProviderConfig, + ModelProviderServiceConfigAmazonBedrockProviderConfigDict, + ModelProviderServiceConfigAmazonBedrockProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ( + ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, + ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, + ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ( + ModelProviderServiceConfigAnthropicProviderConfig, + ModelProviderServiceConfigAnthropicProviderConfigDict, + ModelProviderServiceConfigAnthropicProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ( + ModelProviderServiceConfigAnthropicProviderDirectConfig, + ModelProviderServiceConfigAnthropicProviderDirectConfigDict, + ModelProviderServiceConfigAnthropicProviderDirectConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ( + ModelProviderServiceConfigAnthropicProviderRelayedConfig, + ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, + ModelProviderServiceConfigAnthropicProviderRelayedConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ( + ModelProviderServiceConfigAwsAccessKey, + ModelProviderServiceConfigAwsAccessKeyDict, + ModelProviderServiceConfigAwsAccessKeyParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ( + ModelProviderServiceConfigAzureOpenAiProviderConfig, + ModelProviderServiceConfigAzureOpenAiProviderConfigDict, + ModelProviderServiceConfigAzureOpenAiProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ( + ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, + ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, + ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ( + ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, + ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, + ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ( + ModelProviderServiceConfigCustomProviderConfig, + ModelProviderServiceConfigCustomProviderConfigDict, + ModelProviderServiceConfigCustomProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ( + ModelProviderServiceConfigCustomProviderDirectConfig, + ModelProviderServiceConfigCustomProviderDirectConfigDict, + ModelProviderServiceConfigCustomProviderDirectConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ( + ModelProviderServiceConfigEntraServicePrincipal, + ModelProviderServiceConfigEntraServicePrincipalDict, + ModelProviderServiceConfigEntraServicePrincipalParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ( + ModelProviderServiceConfigExternalModelProviderType, + ModelProviderServiceConfigExternalModelProviderTypeParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ( + ModelProviderServiceConfigGeminiEnterpriseProviderConfig, + ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, + ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ( + ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, + ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, + ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ( + ModelProviderServiceConfigMicrosoftFoundryProviderConfig, + ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, + ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ( + ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, + ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, + ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ( + ModelProviderServiceConfigModelTargetConfig, + ModelProviderServiceConfigModelTargetConfigDict, + ModelProviderServiceConfigModelTargetConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ( + ModelProviderServiceConfigOpenAiProviderConfig, + ModelProviderServiceConfigOpenAiProviderConfigDict, + ModelProviderServiceConfigOpenAiProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ( + ModelProviderServiceConfigOpenAiProviderDirectConfig, + ModelProviderServiceConfigOpenAiProviderDirectConfigDict, + ModelProviderServiceConfigOpenAiProviderDirectConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ( + ModelProviderServiceConfigProviderPricingConfig, + ModelProviderServiceConfigProviderPricingConfigDict, + ModelProviderServiceConfigProviderPricingConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretDict, + ModelProviderServiceConfigProviderSecretParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ( + ModelProviderServiceConfigSecretReference, + ModelProviderServiceConfigSecretReferenceDict, + ModelProviderServiceConfigSecretReferenceParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( + ModelProviderServiceConfigServiceCredential, + ModelProviderServiceConfigServiceCredentialDict, + ModelProviderServiceConfigServiceCredentialParam, +) +from databricks.bundles.model_provider_services._models.privilege import ( + Privilege, + PrivilegeParam, +) +from databricks.bundles.model_provider_services._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) +from databricks.bundles.model_provider_services._models.rate_limit import ( + RateLimit, + RateLimitDict, + RateLimitParam, +) +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import ( + RateLimitRateLimitKey, + RateLimitRateLimitKeyParam, +) +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import ( + RateLimitRateLimitRenewalPeriod, + RateLimitRateLimitRenewalPeriodParam, +) diff --git a/python/databricks/bundles/model_provider_services/_models/inference_table_config.py b/python/databricks/bundles/model_provider_services/_models/inference_table_config.py index c246dd20353..45165416e17 100644 --- a/python/databricks/bundles/model_provider_services/_models/inference_table_config.py +++ b/python/databricks/bundles/model_provider_services/_models/inference_table_config.py @@ -1,54 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InferenceTableConfig: """ @@ -56,6 +18,7 @@ class InferenceTableConfig: inference table. When this configuration is present, payload logging is enabled by default. """ + parent: VariableOr[str] """ Parent Unity Catalog schema where the inference table is created, in the @@ -73,12 +36,11 @@ class InferenceTableConfig: """ @classmethod - def from_dict(cls, value: 'InferenceTableConfigDict') -> 'Self': + def from_dict(cls, value: "InferenceTableConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InferenceTableConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InferenceTableConfigDict": + return _transform_to_json_value(self) # type:ignore class InferenceTableConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/lifecycle.py b/python/databricks/bundles/model_provider_services/_models/lifecycle.py index fd87749f79e..697776a198e 100644 --- a/python/databricks/bundles/model_provider_services/_models/lifecycle.py +++ b/python/databricks/bundles/model_provider_services/_models/lifecycle.py @@ -1,54 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -59,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service.py index 520c98c8609..41a21652ac8 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service.py @@ -1,54 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.model_provider_services._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config import ( + ModelProviderServiceConfig, + ModelProviderServiceConfigParam, +) +from databricks.bundles.model_provider_services._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderService(Resource): """""" @@ -66,12 +45,11 @@ class ModelProviderService(Resource): lifecycle: VariableOrOptional[Lifecycle] = None @classmethod - def from_dict(cls, value: 'ModelProviderServiceDict') -> 'Self': + def from_dict(cls, value: "ModelProviderServiceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config.py index 7b945a3e222..d2be6cc6d90 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config.py @@ -1,54 +1,64 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.model_provider_services._models.inference_table_config import ( + InferenceTableConfig, + InferenceTableConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ( + ModelProviderServiceConfigAmazonBedrockProviderConfig, + ModelProviderServiceConfigAmazonBedrockProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ( + ModelProviderServiceConfigAnthropicProviderConfig, + ModelProviderServiceConfigAnthropicProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ( + ModelProviderServiceConfigAzureOpenAiProviderConfig, + ModelProviderServiceConfigAzureOpenAiProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ( + ModelProviderServiceConfigCustomProviderConfig, + ModelProviderServiceConfigCustomProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ( + ModelProviderServiceConfigExternalModelProviderType, + ModelProviderServiceConfigExternalModelProviderTypeParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ( + ModelProviderServiceConfigGeminiEnterpriseProviderConfig, + ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ( + ModelProviderServiceConfigMicrosoftFoundryProviderConfig, + ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ( + ModelProviderServiceConfigModelTargetConfig, + ModelProviderServiceConfigModelTargetConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ( + ModelProviderServiceConfigOpenAiProviderConfig, + ModelProviderServiceConfigOpenAiProviderConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ( + ModelProviderServiceConfigProviderPricingConfig, + ModelProviderServiceConfigProviderPricingConfigParam, +) +from databricks.bundles.model_provider_services._models.rate_limit import ( + RateLimit, + RateLimitParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfig: """ @@ -58,6 +68,7 @@ class ModelProviderServiceConfig: headers, query parameters, and unmanaged subpaths cross the trust boundary to the upstream provider. """ + allow_all_targets: VariableOrOptional[bool] = None """ When true, accepts any model exposed by the upstream provider; `targets` @@ -65,23 +76,31 @@ class ModelProviderServiceConfig: models listed in `targets` are routable. Defaults to false. """ - amazon_bedrock: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderConfig] = None + amazon_bedrock: VariableOrOptional[ + ModelProviderServiceConfigAmazonBedrockProviderConfig + ] = None """ Amazon Bedrock provider configuration. """ - anthropic: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderConfig] = None + anthropic: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderConfig] = ( + None + ) """ Anthropic provider configuration. Exactly one of `direct` or `relayed` must be set on Create; the two are mutually exclusive. """ - azure_openai: VariableOrOptional[ModelProviderServiceConfigAzureOpenAiProviderConfig] = None + azure_openai: VariableOrOptional[ + ModelProviderServiceConfigAzureOpenAiProviderConfig + ] = None """ Azure OpenAI provider configuration. """ - bedrock_mantle: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderConfig] = None + bedrock_mantle: VariableOrOptional[ + ModelProviderServiceConfigAmazonBedrockProviderConfig + ] = None """ Amazon Bedrock provider configuration. """ @@ -117,7 +136,9 @@ class ModelProviderServiceConfig: rejected whenever Service Policies are attached to this securable. """ - gemini_enterprise: VariableOrOptional[ModelProviderServiceConfigGeminiEnterpriseProviderConfig] = None + gemini_enterprise: VariableOrOptional[ + ModelProviderServiceConfigGeminiEnterpriseProviderConfig + ] = None """ Gemini Enterprise provider configuration. """ @@ -129,7 +150,9 @@ class ModelProviderServiceConfig: service's inference table instead. """ - microsoft_foundry: VariableOrOptional[ModelProviderServiceConfigMicrosoftFoundryProviderConfig] = None + microsoft_foundry: VariableOrOptional[ + ModelProviderServiceConfigMicrosoftFoundryProviderConfig + ] = None """ Microsoft Foundry provider configuration. """ @@ -145,7 +168,9 @@ class ModelProviderServiceConfig: configuration is retained on Create and Update and applies no adjustment. """ - provider_type: VariableOrOptional[ModelProviderServiceConfigExternalModelProviderType] = None + provider_type: VariableOrOptional[ + ModelProviderServiceConfigExternalModelProviderType + ] = None """ External model provider. Required on Create and immutable thereafter. Set the matching provider-specific configuration, such as `openai`, @@ -158,7 +183,9 @@ class ModelProviderServiceConfig: routed through a model service use that model service's rate limits instead. """ - targets: VariableOrList[ModelProviderServiceConfigModelTargetConfig] = field(default_factory=list) + targets: VariableOrList[ModelProviderServiceConfigModelTargetConfig] = field( + default_factory=list + ) """ Models and provider-native API types exposed by this provider service. Each entry must include at least one `native_api_types` value. When @@ -169,12 +196,11 @@ class ModelProviderServiceConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigDict') -> 'Self': + def from_dict(cls, value: "ModelProviderServiceConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigDict(TypedDict, total=False): @@ -187,23 +213,31 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): models listed in `targets` are routable. Defaults to false. """ - amazon_bedrock: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderConfigParam] + amazon_bedrock: VariableOrOptional[ + ModelProviderServiceConfigAmazonBedrockProviderConfigParam + ] """ Amazon Bedrock provider configuration. """ - anthropic: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderConfigParam] + anthropic: VariableOrOptional[ + ModelProviderServiceConfigAnthropicProviderConfigParam + ] """ Anthropic provider configuration. Exactly one of `direct` or `relayed` must be set on Create; the two are mutually exclusive. """ - azure_openai: VariableOrOptional[ModelProviderServiceConfigAzureOpenAiProviderConfigParam] + azure_openai: VariableOrOptional[ + ModelProviderServiceConfigAzureOpenAiProviderConfigParam + ] """ Azure OpenAI provider configuration. """ - bedrock_mantle: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderConfigParam] + bedrock_mantle: VariableOrOptional[ + ModelProviderServiceConfigAmazonBedrockProviderConfigParam + ] """ Amazon Bedrock provider configuration. """ @@ -239,7 +273,9 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): rejected whenever Service Policies are attached to this securable. """ - gemini_enterprise: VariableOrOptional[ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam] + gemini_enterprise: VariableOrOptional[ + ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam + ] """ Gemini Enterprise provider configuration. """ @@ -251,7 +287,9 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): service's inference table instead. """ - microsoft_foundry: VariableOrOptional[ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam] + microsoft_foundry: VariableOrOptional[ + ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam + ] """ Microsoft Foundry provider configuration. """ @@ -267,7 +305,9 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): configuration is retained on Create and Update and applies no adjustment. """ - provider_type: VariableOrOptional[ModelProviderServiceConfigExternalModelProviderTypeParam] + provider_type: VariableOrOptional[ + ModelProviderServiceConfigExternalModelProviderTypeParam + ] """ External model provider. Required on Create and immutable thereafter. Set the matching provider-specific configuration, such as `openai`, @@ -291,4 +331,6 @@ class ModelProviderServiceConfigDict(TypedDict, total=False): """ -ModelProviderServiceConfigParam = ModelProviderServiceConfigDict | ModelProviderServiceConfig +ModelProviderServiceConfigParam = ( + ModelProviderServiceConfigDict | ModelProviderServiceConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_config.py index 4a4823b1188..cf6015f13c5 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_config.py @@ -1,80 +1,55 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ( + ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, + ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigAmazonBedrockProviderConfig: """ Amazon Bedrock provider configuration. """ - direct: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderDirectConfig] = None + + direct: VariableOrOptional[ + ModelProviderServiceConfigAmazonBedrockProviderDirectConfig + ] = None """ Amazon Bedrock region and authentication configuration. """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigAmazonBedrockProviderConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigAmazonBedrockProviderConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigAmazonBedrockProviderConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigAmazonBedrockProviderConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigAmazonBedrockProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam] + direct: VariableOrOptional[ + ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam + ] """ Amazon Bedrock region and authentication configuration. """ -ModelProviderServiceConfigAmazonBedrockProviderConfigParam = ModelProviderServiceConfigAmazonBedrockProviderConfigDict | ModelProviderServiceConfigAmazonBedrockProviderConfig +ModelProviderServiceConfigAmazonBedrockProviderConfigParam = ( + ModelProviderServiceConfigAmazonBedrockProviderConfigDict + | ModelProviderServiceConfigAmazonBedrockProviderConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_direct_config.py index f075c1e30e4..911338ad79e 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_amazon_bedrock_provider_direct_config.py @@ -1,59 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ( + ModelProviderServiceConfigAwsAccessKey, + ModelProviderServiceConfigAwsAccessKeyParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( + ModelProviderServiceConfigServiceCredential, + ModelProviderServiceConfigServiceCredentialParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigAmazonBedrockProviderDirectConfig: """ Direct form of Amazon Bedrock provider config. - + Authentication is one of two mutually exclusive modes, exactly one of which must be supplied on Create: - Access keys: set `aws_access_key`, leave `service_credential` unset. @@ -63,6 +33,7 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfig: held on this message. Setting more than one mode is rejected. """ + aws_access_key: VariableOrOptional[ModelProviderServiceConfigAwsAccessKey] = None """ AWS access-key-pair authentication. Set `access_key_id` and @@ -76,7 +47,9 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfig: Required on Create. """ - service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredential] = None + service_credential: VariableOrOptional[ + ModelProviderServiceConfigServiceCredential + ] = None """ Reference to a Unity Catalog service credential authorizing Bedrock requests. On Create, supply `service_credential.name` in the form @@ -87,15 +60,20 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict( + self, + ) -> "ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict": + return _transform_to_json_value(self) # type:ignore -class ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict(TypedDict, total=False): +class ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict( + TypedDict, total=False +): """""" aws_access_key: VariableOrOptional[ModelProviderServiceConfigAwsAccessKeyParam] @@ -111,7 +89,9 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict(TypedDict, Required on Create. """ - service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredentialParam] + service_credential: VariableOrOptional[ + ModelProviderServiceConfigServiceCredentialParam + ] """ Reference to a Unity Catalog service credential authorizing Bedrock requests. On Create, supply `service_credential.name` in the form @@ -122,4 +102,7 @@ class ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict(TypedDict, """ -ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam = ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict | ModelProviderServiceConfigAmazonBedrockProviderDirectConfig +ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam = ( + ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict + | ModelProviderServiceConfigAmazonBedrockProviderDirectConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_config.py index ea8c077414a..c7b3311ca93 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_config.py @@ -1,67 +1,42 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ( + ModelProviderServiceConfigAnthropicProviderDirectConfig, + ModelProviderServiceConfigAnthropicProviderDirectConfigParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ( + ModelProviderServiceConfigAnthropicProviderRelayedConfig, + ModelProviderServiceConfigAnthropicProviderRelayedConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigAnthropicProviderConfig: """ Anthropic provider configuration. Exactly one of `direct` or `relayed` must be set on Create; the two are mutually exclusive. """ - direct: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderDirectConfig] = None + + direct: VariableOrOptional[ + ModelProviderServiceConfigAnthropicProviderDirectConfig + ] = None """ Direct authentication with an API key supplied in `direct.api_key.plaintext`. Required unless `relayed` is set. """ - relayed: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderRelayedConfig] = None + relayed: VariableOrOptional[ + ModelProviderServiceConfigAnthropicProviderRelayedConfig + ] = None """ Relayed authentication. Each inference request supplies the caller's OAuth token, which is forwarded to Anthropic. No Anthropic credential is @@ -69,24 +44,29 @@ class ModelProviderServiceConfigAnthropicProviderConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigAnthropicProviderConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigAnthropicProviderConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigAnthropicProviderConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigAnthropicProviderConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigAnthropicProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderDirectConfigParam] + direct: VariableOrOptional[ + ModelProviderServiceConfigAnthropicProviderDirectConfigParam + ] """ Direct authentication with an API key supplied in `direct.api_key.plaintext`. Required unless `relayed` is set. """ - relayed: VariableOrOptional[ModelProviderServiceConfigAnthropicProviderRelayedConfigParam] + relayed: VariableOrOptional[ + ModelProviderServiceConfigAnthropicProviderRelayedConfigParam + ] """ Relayed authentication. Each inference request supplies the caller's OAuth token, which is forwarded to Anthropic. No Anthropic credential is @@ -94,4 +74,7 @@ class ModelProviderServiceConfigAnthropicProviderConfigDict(TypedDict, total=Fal """ -ModelProviderServiceConfigAnthropicProviderConfigParam = ModelProviderServiceConfigAnthropicProviderConfigDict | ModelProviderServiceConfigAnthropicProviderConfig +ModelProviderServiceConfigAnthropicProviderConfigParam = ( + ModelProviderServiceConfigAnthropicProviderConfigDict + | ModelProviderServiceConfigAnthropicProviderConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_direct_config.py index b6a0705bb01..d03a79d596d 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_direct_config.py @@ -1,59 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigAnthropicProviderDirectConfig: """ Direct form of Anthropic provider config. """ + api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Anthropic API key. Required when creating the service. Supply the value @@ -61,15 +28,18 @@ class ModelProviderServiceConfigAnthropicProviderDirectConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigAnthropicProviderDirectConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigAnthropicProviderDirectConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigAnthropicProviderDirectConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigAnthropicProviderDirectConfigDict": + return _transform_to_json_value(self) # type:ignore -class ModelProviderServiceConfigAnthropicProviderDirectConfigDict(TypedDict, total=False): +class ModelProviderServiceConfigAnthropicProviderDirectConfigDict( + TypedDict, total=False +): """""" api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecretParam] @@ -79,4 +49,7 @@ class ModelProviderServiceConfigAnthropicProviderDirectConfigDict(TypedDict, tot """ -ModelProviderServiceConfigAnthropicProviderDirectConfigParam = ModelProviderServiceConfigAnthropicProviderDirectConfigDict | ModelProviderServiceConfigAnthropicProviderDirectConfig +ModelProviderServiceConfigAnthropicProviderDirectConfigParam = ( + ModelProviderServiceConfigAnthropicProviderDirectConfigDict + | ModelProviderServiceConfigAnthropicProviderDirectConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_relayed_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_relayed_config.py index 1f3c3a7c38e..7754ff58862 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_relayed_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_anthropic_provider_relayed_config.py @@ -1,54 +1,15 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigAnthropicProviderRelayedConfig: """ @@ -56,17 +17,24 @@ class ModelProviderServiceConfigAnthropicProviderRelayedConfig: the caller's OAuth token, which is forwarded to Anthropic. No Anthropic credential is stored. """ + @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigAnthropicProviderRelayedConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigAnthropicProviderRelayedConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigAnthropicProviderRelayedConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigAnthropicProviderRelayedConfigDict": + return _transform_to_json_value(self) # type:ignore -class ModelProviderServiceConfigAnthropicProviderRelayedConfigDict(TypedDict, total=False): +class ModelProviderServiceConfigAnthropicProviderRelayedConfigDict( + TypedDict, total=False +): """""" -ModelProviderServiceConfigAnthropicProviderRelayedConfigParam = ModelProviderServiceConfigAnthropicProviderRelayedConfigDict | ModelProviderServiceConfigAnthropicProviderRelayedConfig +ModelProviderServiceConfigAnthropicProviderRelayedConfigParam = ( + ModelProviderServiceConfigAnthropicProviderRelayedConfigDict + | ModelProviderServiceConfigAnthropicProviderRelayedConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_aws_access_key.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_aws_access_key.py index 2672d0e6fa8..6c57b6912d8 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_aws_access_key.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_aws_access_key.py @@ -1,59 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigAwsAccessKey: """ AWS access-key-pair auth for Amazon Bedrock: a SigV4-signing key pair. """ + access_key_id: VariableOrOptional[str] = None """ AWS access key ID. Required on Create when using access-key auth. Treated as @@ -61,7 +28,9 @@ class ModelProviderServiceConfigAwsAccessKey: scrubbed from audit logs. """ - secret_access_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None + secret_access_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = ( + None + ) """ AWS secret access key paired with `access_key_id`. Required when creating a service with access-key authentication. Supply the value in @@ -69,12 +38,11 @@ class ModelProviderServiceConfigAwsAccessKey: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigAwsAccessKeyDict') -> 'Self': + def from_dict(cls, value: "ModelProviderServiceConfigAwsAccessKeyDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigAwsAccessKeyDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigAwsAccessKeyDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigAwsAccessKeyDict(TypedDict, total=False): @@ -95,4 +63,6 @@ class ModelProviderServiceConfigAwsAccessKeyDict(TypedDict, total=False): """ -ModelProviderServiceConfigAwsAccessKeyParam = ModelProviderServiceConfigAwsAccessKeyDict | ModelProviderServiceConfigAwsAccessKey +ModelProviderServiceConfigAwsAccessKeyParam = ( + ModelProviderServiceConfigAwsAccessKeyDict | ModelProviderServiceConfigAwsAccessKey +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_config.py index 803e33f89c4..60726eeffed 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_config.py @@ -1,80 +1,55 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ( + ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, + ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigAzureOpenAiProviderConfig: """ Azure OpenAI provider configuration. """ - direct: VariableOrOptional[ModelProviderServiceConfigAzureOpenAiProviderDirectConfig] = None + + direct: VariableOrOptional[ + ModelProviderServiceConfigAzureOpenAiProviderDirectConfig + ] = None """ Azure OpenAI endpoint and authentication configuration. """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigAzureOpenAiProviderConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigAzureOpenAiProviderConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigAzureOpenAiProviderConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigAzureOpenAiProviderConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigAzureOpenAiProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam] + direct: VariableOrOptional[ + ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam + ] """ Azure OpenAI endpoint and authentication configuration. """ -ModelProviderServiceConfigAzureOpenAiProviderConfigParam = ModelProviderServiceConfigAzureOpenAiProviderConfigDict | ModelProviderServiceConfigAzureOpenAiProviderConfig +ModelProviderServiceConfigAzureOpenAiProviderConfigParam = ( + ModelProviderServiceConfigAzureOpenAiProviderConfigDict + | ModelProviderServiceConfigAzureOpenAiProviderConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_direct_config.py index fd5a80176c9..9ba84e99974 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_azure_open_ai_provider_direct_config.py @@ -1,54 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ( + ModelProviderServiceConfigEntraServicePrincipal, + ModelProviderServiceConfigEntraServicePrincipalParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( + ModelProviderServiceConfigServiceCredential, + ModelProviderServiceConfigServiceCredentialParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigAzureOpenAiProviderDirectConfig: """ @@ -65,6 +39,7 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfig: supported on Azure-hosted workspaces. Setting more than one mode is rejected. """ + api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Azure OpenAI API key. Supply the value in `api_key.plaintext`. Mutually @@ -78,14 +53,18 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfig: `https://myresource.openai.azure.com`. Required on Create. """ - entra_service_principal: VariableOrOptional[ModelProviderServiceConfigEntraServicePrincipal] = None + entra_service_principal: VariableOrOptional[ + ModelProviderServiceConfigEntraServicePrincipal + ] = None """ Entra ID service-principal authentication. Set `tenant_id`, `client_id`, and `client_secret.plaintext`. Mutually exclusive with `api_key` and `service_credential`. """ - service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredential] = None + service_credential: VariableOrOptional[ + ModelProviderServiceConfigServiceCredential + ] = None """ Reference to a Unity Catalog service credential authorizing Azure OpenAI requests. On Create, supply `service_credential.name` in the form @@ -96,15 +75,20 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict( + self, + ) -> "ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict": + return _transform_to_json_value(self) # type:ignore -class ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict(TypedDict, total=False): +class ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict( + TypedDict, total=False +): """""" api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecretParam] @@ -120,14 +104,18 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict(TypedDict, t `https://myresource.openai.azure.com`. Required on Create. """ - entra_service_principal: VariableOrOptional[ModelProviderServiceConfigEntraServicePrincipalParam] + entra_service_principal: VariableOrOptional[ + ModelProviderServiceConfigEntraServicePrincipalParam + ] """ Entra ID service-principal authentication. Set `tenant_id`, `client_id`, and `client_secret.plaintext`. Mutually exclusive with `api_key` and `service_credential`. """ - service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredentialParam] + service_credential: VariableOrOptional[ + ModelProviderServiceConfigServiceCredentialParam + ] """ Reference to a Unity Catalog service credential authorizing Azure OpenAI requests. On Create, supply `service_credential.name` in the form @@ -138,4 +126,7 @@ class ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict(TypedDict, t """ -ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam = ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict | ModelProviderServiceConfigAzureOpenAiProviderDirectConfig +ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam = ( + ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict + | ModelProviderServiceConfigAzureOpenAiProviderDirectConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_api_key_header_auth.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_api_key_header_auth.py index 12a4f31be79..78865d1c8f2 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_api_key_header_auth.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_api_key_header_auth.py @@ -1,54 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth: """ @@ -56,6 +22,7 @@ class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth: forwarded on outbound requests under a caller-chosen HTTP header, as `: `. """ + api_key_name: VariableOrOptional[str] = None """ HTTP header name that carries the API key on outbound requests (e.g., @@ -70,15 +37,18 @@ class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict": + return _transform_to_json_value(self) # type:ignore -class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict(TypedDict, total=False): +class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict( + TypedDict, total=False +): """""" api_key_name: VariableOrOptional[str] @@ -95,4 +65,7 @@ class ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict(TypedDict, to """ -ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam = ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict | ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth +ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam = ( + ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict + | ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_config.py index 92e5f98d4fd..25dcaca2838 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_config.py @@ -1,80 +1,55 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ( + ModelProviderServiceConfigCustomProviderDirectConfig, + ModelProviderServiceConfigCustomProviderDirectConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigCustomProviderConfig: """ Custom OpenAI-compatible provider configuration. """ - direct: VariableOrOptional[ModelProviderServiceConfigCustomProviderDirectConfig] = None + + direct: VariableOrOptional[ModelProviderServiceConfigCustomProviderDirectConfig] = ( + None + ) """ Endpoint and authentication configuration for the custom provider. """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigCustomProviderConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigCustomProviderConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigCustomProviderConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigCustomProviderConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigCustomProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ModelProviderServiceConfigCustomProviderDirectConfigParam] + direct: VariableOrOptional[ + ModelProviderServiceConfigCustomProviderDirectConfigParam + ] """ Endpoint and authentication configuration for the custom provider. """ -ModelProviderServiceConfigCustomProviderConfigParam = ModelProviderServiceConfigCustomProviderConfigDict | ModelProviderServiceConfigCustomProviderConfig +ModelProviderServiceConfigCustomProviderConfigParam = ( + ModelProviderServiceConfigCustomProviderConfigDict + | ModelProviderServiceConfigCustomProviderConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_direct_config.py index d16fe13463a..42e4d350a24 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_custom_provider_direct_config.py @@ -1,54 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ( + ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, + ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigCustomProviderDirectConfig: """ @@ -56,6 +26,7 @@ class ModelProviderServiceConfigCustomProviderDirectConfig: secret as an `Authorization` bearer token, or `header_auth` to forward it under a caller-chosen HTTP header. """ + api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Bearer token forwarded in the `Authorization` header. Supply the value @@ -68,7 +39,9 @@ class ModelProviderServiceConfigCustomProviderDirectConfig: `https://api.example.com/v1`). Required on Create. """ - header_auth: VariableOrOptional[ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth] = None + header_auth: VariableOrOptional[ + ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth + ] = None """ Header-based API-key auth: the secret is forwarded on outbound requests under a caller-chosen HTTP header rather than as an `Authorization` @@ -76,12 +49,13 @@ class ModelProviderServiceConfigCustomProviderDirectConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigCustomProviderDirectConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigCustomProviderDirectConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigCustomProviderDirectConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigCustomProviderDirectConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigCustomProviderDirectConfigDict(TypedDict, total=False): @@ -99,7 +73,9 @@ class ModelProviderServiceConfigCustomProviderDirectConfigDict(TypedDict, total= `https://api.example.com/v1`). Required on Create. """ - header_auth: VariableOrOptional[ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam] + header_auth: VariableOrOptional[ + ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam + ] """ Header-based API-key auth: the secret is forwarded on outbound requests under a caller-chosen HTTP header rather than as an `Authorization` @@ -107,4 +83,7 @@ class ModelProviderServiceConfigCustomProviderDirectConfigDict(TypedDict, total= """ -ModelProviderServiceConfigCustomProviderDirectConfigParam = ModelProviderServiceConfigCustomProviderDirectConfigDict | ModelProviderServiceConfigCustomProviderDirectConfig +ModelProviderServiceConfigCustomProviderDirectConfigParam = ( + ModelProviderServiceConfigCustomProviderDirectConfigDict + | ModelProviderServiceConfigCustomProviderDirectConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_entra_service_principal.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_entra_service_principal.py index 8924ee016ee..cf5fcefd569 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_entra_service_principal.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_entra_service_principal.py @@ -1,54 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigEntraServicePrincipal: """ @@ -57,6 +23,7 @@ class ModelProviderServiceConfigEntraServicePrincipal: it. AI Gateway exchanges these credentials for an Entra bearer token for requests to Azure OpenAI or Microsoft Foundry. """ + client_id: VariableOrOptional[str] = None """ Entra ID client (application) ID. Required on Create. @@ -73,12 +40,13 @@ class ModelProviderServiceConfigEntraServicePrincipal: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigEntraServicePrincipalDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigEntraServicePrincipalDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigEntraServicePrincipalDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigEntraServicePrincipalDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigEntraServicePrincipalDict(TypedDict, total=False): @@ -100,4 +68,7 @@ class ModelProviderServiceConfigEntraServicePrincipalDict(TypedDict, total=False """ -ModelProviderServiceConfigEntraServicePrincipalParam = ModelProviderServiceConfigEntraServicePrincipalDict | ModelProviderServiceConfigEntraServicePrincipal +ModelProviderServiceConfigEntraServicePrincipalParam = ( + ModelProviderServiceConfigEntraServicePrincipalDict + | ModelProviderServiceConfigEntraServicePrincipal +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_external_model_provider_type.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_external_model_provider_type.py index 2416cffa698..f3a79bd6cdf 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_external_model_provider_type.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_external_model_provider_type.py @@ -1,61 +1,44 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam +from typing import Literal class ModelProviderServiceConfigExternalModelProviderType(Enum): """ External LLM provider for an EXTERNAL_FOUNDATION_MODEL destination. """ + EXTERNAL_MODEL_PROVIDER_TYPE_OPENAI = "EXTERNAL_MODEL_PROVIDER_TYPE_OPENAI" - EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI = "EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI" + EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI = ( + "EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI" + ) EXTERNAL_MODEL_PROVIDER_TYPE_ANTHROPIC = "EXTERNAL_MODEL_PROVIDER_TYPE_ANTHROPIC" - EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK = "EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK" + EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK = ( + "EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK" + ) EXTERNAL_MODEL_PROVIDER_TYPE_CUSTOM = "EXTERNAL_MODEL_PROVIDER_TYPE_CUSTOM" - EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY = "EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY" - EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE = "EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE" - EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE = "EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE" - -ModelProviderServiceConfigExternalModelProviderTypeParam = Literal["EXTERNAL_MODEL_PROVIDER_TYPE_OPENAI", "EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI", "EXTERNAL_MODEL_PROVIDER_TYPE_ANTHROPIC", "EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK", "EXTERNAL_MODEL_PROVIDER_TYPE_CUSTOM", "EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY", "EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE", "EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE"] | ModelProviderServiceConfigExternalModelProviderType + EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY = ( + "EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY" + ) + EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE = ( + "EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE" + ) + EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE = ( + "EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE" + ) + + +ModelProviderServiceConfigExternalModelProviderTypeParam = ( + Literal[ + "EXTERNAL_MODEL_PROVIDER_TYPE_OPENAI", + "EXTERNAL_MODEL_PROVIDER_TYPE_AZURE_OPENAI", + "EXTERNAL_MODEL_PROVIDER_TYPE_ANTHROPIC", + "EXTERNAL_MODEL_PROVIDER_TYPE_AMAZON_BEDROCK", + "EXTERNAL_MODEL_PROVIDER_TYPE_CUSTOM", + "EXTERNAL_MODEL_PROVIDER_TYPE_MICROSOFT_FOUNDRY", + "EXTERNAL_MODEL_PROVIDER_TYPE_GEMINI_ENTERPRISE", + "EXTERNAL_MODEL_PROVIDER_TYPE_BEDROCK_MANTLE", + ] + | ModelProviderServiceConfigExternalModelProviderType +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_config.py index 9ea1598cc3b..38f005ad002 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_config.py @@ -1,80 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ( + ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, + ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigGeminiEnterpriseProviderConfig: """ Gemini Enterprise provider configuration. """ - direct: VariableOrOptional[ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig] = None + + direct: VariableOrOptional[ + ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig + ] = None """ Gemini Enterprise project, region, and authentication configuration. """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict": + return _transform_to_json_value(self) # type:ignore -class ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict(TypedDict, total=False): +class ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict( + TypedDict, total=False +): """""" - direct: VariableOrOptional[ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam] + direct: VariableOrOptional[ + ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam + ] """ Gemini Enterprise project, region, and authentication configuration. """ -ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam = ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict | ModelProviderServiceConfigGeminiEnterpriseProviderConfig +ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam = ( + ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict + | ModelProviderServiceConfigGeminiEnterpriseProviderConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_direct_config.py index 803dadf3019..157ef00adfd 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_gemini_enterprise_provider_direct_config.py @@ -1,60 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( + ModelProviderServiceConfigServiceCredential, + ModelProviderServiceConfigServiceCredentialParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig: """ Direct Gemini Enterprise provider configuration. An API key is required when creating the service. """ + api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Google Gemini Enterprise API key. Required when creating the service. @@ -72,7 +43,9 @@ class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig: Required on Create. """ - service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredential] = None + service_credential: VariableOrOptional[ + ModelProviderServiceConfigServiceCredential + ] = None """ Reference to a Unity Catalog service credential authorizing Gemini Enterprise requests. On Create, supply `service_credential.name` as @@ -84,15 +57,20 @@ class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict( + self, + ) -> "ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict": + return _transform_to_json_value(self) # type:ignore -class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict(TypedDict, total=False): +class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict( + TypedDict, total=False +): """""" api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecretParam] @@ -112,7 +90,9 @@ class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict(TypedDi Required on Create. """ - service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredentialParam] + service_credential: VariableOrOptional[ + ModelProviderServiceConfigServiceCredentialParam + ] """ Reference to a Unity Catalog service credential authorizing Gemini Enterprise requests. On Create, supply `service_credential.name` as @@ -124,4 +104,7 @@ class ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict(TypedDi """ -ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam = ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict | ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig +ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam = ( + ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict + | ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_config.py index 8d69e4636ee..c63fe203888 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_config.py @@ -1,80 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ( + ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, + ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigMicrosoftFoundryProviderConfig: """ Microsoft Foundry provider configuration. """ - direct: VariableOrOptional[ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig] = None + + direct: VariableOrOptional[ + ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig + ] = None """ Microsoft Foundry endpoint and authentication configuration. """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict": + return _transform_to_json_value(self) # type:ignore -class ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict(TypedDict, total=False): +class ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict( + TypedDict, total=False +): """""" - direct: VariableOrOptional[ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam] + direct: VariableOrOptional[ + ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam + ] """ Microsoft Foundry endpoint and authentication configuration. """ -ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam = ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict | ModelProviderServiceConfigMicrosoftFoundryProviderConfig +ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam = ( + ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict + | ModelProviderServiceConfigMicrosoftFoundryProviderConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_direct_config.py index 6f2d1945bb5..5caad765601 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_microsoft_foundry_provider_direct_config.py @@ -1,59 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ( + ModelProviderServiceConfigEntraServicePrincipal, + ModelProviderServiceConfigEntraServicePrincipalParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretParam, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ( + ModelProviderServiceConfigServiceCredential, + ModelProviderServiceConfigServiceCredentialParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig: """ Direct form of Microsoft Foundry provider config. - + Authentication is one of three mutually exclusive modes, exactly one of which must be supplied on Create: - API key: set `api_key`, leave `entra_service_principal` and @@ -69,6 +43,7 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig: supported on Azure-hosted workspaces. Setting more than one mode is rejected. """ + api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ Microsoft Foundry API key. Supply the value in `api_key.plaintext`. @@ -81,14 +56,18 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig: Microsoft Foundry endpoint URL. Required on Create. """ - entra_service_principal: VariableOrOptional[ModelProviderServiceConfigEntraServicePrincipal] = None + entra_service_principal: VariableOrOptional[ + ModelProviderServiceConfigEntraServicePrincipal + ] = None """ Entra ID service-principal authentication. Set `tenant_id`, `client_id`, and `client_secret.plaintext`. Mutually exclusive with `api_key` and `service_credential`. """ - service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredential] = None + service_credential: VariableOrOptional[ + ModelProviderServiceConfigServiceCredential + ] = None """ Reference to a Unity Catalog service credential authorizing Microsoft Foundry requests. On Create, supply `service_credential.name` in the form @@ -99,15 +78,20 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict( + self, + ) -> "ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict": + return _transform_to_json_value(self) # type:ignore -class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict(TypedDict, total=False): +class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict( + TypedDict, total=False +): """""" api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecretParam] @@ -122,14 +106,18 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict(TypedDi Microsoft Foundry endpoint URL. Required on Create. """ - entra_service_principal: VariableOrOptional[ModelProviderServiceConfigEntraServicePrincipalParam] + entra_service_principal: VariableOrOptional[ + ModelProviderServiceConfigEntraServicePrincipalParam + ] """ Entra ID service-principal authentication. Set `tenant_id`, `client_id`, and `client_secret.plaintext`. Mutually exclusive with `api_key` and `service_credential`. """ - service_credential: VariableOrOptional[ModelProviderServiceConfigServiceCredentialParam] + service_credential: VariableOrOptional[ + ModelProviderServiceConfigServiceCredentialParam + ] """ Reference to a Unity Catalog service credential authorizing Microsoft Foundry requests. On Create, supply `service_credential.name` in the form @@ -140,4 +128,7 @@ class ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict(TypedDi """ -ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam = ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict | ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig +ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam = ( + ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict + | ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_model_target_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_model_target_config.py index 6d91404cc1e..671c9eb4490 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_model_target_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_model_target_config.py @@ -1,60 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigModelTargetConfig: """ Model target configuration shared by model provider service targets and external model destinations on model services. """ + model: VariableOr[str] """ Provider-side model identifier, such as `gpt-5` or `claude-opus-4-7`. @@ -82,12 +49,13 @@ class ModelProviderServiceConfigModelTargetConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigModelTargetConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigModelTargetConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigModelTargetConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigModelTargetConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigModelTargetConfigDict(TypedDict, total=False): @@ -120,4 +88,7 @@ class ModelProviderServiceConfigModelTargetConfigDict(TypedDict, total=False): """ -ModelProviderServiceConfigModelTargetConfigParam = ModelProviderServiceConfigModelTargetConfigDict | ModelProviderServiceConfigModelTargetConfig +ModelProviderServiceConfigModelTargetConfigParam = ( + ModelProviderServiceConfigModelTargetConfigDict + | ModelProviderServiceConfigModelTargetConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_config.py index 598b38067f6..70200550381 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_config.py @@ -1,80 +1,55 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ( + ModelProviderServiceConfigOpenAiProviderDirectConfig, + ModelProviderServiceConfigOpenAiProviderDirectConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigOpenAiProviderConfig: """ OpenAI provider configuration. """ - direct: VariableOrOptional[ModelProviderServiceConfigOpenAiProviderDirectConfig] = None + + direct: VariableOrOptional[ModelProviderServiceConfigOpenAiProviderDirectConfig] = ( + None + ) """ OpenAI configuration with an API key supplied in the request. """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigOpenAiProviderConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigOpenAiProviderConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigOpenAiProviderConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigOpenAiProviderConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigOpenAiProviderConfigDict(TypedDict, total=False): """""" - direct: VariableOrOptional[ModelProviderServiceConfigOpenAiProviderDirectConfigParam] + direct: VariableOrOptional[ + ModelProviderServiceConfigOpenAiProviderDirectConfigParam + ] """ OpenAI configuration with an API key supplied in the request. """ -ModelProviderServiceConfigOpenAiProviderConfigParam = ModelProviderServiceConfigOpenAiProviderConfigDict | ModelProviderServiceConfigOpenAiProviderConfig +ModelProviderServiceConfigOpenAiProviderConfigParam = ( + ModelProviderServiceConfigOpenAiProviderConfigDict + | ModelProviderServiceConfigOpenAiProviderConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_direct_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_direct_config.py index 148137f7f90..128ce50e14e 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_direct_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_open_ai_provider_direct_config.py @@ -1,59 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ( + ModelProviderServiceConfigProviderSecret, + ModelProviderServiceConfigProviderSecretParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigOpenAiProviderDirectConfig: """ Direct (inline-credentials) form of the OpenAI provider config. """ + api_key: VariableOrOptional[ModelProviderServiceConfigProviderSecret] = None """ OpenAI API key. Required when creating the service. Supply the value in @@ -73,12 +40,13 @@ class ModelProviderServiceConfigOpenAiProviderDirectConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigOpenAiProviderDirectConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigOpenAiProviderDirectConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigOpenAiProviderDirectConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigOpenAiProviderDirectConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigOpenAiProviderDirectConfigDict(TypedDict, total=False): @@ -103,4 +71,7 @@ class ModelProviderServiceConfigOpenAiProviderDirectConfigDict(TypedDict, total= """ -ModelProviderServiceConfigOpenAiProviderDirectConfigParam = ModelProviderServiceConfigOpenAiProviderDirectConfigDict | ModelProviderServiceConfigOpenAiProviderDirectConfig +ModelProviderServiceConfigOpenAiProviderDirectConfigParam = ( + ModelProviderServiceConfigOpenAiProviderDirectConfigDict + | ModelProviderServiceConfigOpenAiProviderDirectConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_pricing_config.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_pricing_config.py index bbbef776c6e..88e0d02484c 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_pricing_config.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_pricing_config.py @@ -1,60 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigProviderPricingConfig: """ Pricing adjustments applied to this provider service's external-model spend estimates. """ + default_discount_basis_points: VariableOrOptional[int] = None """ Provider-wide adjustment to external-model spend estimates in basis points: @@ -64,12 +27,13 @@ class ModelProviderServiceConfigProviderPricingConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigProviderPricingConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigProviderPricingConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigProviderPricingConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigProviderPricingConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigProviderPricingConfigDict(TypedDict, total=False): @@ -84,4 +48,7 @@ class ModelProviderServiceConfigProviderPricingConfigDict(TypedDict, total=False """ -ModelProviderServiceConfigProviderPricingConfigParam = ModelProviderServiceConfigProviderPricingConfigDict | ModelProviderServiceConfigProviderPricingConfig +ModelProviderServiceConfigProviderPricingConfigParam = ( + ModelProviderServiceConfigProviderPricingConfigDict + | ModelProviderServiceConfigProviderPricingConfig +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_secret.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_secret.py index 6839557023a..fd6cef72cf8 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_secret.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_provider_secret.py @@ -1,54 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ( + ModelProviderServiceConfigSecretReference, + ModelProviderServiceConfigSecretReferenceParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigProviderSecret: """ @@ -57,6 +23,7 @@ class ModelProviderServiceConfigProviderSecret: encrypted. The `plaintext` field is `INPUT_ONLY` and never round-trips on reads. """ + plaintext: VariableOrOptional[str] = None """ Inline plaintext credential. INPUT_ONLY: the value never round-trips on @@ -64,7 +31,9 @@ class ModelProviderServiceConfigProviderSecret: object remains present to indicate that a secret is configured. """ - secret_reference: VariableOrOptional[ModelProviderServiceConfigSecretReference] = None + secret_reference: VariableOrOptional[ModelProviderServiceConfigSecretReference] = ( + None + ) """ Reference to a customer-owned UC Secret that carries this secret value. The value is read at invoke time under the model provider service @@ -75,12 +44,11 @@ class ModelProviderServiceConfigProviderSecret: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigProviderSecretDict') -> 'Self': + def from_dict(cls, value: "ModelProviderServiceConfigProviderSecretDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigProviderSecretDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigProviderSecretDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigProviderSecretDict(TypedDict, total=False): @@ -104,4 +72,7 @@ class ModelProviderServiceConfigProviderSecretDict(TypedDict, total=False): """ -ModelProviderServiceConfigProviderSecretParam = ModelProviderServiceConfigProviderSecretDict | ModelProviderServiceConfigProviderSecret +ModelProviderServiceConfigProviderSecretParam = ( + ModelProviderServiceConfigProviderSecretDict + | ModelProviderServiceConfigProviderSecret +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_secret_reference.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_secret_reference.py index f4de98a2c4d..9c7e95309ff 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_secret_reference.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_secret_reference.py @@ -1,60 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigSecretReference: """ Reference to a customer-owned UC Secret backing a secret-bearing provider field, in the `ProviderSecret.secret_reference` arm. """ + name: VariableOr[str] """ Resource name of the bound UC Secret, in the form @@ -63,12 +26,13 @@ class ModelProviderServiceConfigSecretReference: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigSecretReferenceDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigSecretReferenceDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigSecretReferenceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigSecretReferenceDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigSecretReferenceDict(TypedDict, total=False): @@ -82,4 +46,7 @@ class ModelProviderServiceConfigSecretReferenceDict(TypedDict, total=False): """ -ModelProviderServiceConfigSecretReferenceParam = ModelProviderServiceConfigSecretReferenceDict | ModelProviderServiceConfigSecretReference +ModelProviderServiceConfigSecretReferenceParam = ( + ModelProviderServiceConfigSecretReferenceDict + | ModelProviderServiceConfigSecretReference +) diff --git a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_service_credential.py b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_service_credential.py index db571e166cd..12753ea2865 100644 --- a/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_service_credential.py +++ b/python/databricks/bundles/model_provider_services/_models/model_provider_service_config_service_credential.py @@ -1,60 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigServiceCredential: """ The customer-owned Unity Catalog service credential a model provider service uses to authenticate to its provider, referenced by name. """ + name: VariableOr[str] """ Resource name of the bound Unity Catalog service credential, in the form @@ -64,12 +27,13 @@ class ModelProviderServiceConfigServiceCredential: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigServiceCredentialDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigServiceCredentialDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigServiceCredentialDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigServiceCredentialDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigServiceCredentialDict(TypedDict, total=False): @@ -84,4 +48,7 @@ class ModelProviderServiceConfigServiceCredentialDict(TypedDict, total=False): """ -ModelProviderServiceConfigServiceCredentialParam = ModelProviderServiceConfigServiceCredentialDict | ModelProviderServiceConfigServiceCredential +ModelProviderServiceConfigServiceCredentialParam = ( + ModelProviderServiceConfigServiceCredentialDict + | ModelProviderServiceConfigServiceCredential +) diff --git a/python/databricks/bundles/model_provider_services/_models/privilege.py b/python/databricks/bundles/model_provider_services/_models/privilege.py index da63f51e84a..df45211a76a 100644 --- a/python/databricks/bundles/model_provider_services/_models/privilege.py +++ b/python/databricks/bundles/model_provider_services/_models/privilege.py @@ -1,48 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam +from typing import Literal class Privilege(Enum): @@ -99,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/model_provider_services/_models/privilege_assignment.py b/python/databricks/bundles/model_provider_services/_models/privilege_assignment.py index 4512b6d63e8..810d4ebb627 100644 --- a/python/databricks/bundles/model_provider_services/_models/privilege_assignment.py +++ b/python/databricks/bundles/model_provider_services/_models/privilege_assignment.py @@ -1,54 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.model_provider_services._models.privilege import ( + Privilege, + PrivilegeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -65,12 +31,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/rate_limit.py b/python/databricks/bundles/model_provider_services/_models/rate_limit.py index 8b75b82fe94..5bbb533157a 100644 --- a/python/databricks/bundles/model_provider_services/_models/rate_limit.py +++ b/python/databricks/bundles/model_provider_services/_models/rate_limit.py @@ -1,54 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import ( + RateLimitRateLimitKey, + RateLimitRateLimitKeyParam, +) +from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import ( + RateLimitRateLimitRenewalPeriod, + RateLimitRateLimitRenewalPeriodParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RateLimit: """ @@ -56,6 +26,7 @@ class RateLimit: unset to impose no limit on that dimension; set a value to cap that dimension within the renewal period. """ + key: VariableOr[RateLimitRateLimitKey] """ Scope of the rate limit. Depending on this value, the limit applies to a @@ -87,12 +58,11 @@ class RateLimit: """ @classmethod - def from_dict(cls, value: 'RateLimitDict') -> 'Self': + def from_dict(cls, value: "RateLimitDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RateLimitDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RateLimitDict": + return _transform_to_json_value(self) # type:ignore class RateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_key.py b/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_key.py index 840e8e25e56..fb907da810f 100644 --- a/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_key.py +++ b/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_key.py @@ -1,58 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam +from typing import Literal class RateLimitRateLimitKey(Enum): """ Scope key for a rate limit. """ + RATE_LIMIT_KEY_USER = "RATE_LIMIT_KEY_USER" RATE_LIMIT_KEY_USER_GROUP = "RATE_LIMIT_KEY_USER_GROUP" RATE_LIMIT_KEY_SERVICE_PRINCIPAL = "RATE_LIMIT_KEY_SERVICE_PRINCIPAL" RATE_LIMIT_KEY_SERVICE = "RATE_LIMIT_KEY_SERVICE" RATE_LIMIT_KEY_USER_DEFAULT = "RATE_LIMIT_KEY_USER_DEFAULT" -RateLimitRateLimitKeyParam = Literal["RATE_LIMIT_KEY_USER", "RATE_LIMIT_KEY_USER_GROUP", "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", "RATE_LIMIT_KEY_SERVICE", "RATE_LIMIT_KEY_USER_DEFAULT"] | RateLimitRateLimitKey + +RateLimitRateLimitKeyParam = ( + Literal[ + "RATE_LIMIT_KEY_USER", + "RATE_LIMIT_KEY_USER_GROUP", + "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", + "RATE_LIMIT_KEY_SERVICE", + "RATE_LIMIT_KEY_USER_DEFAULT", + ] + | RateLimitRateLimitKey +) diff --git a/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_renewal_period.py b/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_renewal_period.py index b0ff8ae0779..58636c784c0 100644 --- a/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_renewal_period.py +++ b/python/databricks/bundles/model_provider_services/_models/rate_limit_rate_limit_renewal_period.py @@ -1,55 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_provider_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig, ModelProviderServiceConfigDict, ModelProviderServiceConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_config import ModelProviderServiceConfigAmazonBedrockProviderConfig, ModelProviderServiceConfigAmazonBedrockProviderConfigDict, ModelProviderServiceConfigAmazonBedrockProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_amazon_bedrock_provider_direct_config import ModelProviderServiceConfigAmazonBedrockProviderDirectConfig, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigDict, ModelProviderServiceConfigAmazonBedrockProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_config import ModelProviderServiceConfigAnthropicProviderConfig, ModelProviderServiceConfigAnthropicProviderConfigDict, ModelProviderServiceConfigAnthropicProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_direct_config import ModelProviderServiceConfigAnthropicProviderDirectConfig, ModelProviderServiceConfigAnthropicProviderDirectConfigDict, ModelProviderServiceConfigAnthropicProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_anthropic_provider_relayed_config import ModelProviderServiceConfigAnthropicProviderRelayedConfig, ModelProviderServiceConfigAnthropicProviderRelayedConfigDict, ModelProviderServiceConfigAnthropicProviderRelayedConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_aws_access_key import ModelProviderServiceConfigAwsAccessKey, ModelProviderServiceConfigAwsAccessKeyDict, ModelProviderServiceConfigAwsAccessKeyParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_config import ModelProviderServiceConfigAzureOpenAiProviderConfig, ModelProviderServiceConfigAzureOpenAiProviderConfigDict, ModelProviderServiceConfigAzureOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_azure_open_ai_provider_direct_config import ModelProviderServiceConfigAzureOpenAiProviderDirectConfig, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigDict, ModelProviderServiceConfigAzureOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_api_key_header_auth import ModelProviderServiceConfigCustomProviderApiKeyHeaderAuth, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthDict, ModelProviderServiceConfigCustomProviderApiKeyHeaderAuthParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_config import ModelProviderServiceConfigCustomProviderConfig, ModelProviderServiceConfigCustomProviderConfigDict, ModelProviderServiceConfigCustomProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_custom_provider_direct_config import ModelProviderServiceConfigCustomProviderDirectConfig, ModelProviderServiceConfigCustomProviderDirectConfigDict, ModelProviderServiceConfigCustomProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_entra_service_principal import ModelProviderServiceConfigEntraServicePrincipal, ModelProviderServiceConfigEntraServicePrincipalDict, ModelProviderServiceConfigEntraServicePrincipalParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_config import ModelProviderServiceConfigGeminiEnterpriseProviderConfig, ModelProviderServiceConfigGeminiEnterpriseProviderConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_gemini_enterprise_provider_direct_config import ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfig, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigDict, ModelProviderServiceConfigGeminiEnterpriseProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_config import ModelProviderServiceConfigMicrosoftFoundryProviderConfig, ModelProviderServiceConfigMicrosoftFoundryProviderConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_microsoft_foundry_provider_direct_config import ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfig, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigDict, ModelProviderServiceConfigMicrosoftFoundryProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_config import ModelProviderServiceConfigOpenAiProviderConfig, ModelProviderServiceConfigOpenAiProviderConfigDict, ModelProviderServiceConfigOpenAiProviderConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_open_ai_provider_direct_config import ModelProviderServiceConfigOpenAiProviderDirectConfig, ModelProviderServiceConfigOpenAiProviderDirectConfigDict, ModelProviderServiceConfigOpenAiProviderDirectConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_pricing_config import ModelProviderServiceConfigProviderPricingConfig, ModelProviderServiceConfigProviderPricingConfigDict, ModelProviderServiceConfigProviderPricingConfigParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_provider_secret import ModelProviderServiceConfigProviderSecret, ModelProviderServiceConfigProviderSecretDict, ModelProviderServiceConfigProviderSecretParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_secret_reference import ModelProviderServiceConfigSecretReference, ModelProviderServiceConfigSecretReferenceDict, ModelProviderServiceConfigSecretReferenceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_service_credential import ModelProviderServiceConfigServiceCredential, ModelProviderServiceConfigServiceCredentialDict, ModelProviderServiceConfigServiceCredentialParam -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_provider_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService, ModelProviderServiceDict, ModelProviderServiceParam -from databricks.bundles.model_provider_services._models.model_provider_service_config_external_model_provider_type import ModelProviderServiceConfigExternalModelProviderType, ModelProviderServiceConfigExternalModelProviderTypeParam - -from databricks.bundles.model_provider_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_provider_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam +from typing import Literal class RateLimitRateLimitRenewalPeriod(Enum): """ Renewal period for a rate limit. """ + RATE_LIMIT_RENEWAL_PERIOD_MINUTE = "RATE_LIMIT_RENEWAL_PERIOD_MINUTE" RATE_LIMIT_RENEWAL_PERIOD_HOUR = "RATE_LIMIT_RENEWAL_PERIOD_HOUR" -RateLimitRateLimitRenewalPeriodParam = Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] | RateLimitRateLimitRenewalPeriod + +RateLimitRateLimitRenewalPeriodParam = ( + Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] + | RateLimitRateLimitRenewalPeriod +) diff --git a/python/databricks/bundles/model_services/__init__.py b/python/databricks/bundles/model_services/__init__.py index 86cdfa8031b..9e10e6df19c 100644 --- a/python/databricks/bundles/model_services/__init__.py +++ b/python/databricks/bundles/model_services/__init__.py @@ -1,74 +1,134 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'InferenceTableConfig', - 'InferenceTableConfigDict', - 'InferenceTableConfigParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'ModelProviderServiceConfigModelTargetConfig', - 'ModelProviderServiceConfigModelTargetConfigDict', - 'ModelProviderServiceConfigModelTargetConfigParam', - 'ModelService', - 'ModelServiceConfig', - 'ModelServiceConfigDestinationConfig', - 'ModelServiceConfigDestinationConfigDestinationType', - 'ModelServiceConfigDestinationConfigDestinationTypeParam', - 'ModelServiceConfigDestinationConfigDict', - 'ModelServiceConfigDestinationConfigParam', - 'ModelServiceConfigDict', - 'ModelServiceConfigExternalModelConfig', - 'ModelServiceConfigExternalModelConfigDict', - 'ModelServiceConfigExternalModelConfigParam', - 'ModelServiceConfigFallbackConfig', - 'ModelServiceConfigFallbackConfigDict', - 'ModelServiceConfigFallbackConfigParam', - 'ModelServiceConfigParam', - 'ModelServiceConfigPayPerTokenConfig', - 'ModelServiceConfigPayPerTokenConfigDict', - 'ModelServiceConfigPayPerTokenConfigParam', - 'ModelServiceConfigProvisionedThroughputConfig', - 'ModelServiceConfigProvisionedThroughputConfigDict', - 'ModelServiceConfigProvisionedThroughputConfigParam', - 'ModelServiceConfigRoutingConfig', - 'ModelServiceConfigRoutingConfigDict', - 'ModelServiceConfigRoutingConfigParam', - 'ModelServiceDict', - 'ModelServiceParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'RateLimit', - 'RateLimitDict', - 'RateLimitParam', - 'RateLimitRateLimitKey', - 'RateLimitRateLimitKeyParam', - 'RateLimitRateLimitRenewalPeriod', - 'RateLimitRateLimitRenewalPeriodParam', + "InferenceTableConfig", + "InferenceTableConfigDict", + "InferenceTableConfigParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "ModelProviderServiceConfigModelTargetConfig", + "ModelProviderServiceConfigModelTargetConfigDict", + "ModelProviderServiceConfigModelTargetConfigParam", + "ModelService", + "ModelServiceConfig", + "ModelServiceConfigDestinationConfig", + "ModelServiceConfigDestinationConfigDestinationType", + "ModelServiceConfigDestinationConfigDestinationTypeParam", + "ModelServiceConfigDestinationConfigDict", + "ModelServiceConfigDestinationConfigParam", + "ModelServiceConfigDict", + "ModelServiceConfigExternalModelConfig", + "ModelServiceConfigExternalModelConfigDict", + "ModelServiceConfigExternalModelConfigParam", + "ModelServiceConfigFallbackConfig", + "ModelServiceConfigFallbackConfigDict", + "ModelServiceConfigFallbackConfigParam", + "ModelServiceConfigParam", + "ModelServiceConfigPayPerTokenConfig", + "ModelServiceConfigPayPerTokenConfigDict", + "ModelServiceConfigPayPerTokenConfigParam", + "ModelServiceConfigProvisionedThroughputConfig", + "ModelServiceConfigProvisionedThroughputConfigDict", + "ModelServiceConfigProvisionedThroughputConfigParam", + "ModelServiceConfigRoutingConfig", + "ModelServiceConfigRoutingConfigDict", + "ModelServiceConfigRoutingConfigParam", + "ModelServiceDict", + "ModelServiceParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", + "RateLimit", + "RateLimitDict", + "RateLimitParam", + "RateLimitRateLimitKey", + "RateLimitRateLimitKeyParam", + "RateLimitRateLimitRenewalPeriod", + "RateLimitRateLimitRenewalPeriodParam", ] -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.model_services._models.inference_table_config import ( + InferenceTableConfig, + InferenceTableConfigDict, + InferenceTableConfigParam, +) +from databricks.bundles.model_services._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ( + ModelProviderServiceConfigModelTargetConfig, + ModelProviderServiceConfigModelTargetConfigDict, + ModelProviderServiceConfigModelTargetConfigParam, +) +from databricks.bundles.model_services._models.model_service import ( + ModelService, + ModelServiceDict, + ModelServiceParam, +) +from databricks.bundles.model_services._models.model_service_config import ( + ModelServiceConfig, + ModelServiceConfigDict, + ModelServiceConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_destination_config import ( + ModelServiceConfigDestinationConfig, + ModelServiceConfigDestinationConfigDict, + ModelServiceConfigDestinationConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ( + ModelServiceConfigDestinationConfigDestinationType, + ModelServiceConfigDestinationConfigDestinationTypeParam, +) +from databricks.bundles.model_services._models.model_service_config_external_model_config import ( + ModelServiceConfigExternalModelConfig, + ModelServiceConfigExternalModelConfigDict, + ModelServiceConfigExternalModelConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_fallback_config import ( + ModelServiceConfigFallbackConfig, + ModelServiceConfigFallbackConfigDict, + ModelServiceConfigFallbackConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ( + ModelServiceConfigPayPerTokenConfig, + ModelServiceConfigPayPerTokenConfigDict, + ModelServiceConfigPayPerTokenConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ( + ModelServiceConfigProvisionedThroughputConfig, + ModelServiceConfigProvisionedThroughputConfigDict, + ModelServiceConfigProvisionedThroughputConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_routing_config import ( + ModelServiceConfigRoutingConfig, + ModelServiceConfigRoutingConfigDict, + ModelServiceConfigRoutingConfigParam, +) +from databricks.bundles.model_services._models.privilege import ( + Privilege, + PrivilegeParam, +) +from databricks.bundles.model_services._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) +from databricks.bundles.model_services._models.rate_limit import ( + RateLimit, + RateLimitDict, + RateLimitParam, +) +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import ( + RateLimitRateLimitKey, + RateLimitRateLimitKeyParam, +) +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import ( + RateLimitRateLimitRenewalPeriod, + RateLimitRateLimitRenewalPeriodParam, +) diff --git a/python/databricks/bundles/model_services/_models/inference_table_config.py b/python/databricks/bundles/model_services/_models/inference_table_config.py index 159ec1af624..45165416e17 100644 --- a/python/databricks/bundles/model_services/_models/inference_table_config.py +++ b/python/databricks/bundles/model_services/_models/inference_table_config.py @@ -1,38 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InferenceTableConfig: """ @@ -40,6 +18,7 @@ class InferenceTableConfig: inference table. When this configuration is present, payload logging is enabled by default. """ + parent: VariableOr[str] """ Parent Unity Catalog schema where the inference table is created, in the @@ -57,12 +36,11 @@ class InferenceTableConfig: """ @classmethod - def from_dict(cls, value: 'InferenceTableConfigDict') -> 'Self': + def from_dict(cls, value: "InferenceTableConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InferenceTableConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InferenceTableConfigDict": + return _transform_to_json_value(self) # type:ignore class InferenceTableConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/lifecycle.py b/python/databricks/bundles/model_services/_models/lifecycle.py index f84b9e7e08b..697776a198e 100644 --- a/python/databricks/bundles/model_services/_models/lifecycle.py +++ b/python/databricks/bundles/model_services/_models/lifecycle.py @@ -1,38 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -43,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/model_provider_service_config_model_target_config.py b/python/databricks/bundles/model_services/_models/model_provider_service_config_model_target_config.py index 50a9d818ab6..671c9eb4490 100644 --- a/python/databricks/bundles/model_services/_models/model_provider_service_config_model_target_config.py +++ b/python/databricks/bundles/model_services/_models/model_provider_service_config_model_target_config.py @@ -1,44 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelProviderServiceConfigModelTargetConfig: """ Model target configuration shared by model provider service targets and external model destinations on model services. """ + model: VariableOr[str] """ Provider-side model identifier, such as `gpt-5` or `claude-opus-4-7`. @@ -66,12 +49,13 @@ class ModelProviderServiceConfigModelTargetConfig: """ @classmethod - def from_dict(cls, value: 'ModelProviderServiceConfigModelTargetConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelProviderServiceConfigModelTargetConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelProviderServiceConfigModelTargetConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelProviderServiceConfigModelTargetConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelProviderServiceConfigModelTargetConfigDict(TypedDict, total=False): @@ -104,4 +88,7 @@ class ModelProviderServiceConfigModelTargetConfigDict(TypedDict, total=False): """ -ModelProviderServiceConfigModelTargetConfigParam = ModelProviderServiceConfigModelTargetConfigDict | ModelProviderServiceConfigModelTargetConfig +ModelProviderServiceConfigModelTargetConfigParam = ( + ModelProviderServiceConfigModelTargetConfigDict + | ModelProviderServiceConfigModelTargetConfig +) diff --git a/python/databricks/bundles/model_services/_models/model_service.py b/python/databricks/bundles/model_services/_models/model_service.py index f234c7746fd..88f9d8cdeb7 100644 --- a/python/databricks/bundles/model_services/_models/model_service.py +++ b/python/databricks/bundles/model_services/_models/model_service.py @@ -1,38 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.model_services._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.model_services._models.model_service_config import ( + ModelServiceConfig, + ModelServiceConfigParam, +) +from databricks.bundles.model_services._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelService(Resource): """""" @@ -50,12 +45,11 @@ class ModelService(Resource): lifecycle: VariableOrOptional[Lifecycle] = None @classmethod - def from_dict(cls, value: 'ModelServiceDict') -> 'Self': + def from_dict(cls, value: "ModelServiceDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServiceDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServiceDict": + return _transform_to_json_value(self) # type:ignore class ModelServiceDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/model_service_config.py b/python/databricks/bundles/model_services/_models/model_service_config.py index 50c7c68458c..88fb652d3ce 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config.py @@ -1,43 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.model_services._models.inference_table_config import ( + InferenceTableConfig, + InferenceTableConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_routing_config import ( + ModelServiceConfigRoutingConfig, + ModelServiceConfigRoutingConfigParam, +) +from databricks.bundles.model_services._models.rate_limit import ( + RateLimit, + RateLimitParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelServiceConfig: """ Operational configuration wrapped around the ModelService resource. """ + inference_table: VariableOrOptional[InferenceTableConfig] = None """ Inference table configuration for payload logging. @@ -54,12 +45,11 @@ class ModelServiceConfig: """ @classmethod - def from_dict(cls, value: 'ModelServiceConfigDict') -> 'Self': + def from_dict(cls, value: "ModelServiceConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServiceConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServiceConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelServiceConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/model_service_config_destination_config.py b/python/databricks/bundles/model_services/_models/model_service_config_destination_config.py index a18b409d074..e4277476e5f 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_destination_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_destination_config.py @@ -1,38 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ( + ModelServiceConfigDestinationConfigDestinationType, + ModelServiceConfigDestinationConfigDestinationTypeParam, +) +from databricks.bundles.model_services._models.model_service_config_external_model_config import ( + ModelServiceConfigExternalModelConfig, + ModelServiceConfigExternalModelConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ( + ModelServiceConfigPayPerTokenConfig, + ModelServiceConfigPayPerTokenConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ( + ModelServiceConfigProvisionedThroughputConfig, + ModelServiceConfigProvisionedThroughputConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelServiceConfigDestinationConfig: """ @@ -40,6 +34,7 @@ class ModelServiceConfigDestinationConfig: per-type configs inside `type_config` must be set, and it must match `destination_type`. """ + destination_type: VariableOr[ModelServiceConfigDestinationConfigDestinationType] """ Backing-model category. Provide the matching type-specific configuration @@ -51,7 +46,9 @@ class ModelServiceConfigDestinationConfig: User-facing label for this destination, used in routing references. """ - external_model_config: VariableOrOptional[ModelServiceConfigExternalModelConfig] = None + external_model_config: VariableOrOptional[ModelServiceConfigExternalModelConfig] = ( + None + ) """ Configuration for an external model reached through a model provider service. """ @@ -61,7 +58,9 @@ class ModelServiceConfigDestinationConfig: Configuration for a pay-per-token Databricks foundation model. """ - provisioned_throughput_config: VariableOrOptional[ModelServiceConfigProvisionedThroughputConfig] = None + provisioned_throughput_config: VariableOrOptional[ + ModelServiceConfigProvisionedThroughputConfig + ] = None """ Configuration for a provisioned-throughput Databricks foundation model. """ @@ -75,18 +74,19 @@ class ModelServiceConfigDestinationConfig: """ @classmethod - def from_dict(cls, value: 'ModelServiceConfigDestinationConfigDict') -> 'Self': + def from_dict(cls, value: "ModelServiceConfigDestinationConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServiceConfigDestinationConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServiceConfigDestinationConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelServiceConfigDestinationConfigDict(TypedDict, total=False): """""" - destination_type: VariableOr[ModelServiceConfigDestinationConfigDestinationTypeParam] + destination_type: VariableOr[ + ModelServiceConfigDestinationConfigDestinationTypeParam + ] """ Backing-model category. Provide the matching type-specific configuration and leave the other type-specific configurations unset. @@ -97,7 +97,9 @@ class ModelServiceConfigDestinationConfigDict(TypedDict, total=False): User-facing label for this destination, used in routing references. """ - external_model_config: VariableOrOptional[ModelServiceConfigExternalModelConfigParam] + external_model_config: VariableOrOptional[ + ModelServiceConfigExternalModelConfigParam + ] """ Configuration for an external model reached through a model provider service. """ @@ -107,7 +109,9 @@ class ModelServiceConfigDestinationConfigDict(TypedDict, total=False): Configuration for a pay-per-token Databricks foundation model. """ - provisioned_throughput_config: VariableOrOptional[ModelServiceConfigProvisionedThroughputConfigParam] + provisioned_throughput_config: VariableOrOptional[ + ModelServiceConfigProvisionedThroughputConfigParam + ] """ Configuration for a provisioned-throughput Databricks foundation model. """ @@ -121,4 +125,6 @@ class ModelServiceConfigDestinationConfigDict(TypedDict, total=False): """ -ModelServiceConfigDestinationConfigParam = ModelServiceConfigDestinationConfigDict | ModelServiceConfigDestinationConfig +ModelServiceConfigDestinationConfigParam = ( + ModelServiceConfigDestinationConfigDict | ModelServiceConfigDestinationConfig +) diff --git a/python/databricks/bundles/model_services/_models/model_service_config_destination_config_destination_type.py b/python/databricks/bundles/model_services/_models/model_service_config_destination_config_destination_type.py index 96aacf91f80..a98c2ca8161 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_destination_config_destination_type.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_destination_config_destination_type.py @@ -1,40 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam +from typing import Literal class ModelServiceConfigDestinationConfigDestinationType(Enum): """ Backing-model category for a model service destination. """ - DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL = "DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL" - DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL = "DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL" - DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL = "DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL" -ModelServiceConfigDestinationConfigDestinationTypeParam = Literal["DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL", "DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL", "DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL"] | ModelServiceConfigDestinationConfigDestinationType + DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL = ( + "DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL" + ) + DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL = ( + "DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL" + ) + DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL = ( + "DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL" + ) + + +ModelServiceConfigDestinationConfigDestinationTypeParam = ( + Literal[ + "DESTINATION_TYPE_PAY_PER_TOKEN_FOUNDATION_MODEL", + "DESTINATION_TYPE_PROVISIONED_THROUGHPUT_FOUNDATION_MODEL", + "DESTINATION_TYPE_EXTERNAL_FOUNDATION_MODEL", + ] + | ModelServiceConfigDestinationConfigDestinationType +) diff --git a/python/databricks/bundles/model_services/_models/model_service_config_external_model_config.py b/python/databricks/bundles/model_services/_models/model_service_config_external_model_config.py index 9d2ea2b3ed7..d82c7ad7d19 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_external_model_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_external_model_config.py @@ -1,38 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr +from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ( + ModelProviderServiceConfigModelTargetConfig, + ModelProviderServiceConfigModelTargetConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelServiceConfigExternalModelConfig: """ @@ -41,6 +23,7 @@ class ModelServiceConfigExternalModelConfig: ModelProviderService entity referenced via `model_provider_service`; the platform resolves the provider at invocation time. """ + model_provider_service: VariableOr[str] """ Resource name of the governed ModelProviderService that owns provider @@ -59,12 +42,11 @@ class ModelServiceConfigExternalModelConfig: """ @classmethod - def from_dict(cls, value: 'ModelServiceConfigExternalModelConfigDict') -> 'Self': + def from_dict(cls, value: "ModelServiceConfigExternalModelConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServiceConfigExternalModelConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServiceConfigExternalModelConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelServiceConfigExternalModelConfigDict(TypedDict, total=False): @@ -88,4 +70,6 @@ class ModelServiceConfigExternalModelConfigDict(TypedDict, total=False): """ -ModelServiceConfigExternalModelConfigParam = ModelServiceConfigExternalModelConfigDict | ModelServiceConfigExternalModelConfig +ModelServiceConfigExternalModelConfigParam = ( + ModelServiceConfigExternalModelConfigDict | ModelServiceConfigExternalModelConfig +) diff --git a/python/databricks/bundles/model_services/_models/model_service_config_fallback_config.py b/python/databricks/bundles/model_services/_models/model_service_config_fallback_config.py index 70e65e6b069..e450152f88d 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_fallback_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_fallback_config.py @@ -1,56 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.model_services._models.model_service_config_destination_config import ( + ModelServiceConfigDestinationConfig, + ModelServiceConfigDestinationConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelServiceConfigFallbackConfig: """ Fallback routing applied after a primary destination fails. Fallback destinations are tried in the listed order. """ - destinations: VariableOrList[ModelServiceConfigDestinationConfig] = field(default_factory=list) + + destinations: VariableOrList[ModelServiceConfigDestinationConfig] = field( + default_factory=list + ) """ Fallback destinations, tried in the listed order. At most 5 are allowed. """ @classmethod - def from_dict(cls, value: 'ModelServiceConfigFallbackConfigDict') -> 'Self': + def from_dict(cls, value: "ModelServiceConfigFallbackConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServiceConfigFallbackConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServiceConfigFallbackConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelServiceConfigFallbackConfigDict(TypedDict, total=False): @@ -62,4 +46,6 @@ class ModelServiceConfigFallbackConfigDict(TypedDict, total=False): """ -ModelServiceConfigFallbackConfigParam = ModelServiceConfigFallbackConfigDict | ModelServiceConfigFallbackConfig +ModelServiceConfigFallbackConfigParam = ( + ModelServiceConfigFallbackConfigDict | ModelServiceConfigFallbackConfig +) diff --git a/python/databricks/bundles/model_services/_models/model_service_config_pay_per_token_config.py b/python/databricks/bundles/model_services/_models/model_service_config_pay_per_token_config.py index 4e4062cd410..a42bbed4465 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_pay_per_token_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_pay_per_token_config.py @@ -1,38 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelServiceConfigPayPerTokenConfig: """ @@ -40,6 +18,7 @@ class ModelServiceConfigPayPerTokenConfig: the foundation model by its UC resource name; the platform resolves it to a Model Serving endpoint at request time. """ + model: VariableOr[str] """ Resource name of the Unity Catalog model. @@ -47,12 +26,11 @@ class ModelServiceConfigPayPerTokenConfig: """ @classmethod - def from_dict(cls, value: 'ModelServiceConfigPayPerTokenConfigDict') -> 'Self': + def from_dict(cls, value: "ModelServiceConfigPayPerTokenConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServiceConfigPayPerTokenConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServiceConfigPayPerTokenConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelServiceConfigPayPerTokenConfigDict(TypedDict, total=False): @@ -65,4 +43,6 @@ class ModelServiceConfigPayPerTokenConfigDict(TypedDict, total=False): """ -ModelServiceConfigPayPerTokenConfigParam = ModelServiceConfigPayPerTokenConfigDict | ModelServiceConfigPayPerTokenConfig +ModelServiceConfigPayPerTokenConfigParam = ( + ModelServiceConfigPayPerTokenConfigDict | ModelServiceConfigPayPerTokenConfig +) diff --git a/python/databricks/bundles/model_services/_models/model_service_config_provisioned_throughput_config.py b/python/databricks/bundles/model_services/_models/model_service_config_provisioned_throughput_config.py index ad88b18c112..d8a0c63cc3d 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_provisioned_throughput_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_provisioned_throughput_config.py @@ -1,38 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelServiceConfigProvisionedThroughputConfig: """ @@ -41,6 +19,7 @@ class ModelServiceConfigProvisionedThroughputConfig: sizing (provisioned throughput, burst scaling, model version) is owned by the Model Serving endpoint itself, not by this message. """ + model_serving_endpoint: VariableOr[str] """ Name of the backing Model Serving endpoint serving the provisioned- @@ -52,12 +31,13 @@ class ModelServiceConfigProvisionedThroughputConfig: """ @classmethod - def from_dict(cls, value: 'ModelServiceConfigProvisionedThroughputConfigDict') -> 'Self': + def from_dict( + cls, value: "ModelServiceConfigProvisionedThroughputConfigDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServiceConfigProvisionedThroughputConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServiceConfigProvisionedThroughputConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelServiceConfigProvisionedThroughputConfigDict(TypedDict, total=False): @@ -74,4 +54,7 @@ class ModelServiceConfigProvisionedThroughputConfigDict(TypedDict, total=False): """ -ModelServiceConfigProvisionedThroughputConfigParam = ModelServiceConfigProvisionedThroughputConfigDict | ModelServiceConfigProvisionedThroughputConfig +ModelServiceConfigProvisionedThroughputConfigParam = ( + ModelServiceConfigProvisionedThroughputConfigDict + | ModelServiceConfigProvisionedThroughputConfig +) diff --git a/python/databricks/bundles/model_services/_models/model_service_config_routing_config.py b/python/databricks/bundles/model_services/_models/model_service_config_routing_config.py index 6054d43e882..36b547f806f 100644 --- a/python/databricks/bundles/model_services/_models/model_service_config_routing_config.py +++ b/python/databricks/bundles/model_services/_models/model_service_config_routing_config.py @@ -1,45 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.model_services._models.model_service_config_destination_config import ( + ModelServiceConfigDestinationConfig, + ModelServiceConfigDestinationConfigParam, +) +from databricks.bundles.model_services._models.model_service_config_fallback_config import ( + ModelServiceConfigFallbackConfig, + ModelServiceConfigFallbackConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelServiceConfigRoutingConfig: """ Routing configuration for a model service, nesting destinations and fallback under a single sub-message. """ - destinations: VariableOrList[ModelServiceConfigDestinationConfig] = field(default_factory=list) + + destinations: VariableOrList[ModelServiceConfigDestinationConfig] = field( + default_factory=list + ) """ Primary routing destinations. At most 10 are allowed. At least one is required on Create. On Update, provide this list when replacing the full @@ -55,12 +44,11 @@ class ModelServiceConfigRoutingConfig: """ @classmethod - def from_dict(cls, value: 'ModelServiceConfigRoutingConfigDict') -> 'Self': + def from_dict(cls, value: "ModelServiceConfigRoutingConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServiceConfigRoutingConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServiceConfigRoutingConfigDict": + return _transform_to_json_value(self) # type:ignore class ModelServiceConfigRoutingConfigDict(TypedDict, total=False): @@ -82,4 +70,6 @@ class ModelServiceConfigRoutingConfigDict(TypedDict, total=False): """ -ModelServiceConfigRoutingConfigParam = ModelServiceConfigRoutingConfigDict | ModelServiceConfigRoutingConfig +ModelServiceConfigRoutingConfigParam = ( + ModelServiceConfigRoutingConfigDict | ModelServiceConfigRoutingConfig +) diff --git a/python/databricks/bundles/model_services/_models/privilege.py b/python/databricks/bundles/model_services/_models/privilege.py index a51483854b2..df45211a76a 100644 --- a/python/databricks/bundles/model_services/_models/privilege.py +++ b/python/databricks/bundles/model_services/_models/privilege.py @@ -1,32 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam +from typing import Literal class Privilege(Enum): @@ -83,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/model_services/_models/privilege_assignment.py b/python/databricks/bundles/model_services/_models/privilege_assignment.py index 5f91055ff3a..7bbd0ac80ee 100644 --- a/python/databricks/bundles/model_services/_models/privilege_assignment.py +++ b/python/databricks/bundles/model_services/_models/privilege_assignment.py @@ -1,38 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.model_services._models.privilege import ( + Privilege, + PrivilegeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -49,12 +31,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/rate_limit.py b/python/databricks/bundles/model_services/_models/rate_limit.py index 9f6b7feb8bb..355414bc28c 100644 --- a/python/databricks/bundles/model_services/_models/rate_limit.py +++ b/python/databricks/bundles/model_services/_models/rate_limit.py @@ -1,38 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.model_services._models.rate_limit_rate_limit_key import ( + RateLimitRateLimitKey, + RateLimitRateLimitKeyParam, +) +from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import ( + RateLimitRateLimitRenewalPeriod, + RateLimitRateLimitRenewalPeriodParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RateLimit: """ @@ -40,6 +26,7 @@ class RateLimit: unset to impose no limit on that dimension; set a value to cap that dimension within the renewal period. """ + key: VariableOr[RateLimitRateLimitKey] """ Scope of the rate limit. Depending on this value, the limit applies to a @@ -71,12 +58,11 @@ class RateLimit: """ @classmethod - def from_dict(cls, value: 'RateLimitDict') -> 'Self': + def from_dict(cls, value: "RateLimitDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RateLimitDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RateLimitDict": + return _transform_to_json_value(self) # type:ignore class RateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_key.py b/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_key.py index fef5d8b91ac..fb907da810f 100644 --- a/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_key.py +++ b/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_key.py @@ -1,42 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_renewal_period import RateLimitRateLimitRenewalPeriod, RateLimitRateLimitRenewalPeriodParam +from typing import Literal class RateLimitRateLimitKey(Enum): """ Scope key for a rate limit. """ + RATE_LIMIT_KEY_USER = "RATE_LIMIT_KEY_USER" RATE_LIMIT_KEY_USER_GROUP = "RATE_LIMIT_KEY_USER_GROUP" RATE_LIMIT_KEY_SERVICE_PRINCIPAL = "RATE_LIMIT_KEY_SERVICE_PRINCIPAL" RATE_LIMIT_KEY_SERVICE = "RATE_LIMIT_KEY_SERVICE" RATE_LIMIT_KEY_USER_DEFAULT = "RATE_LIMIT_KEY_USER_DEFAULT" -RateLimitRateLimitKeyParam = Literal["RATE_LIMIT_KEY_USER", "RATE_LIMIT_KEY_USER_GROUP", "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", "RATE_LIMIT_KEY_SERVICE", "RATE_LIMIT_KEY_USER_DEFAULT"] | RateLimitRateLimitKey + +RateLimitRateLimitKeyParam = ( + Literal[ + "RATE_LIMIT_KEY_USER", + "RATE_LIMIT_KEY_USER_GROUP", + "RATE_LIMIT_KEY_SERVICE_PRINCIPAL", + "RATE_LIMIT_KEY_SERVICE", + "RATE_LIMIT_KEY_USER_DEFAULT", + ] + | RateLimitRateLimitKey +) diff --git a/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_renewal_period.py b/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_renewal_period.py index f4fbc895749..58636c784c0 100644 --- a/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_renewal_period.py +++ b/python/databricks/bundles/model_services/_models/rate_limit_rate_limit_renewal_period.py @@ -1,39 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_services._models.inference_table_config import InferenceTableConfig, InferenceTableConfigDict, InferenceTableConfigParam -from databricks.bundles.model_services._models.model_provider_service_config_model_target_config import ModelProviderServiceConfigModelTargetConfig, ModelProviderServiceConfigModelTargetConfigDict, ModelProviderServiceConfigModelTargetConfigParam -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig, ModelServiceConfigDict, ModelServiceConfigParam -from databricks.bundles.model_services._models.model_service_config_destination_config import ModelServiceConfigDestinationConfig, ModelServiceConfigDestinationConfigDict, ModelServiceConfigDestinationConfigParam -from databricks.bundles.model_services._models.model_service_config_external_model_config import ModelServiceConfigExternalModelConfig, ModelServiceConfigExternalModelConfigDict, ModelServiceConfigExternalModelConfigParam -from databricks.bundles.model_services._models.model_service_config_fallback_config import ModelServiceConfigFallbackConfig, ModelServiceConfigFallbackConfigDict, ModelServiceConfigFallbackConfigParam -from databricks.bundles.model_services._models.model_service_config_pay_per_token_config import ModelServiceConfigPayPerTokenConfig, ModelServiceConfigPayPerTokenConfigDict, ModelServiceConfigPayPerTokenConfigParam -from databricks.bundles.model_services._models.model_service_config_provisioned_throughput_config import ModelServiceConfigProvisionedThroughputConfig, ModelServiceConfigProvisionedThroughputConfigDict, ModelServiceConfigProvisionedThroughputConfigParam -from databricks.bundles.model_services._models.model_service_config_routing_config import ModelServiceConfigRoutingConfig, ModelServiceConfigRoutingConfigDict, ModelServiceConfigRoutingConfigParam -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.model_services._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_services._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_services._models.model_service import ModelService, ModelServiceDict, ModelServiceParam -from databricks.bundles.model_services._models.model_service_config_destination_config_destination_type import ModelServiceConfigDestinationConfigDestinationType, ModelServiceConfigDestinationConfigDestinationTypeParam - -from databricks.bundles.model_services._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.model_services._models.rate_limit_rate_limit_key import RateLimitRateLimitKey, RateLimitRateLimitKeyParam +from typing import Literal class RateLimitRateLimitRenewalPeriod(Enum): """ Renewal period for a rate limit. """ + RATE_LIMIT_RENEWAL_PERIOD_MINUTE = "RATE_LIMIT_RENEWAL_PERIOD_MINUTE" RATE_LIMIT_RENEWAL_PERIOD_HOUR = "RATE_LIMIT_RENEWAL_PERIOD_HOUR" -RateLimitRateLimitRenewalPeriodParam = Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] | RateLimitRateLimitRenewalPeriod + +RateLimitRateLimitRenewalPeriodParam = ( + Literal["RATE_LIMIT_RENEWAL_PERIOD_MINUTE", "RATE_LIMIT_RENEWAL_PERIOD_HOUR"] + | RateLimitRateLimitRenewalPeriod +) diff --git a/python/databricks/bundles/model_serving_endpoints/__init__.py b/python/databricks/bundles/model_serving_endpoints/__init__.py index c9be1165854..0137c5e76b9 100644 --- a/python/databricks/bundles/model_serving_endpoints/__init__.py +++ b/python/databricks/bundles/model_serving_endpoints/__init__.py @@ -1,190 +1,352 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Ai21LabsConfig', - 'Ai21LabsConfigDict', - 'Ai21LabsConfigParam', - 'AiGatewayConfig', - 'AiGatewayConfigDict', - 'AiGatewayConfigParam', - 'AiGatewayGuardrailParameters', - 'AiGatewayGuardrailParametersDict', - 'AiGatewayGuardrailParametersParam', - 'AiGatewayGuardrailPiiBehavior', - 'AiGatewayGuardrailPiiBehaviorBehavior', - 'AiGatewayGuardrailPiiBehaviorBehaviorParam', - 'AiGatewayGuardrailPiiBehaviorDict', - 'AiGatewayGuardrailPiiBehaviorParam', - 'AiGatewayGuardrails', - 'AiGatewayGuardrailsDict', - 'AiGatewayGuardrailsParam', - 'AiGatewayInferenceTableConfig', - 'AiGatewayInferenceTableConfigDict', - 'AiGatewayInferenceTableConfigParam', - 'AiGatewayRateLimit', - 'AiGatewayRateLimitDict', - 'AiGatewayRateLimitKey', - 'AiGatewayRateLimitKeyParam', - 'AiGatewayRateLimitParam', - 'AiGatewayRateLimitRenewalPeriod', - 'AiGatewayRateLimitRenewalPeriodParam', - 'AiGatewayUsageTrackingConfig', - 'AiGatewayUsageTrackingConfigDict', - 'AiGatewayUsageTrackingConfigParam', - 'AmazonBedrockConfig', - 'AmazonBedrockConfigBedrockProvider', - 'AmazonBedrockConfigBedrockProviderParam', - 'AmazonBedrockConfigDict', - 'AmazonBedrockConfigParam', - 'AnthropicConfig', - 'AnthropicConfigDict', - 'AnthropicConfigParam', - 'ApiKeyAuth', - 'ApiKeyAuthDict', - 'ApiKeyAuthParam', - 'AutoCaptureConfigInput', - 'AutoCaptureConfigInputDict', - 'AutoCaptureConfigInputParam', - 'BearerTokenAuth', - 'BearerTokenAuthDict', - 'BearerTokenAuthParam', - 'CohereConfig', - 'CohereConfigDict', - 'CohereConfigParam', - 'CustomProviderConfig', - 'CustomProviderConfigDict', - 'CustomProviderConfigParam', - 'DatabricksModelServingConfig', - 'DatabricksModelServingConfigDict', - 'DatabricksModelServingConfigParam', - 'EmailNotifications', - 'EmailNotificationsDict', - 'EmailNotificationsParam', - 'EndpointCoreConfigInput', - 'EndpointCoreConfigInputDict', - 'EndpointCoreConfigInputParam', - 'EndpointTag', - 'EndpointTagDict', - 'EndpointTagParam', - 'ExternalModel', - 'ExternalModelDict', - 'ExternalModelParam', - 'ExternalModelProvider', - 'ExternalModelProviderParam', - 'FallbackConfig', - 'FallbackConfigDict', - 'FallbackConfigParam', - 'GoogleCloudVertexAiConfig', - 'GoogleCloudVertexAiConfigDict', - 'GoogleCloudVertexAiConfigParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'ModelServingEndpoint', - 'ModelServingEndpointDict', - 'ModelServingEndpointParam', - 'ModelServingEndpointPermission', - 'ModelServingEndpointPermissionDict', - 'ModelServingEndpointPermissionParam', - 'OpenAiConfig', - 'OpenAiConfigDict', - 'OpenAiConfigParam', - 'PaLmConfig', - 'PaLmConfigDict', - 'PaLmConfigParam', - 'RateLimit', - 'RateLimitDict', - 'RateLimitKey', - 'RateLimitKeyParam', - 'RateLimitParam', - 'RateLimitRenewalPeriod', - 'RateLimitRenewalPeriodParam', - 'Route', - 'RouteDict', - 'RouteParam', - 'ServedEntityInput', - 'ServedEntityInputDict', - 'ServedEntityInputParam', - 'ServedModelInput', - 'ServedModelInputDict', - 'ServedModelInputParam', - 'ServedModelInputWorkloadType', - 'ServedModelInputWorkloadTypeParam', - 'ServingEndpointPermissionLevel', - 'ServingEndpointPermissionLevelParam', - 'ServingModelWorkloadType', - 'ServingModelWorkloadTypeParam', - 'TelemetryConfig', - 'TelemetryConfigDict', - 'TelemetryConfigParam', - 'TelemetryFeature', - 'TelemetryFeatureParam', - 'TelemetryInferenceTableConfig', - 'TelemetryInferenceTableConfigDict', - 'TelemetryInferenceTableConfigParam', - 'TrafficConfig', - 'TrafficConfigDict', - 'TrafficConfigParam', - 'UnityCatalogTableNames', - 'UnityCatalogTableNamesDict', - 'UnityCatalogTableNamesParam', + "Ai21LabsConfig", + "Ai21LabsConfigDict", + "Ai21LabsConfigParam", + "AiGatewayConfig", + "AiGatewayConfigDict", + "AiGatewayConfigParam", + "AiGatewayGuardrailParameters", + "AiGatewayGuardrailParametersDict", + "AiGatewayGuardrailParametersParam", + "AiGatewayGuardrailPiiBehavior", + "AiGatewayGuardrailPiiBehaviorBehavior", + "AiGatewayGuardrailPiiBehaviorBehaviorParam", + "AiGatewayGuardrailPiiBehaviorDict", + "AiGatewayGuardrailPiiBehaviorParam", + "AiGatewayGuardrails", + "AiGatewayGuardrailsDict", + "AiGatewayGuardrailsParam", + "AiGatewayInferenceTableConfig", + "AiGatewayInferenceTableConfigDict", + "AiGatewayInferenceTableConfigParam", + "AiGatewayRateLimit", + "AiGatewayRateLimitDict", + "AiGatewayRateLimitKey", + "AiGatewayRateLimitKeyParam", + "AiGatewayRateLimitParam", + "AiGatewayRateLimitRenewalPeriod", + "AiGatewayRateLimitRenewalPeriodParam", + "AiGatewayUsageTrackingConfig", + "AiGatewayUsageTrackingConfigDict", + "AiGatewayUsageTrackingConfigParam", + "AmazonBedrockConfig", + "AmazonBedrockConfigBedrockProvider", + "AmazonBedrockConfigBedrockProviderParam", + "AmazonBedrockConfigDict", + "AmazonBedrockConfigParam", + "AnthropicConfig", + "AnthropicConfigDict", + "AnthropicConfigParam", + "ApiKeyAuth", + "ApiKeyAuthDict", + "ApiKeyAuthParam", + "AutoCaptureConfigInput", + "AutoCaptureConfigInputDict", + "AutoCaptureConfigInputParam", + "BearerTokenAuth", + "BearerTokenAuthDict", + "BearerTokenAuthParam", + "CohereConfig", + "CohereConfigDict", + "CohereConfigParam", + "CustomProviderConfig", + "CustomProviderConfigDict", + "CustomProviderConfigParam", + "DatabricksModelServingConfig", + "DatabricksModelServingConfigDict", + "DatabricksModelServingConfigParam", + "EmailNotifications", + "EmailNotificationsDict", + "EmailNotificationsParam", + "EndpointCoreConfigInput", + "EndpointCoreConfigInputDict", + "EndpointCoreConfigInputParam", + "EndpointTag", + "EndpointTagDict", + "EndpointTagParam", + "ExternalModel", + "ExternalModelDict", + "ExternalModelParam", + "ExternalModelProvider", + "ExternalModelProviderParam", + "FallbackConfig", + "FallbackConfigDict", + "FallbackConfigParam", + "GoogleCloudVertexAiConfig", + "GoogleCloudVertexAiConfigDict", + "GoogleCloudVertexAiConfigParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "ModelServingEndpoint", + "ModelServingEndpointDict", + "ModelServingEndpointParam", + "ModelServingEndpointPermission", + "ModelServingEndpointPermissionDict", + "ModelServingEndpointPermissionParam", + "OpenAiConfig", + "OpenAiConfigDict", + "OpenAiConfigParam", + "PaLmConfig", + "PaLmConfigDict", + "PaLmConfigParam", + "RateLimit", + "RateLimitDict", + "RateLimitKey", + "RateLimitKeyParam", + "RateLimitParam", + "RateLimitRenewalPeriod", + "RateLimitRenewalPeriodParam", + "Route", + "RouteDict", + "RouteParam", + "ServedEntityInput", + "ServedEntityInputDict", + "ServedEntityInputParam", + "ServedModelInput", + "ServedModelInputDict", + "ServedModelInputParam", + "ServedModelInputWorkloadType", + "ServedModelInputWorkloadTypeParam", + "ServingEndpointPermissionLevel", + "ServingEndpointPermissionLevelParam", + "ServingModelWorkloadType", + "ServingModelWorkloadTypeParam", + "TelemetryConfig", + "TelemetryConfigDict", + "TelemetryConfigParam", + "TelemetryFeature", + "TelemetryFeatureParam", + "TelemetryInferenceTableConfig", + "TelemetryInferenceTableConfigDict", + "TelemetryInferenceTableConfigParam", + "TrafficConfig", + "TrafficConfigDict", + "TrafficConfigParam", + "UnityCatalogTableNames", + "UnityCatalogTableNamesDict", + "UnityCatalogTableNamesParam", ] -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import ( + Ai21LabsConfig, + Ai21LabsConfigDict, + Ai21LabsConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import ( + AiGatewayConfig, + AiGatewayConfigDict, + AiGatewayConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import ( + AiGatewayGuardrailParameters, + AiGatewayGuardrailParametersDict, + AiGatewayGuardrailParametersParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import ( + AiGatewayGuardrailPiiBehavior, + AiGatewayGuardrailPiiBehaviorDict, + AiGatewayGuardrailPiiBehaviorParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import ( + AiGatewayGuardrailPiiBehaviorBehavior, + AiGatewayGuardrailPiiBehaviorBehaviorParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import ( + AiGatewayGuardrails, + AiGatewayGuardrailsDict, + AiGatewayGuardrailsParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import ( + AiGatewayInferenceTableConfig, + AiGatewayInferenceTableConfigDict, + AiGatewayInferenceTableConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import ( + AiGatewayRateLimit, + AiGatewayRateLimitDict, + AiGatewayRateLimitParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import ( + AiGatewayRateLimitKey, + AiGatewayRateLimitKeyParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import ( + AiGatewayRateLimitRenewalPeriod, + AiGatewayRateLimitRenewalPeriodParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import ( + AiGatewayUsageTrackingConfig, + AiGatewayUsageTrackingConfigDict, + AiGatewayUsageTrackingConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import ( + AmazonBedrockConfig, + AmazonBedrockConfigDict, + AmazonBedrockConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import ( + AmazonBedrockConfigBedrockProvider, + AmazonBedrockConfigBedrockProviderParam, +) +from databricks.bundles.model_serving_endpoints._models.anthropic_config import ( + AnthropicConfig, + AnthropicConfigDict, + AnthropicConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ( + ApiKeyAuth, + ApiKeyAuthDict, + ApiKeyAuthParam, +) +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import ( + AutoCaptureConfigInput, + AutoCaptureConfigInputDict, + AutoCaptureConfigInputParam, +) +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import ( + BearerTokenAuth, + BearerTokenAuthDict, + BearerTokenAuthParam, +) +from databricks.bundles.model_serving_endpoints._models.cohere_config import ( + CohereConfig, + CohereConfigDict, + CohereConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import ( + CustomProviderConfig, + CustomProviderConfigDict, + CustomProviderConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import ( + DatabricksModelServingConfig, + DatabricksModelServingConfigDict, + DatabricksModelServingConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.email_notifications import ( + EmailNotifications, + EmailNotificationsDict, + EmailNotificationsParam, +) +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import ( + EndpointCoreConfigInput, + EndpointCoreConfigInputDict, + EndpointCoreConfigInputParam, +) +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import ( + EndpointTag, + EndpointTagDict, + EndpointTagParam, +) +from databricks.bundles.model_serving_endpoints._models.external_model import ( + ExternalModel, + ExternalModelDict, + ExternalModelParam, +) +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ( + ExternalModelProvider, + ExternalModelProviderParam, +) +from databricks.bundles.model_serving_endpoints._models.fallback_config import ( + FallbackConfig, + FallbackConfigDict, + FallbackConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import ( + GoogleCloudVertexAiConfig, + GoogleCloudVertexAiConfigDict, + GoogleCloudVertexAiConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( + ModelServingEndpoint, + ModelServingEndpointDict, + ModelServingEndpointParam, +) +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ( + ModelServingEndpointPermission, + ModelServingEndpointPermissionDict, + ModelServingEndpointPermissionParam, +) +from databricks.bundles.model_serving_endpoints._models.open_ai_config import ( + OpenAiConfig, + OpenAiConfigDict, + OpenAiConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import ( + PaLmConfig, + PaLmConfigDict, + PaLmConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.rate_limit import ( + RateLimit, + RateLimitDict, + RateLimitParam, +) +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import ( + RateLimitKey, + RateLimitKeyParam, +) +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import ( + RateLimitRenewalPeriod, + RateLimitRenewalPeriodParam, +) +from databricks.bundles.model_serving_endpoints._models.route import ( + Route, + RouteDict, + RouteParam, +) +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ( + ServedEntityInput, + ServedEntityInputDict, + ServedEntityInputParam, +) +from databricks.bundles.model_serving_endpoints._models.served_model_input import ( + ServedModelInput, + ServedModelInputDict, + ServedModelInputParam, +) +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ( + ServedModelInputWorkloadType, + ServedModelInputWorkloadTypeParam, +) +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ( + ServingEndpointPermissionLevel, + ServingEndpointPermissionLevelParam, +) +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ( + ServingModelWorkloadType, + ServingModelWorkloadTypeParam, +) +from databricks.bundles.model_serving_endpoints._models.telemetry_config import ( + TelemetryConfig, + TelemetryConfigDict, + TelemetryConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import ( + TelemetryFeature, + TelemetryFeatureParam, +) +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import ( + TelemetryInferenceTableConfig, + TelemetryInferenceTableConfigDict, + TelemetryInferenceTableConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.traffic_config import ( + TrafficConfig, + TrafficConfigDict, + TrafficConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import ( + UnityCatalogTableNames, + UnityCatalogTableNamesDict, + UnityCatalogTableNamesParam, +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai21_labs_config.py b/python/databricks/bundles/model_serving_endpoints/_models/ai21_labs_config.py index 46afde31118..af06b6f6f1e 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai21_labs_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai21_labs_config.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Ai21LabsConfig: """""" @@ -90,12 +32,11 @@ class Ai21LabsConfig: """ @classmethod - def from_dict(cls, value: 'Ai21LabsConfigDict') -> 'Self': + def from_dict(cls, value: "Ai21LabsConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'Ai21LabsConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "Ai21LabsConfigDict": + return _transform_to_json_value(self) # type:ignore class Ai21LabsConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_config.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_config.py index 9c1a3e23361..24a4a967599 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_config.py @@ -1,74 +1,36 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import ( + AiGatewayGuardrails, + AiGatewayGuardrailsParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import ( + AiGatewayInferenceTableConfig, + AiGatewayInferenceTableConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import ( + AiGatewayRateLimit, + AiGatewayRateLimitParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import ( + AiGatewayUsageTrackingConfig, + AiGatewayUsageTrackingConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.fallback_config import ( + FallbackConfig, + FallbackConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AiGatewayConfig: """""" @@ -102,12 +64,11 @@ class AiGatewayConfig: """ @classmethod - def from_dict(cls, value: 'AiGatewayConfigDict') -> 'Self': + def from_dict(cls, value: "AiGatewayConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AiGatewayConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AiGatewayConfigDict": + return _transform_to_json_value(self) # type:ignore class AiGatewayConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_parameters.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_parameters.py index 82b90be32b2..55b70a82669 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_parameters.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_parameters.py @@ -1,74 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import ( + AiGatewayGuardrailPiiBehavior, + AiGatewayGuardrailPiiBehaviorParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AiGatewayGuardrailParameters: """""" @@ -96,12 +42,11 @@ class AiGatewayGuardrailParameters: """ @classmethod - def from_dict(cls, value: 'AiGatewayGuardrailParametersDict') -> 'Self': + def from_dict(cls, value: "AiGatewayGuardrailParametersDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AiGatewayGuardrailParametersDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AiGatewayGuardrailParametersDict": + return _transform_to_json_value(self) # type:ignore class AiGatewayGuardrailParametersDict(TypedDict, total=False): @@ -130,4 +75,6 @@ class AiGatewayGuardrailParametersDict(TypedDict, total=False): """ -AiGatewayGuardrailParametersParam = AiGatewayGuardrailParametersDict | AiGatewayGuardrailParameters +AiGatewayGuardrailParametersParam = ( + AiGatewayGuardrailParametersDict | AiGatewayGuardrailParameters +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior.py index 4d34c2ac3ac..577af194a55 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior.py @@ -1,74 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import ( + AiGatewayGuardrailPiiBehaviorBehavior, + AiGatewayGuardrailPiiBehaviorBehaviorParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AiGatewayGuardrailPiiBehavior: """""" @@ -79,12 +25,11 @@ class AiGatewayGuardrailPiiBehavior: """ @classmethod - def from_dict(cls, value: 'AiGatewayGuardrailPiiBehaviorDict') -> 'Self': + def from_dict(cls, value: "AiGatewayGuardrailPiiBehaviorDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AiGatewayGuardrailPiiBehaviorDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AiGatewayGuardrailPiiBehaviorDict": + return _transform_to_json_value(self) # type:ignore class AiGatewayGuardrailPiiBehaviorDict(TypedDict, total=False): @@ -96,4 +41,6 @@ class AiGatewayGuardrailPiiBehaviorDict(TypedDict, total=False): """ -AiGatewayGuardrailPiiBehaviorParam = AiGatewayGuardrailPiiBehaviorDict | AiGatewayGuardrailPiiBehavior +AiGatewayGuardrailPiiBehaviorParam = ( + AiGatewayGuardrailPiiBehaviorDict | AiGatewayGuardrailPiiBehavior +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior_behavior.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior_behavior.py index f93de2dc112..bf9bf86fc68 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior_behavior.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrail_pii_behavior_behavior.py @@ -1,68 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class AiGatewayGuardrailPiiBehaviorBehavior(Enum): @@ -70,4 +9,7 @@ class AiGatewayGuardrailPiiBehaviorBehavior(Enum): BLOCK = "BLOCK" MASK = "MASK" -AiGatewayGuardrailPiiBehaviorBehaviorParam = Literal["NONE", "BLOCK", "MASK"] | AiGatewayGuardrailPiiBehaviorBehavior + +AiGatewayGuardrailPiiBehaviorBehaviorParam = ( + Literal["NONE", "BLOCK", "MASK"] | AiGatewayGuardrailPiiBehaviorBehavior +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrails.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrails.py index c601b74f99b..299859aa015 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrails.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_guardrails.py @@ -1,74 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import ( + AiGatewayGuardrailParameters, + AiGatewayGuardrailParametersParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AiGatewayGuardrails: """""" @@ -84,12 +30,11 @@ class AiGatewayGuardrails: """ @classmethod - def from_dict(cls, value: 'AiGatewayGuardrailsDict') -> 'Self': + def from_dict(cls, value: "AiGatewayGuardrailsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AiGatewayGuardrailsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AiGatewayGuardrailsDict": + return _transform_to_json_value(self) # type:ignore class AiGatewayGuardrailsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_inference_table_config.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_inference_table_config.py index fc51a362abc..a4d748dee3a 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_inference_table_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_inference_table_config.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AiGatewayInferenceTableConfig: """""" @@ -97,12 +39,11 @@ class AiGatewayInferenceTableConfig: """ @classmethod - def from_dict(cls, value: 'AiGatewayInferenceTableConfigDict') -> 'Self': + def from_dict(cls, value: "AiGatewayInferenceTableConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AiGatewayInferenceTableConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AiGatewayInferenceTableConfigDict": + return _transform_to_json_value(self) # type:ignore class AiGatewayInferenceTableConfigDict(TypedDict, total=False): @@ -132,4 +73,6 @@ class AiGatewayInferenceTableConfigDict(TypedDict, total=False): """ -AiGatewayInferenceTableConfigParam = AiGatewayInferenceTableConfigDict | AiGatewayInferenceTableConfig +AiGatewayInferenceTableConfigParam = ( + AiGatewayInferenceTableConfigDict | AiGatewayInferenceTableConfig +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit.py index 089e3ad35c6..5848e04f895 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit.py @@ -1,74 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import ( + AiGatewayRateLimitKey, + AiGatewayRateLimitKeyParam, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import ( + AiGatewayRateLimitRenewalPeriod, + AiGatewayRateLimitRenewalPeriodParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AiGatewayRateLimit: """""" @@ -100,12 +50,11 @@ class AiGatewayRateLimit: """ @classmethod - def from_dict(cls, value: 'AiGatewayRateLimitDict') -> 'Self': + def from_dict(cls, value: "AiGatewayRateLimitDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AiGatewayRateLimitDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AiGatewayRateLimitDict": + return _transform_to_json_value(self) # type:ignore class AiGatewayRateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_key.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_key.py index 2b81fda6005..f166207b7e7 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_key.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_key.py @@ -1,68 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class AiGatewayRateLimitKey(Enum): @@ -71,4 +10,8 @@ class AiGatewayRateLimitKey(Enum): USER_GROUP = "user_group" SERVICE_PRINCIPAL = "service_principal" -AiGatewayRateLimitKeyParam = Literal["user", "endpoint", "user_group", "service_principal"] | AiGatewayRateLimitKey + +AiGatewayRateLimitKeyParam = ( + Literal["user", "endpoint", "user_group", "service_principal"] + | AiGatewayRateLimitKey +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_renewal_period.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_renewal_period.py index d64c74e995d..e315d878c83 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_renewal_period.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_rate_limit_renewal_period.py @@ -1,71 +1,13 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class AiGatewayRateLimitRenewalPeriod(Enum): MINUTE = "minute" -AiGatewayRateLimitRenewalPeriodParam = Literal["minute"] | AiGatewayRateLimitRenewalPeriod + +AiGatewayRateLimitRenewalPeriodParam = ( + Literal["minute"] | AiGatewayRateLimitRenewalPeriod +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_usage_tracking_config.py b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_usage_tracking_config.py index 8648140e77d..bd5b5dd2806 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_usage_tracking_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/ai_gateway_usage_tracking_config.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AiGatewayUsageTrackingConfig: """""" @@ -79,12 +21,11 @@ class AiGatewayUsageTrackingConfig: """ @classmethod - def from_dict(cls, value: 'AiGatewayUsageTrackingConfigDict') -> 'Self': + def from_dict(cls, value: "AiGatewayUsageTrackingConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AiGatewayUsageTrackingConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AiGatewayUsageTrackingConfigDict": + return _transform_to_json_value(self) # type:ignore class AiGatewayUsageTrackingConfigDict(TypedDict, total=False): @@ -96,4 +37,6 @@ class AiGatewayUsageTrackingConfigDict(TypedDict, total=False): """ -AiGatewayUsageTrackingConfigParam = AiGatewayUsageTrackingConfigDict | AiGatewayUsageTrackingConfig +AiGatewayUsageTrackingConfigParam = ( + AiGatewayUsageTrackingConfigDict | AiGatewayUsageTrackingConfig +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config.py b/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config.py index 9eb73c0a2b0..0a51df09ba8 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config.py @@ -1,74 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import ( + AmazonBedrockConfigBedrockProvider, + AmazonBedrockConfigBedrockProviderParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AmazonBedrockConfig: """""" @@ -137,12 +83,11 @@ class AmazonBedrockConfig: """ @classmethod - def from_dict(cls, value: 'AmazonBedrockConfigDict') -> 'Self': + def from_dict(cls, value: "AmazonBedrockConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AmazonBedrockConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AmazonBedrockConfigDict": + return _transform_to_json_value(self) # type:ignore class AmazonBedrockConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config_bedrock_provider.py b/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config_bedrock_provider.py index 0242f334506..bc3de02a2e0 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config_bedrock_provider.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/amazon_bedrock_config_bedrock_provider.py @@ -1,68 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class AmazonBedrockConfigBedrockProvider(Enum): @@ -71,4 +10,8 @@ class AmazonBedrockConfigBedrockProvider(Enum): AI21LABS = "ai21labs" AMAZON = "amazon" -AmazonBedrockConfigBedrockProviderParam = Literal["anthropic", "cohere", "ai21labs", "amazon"] | AmazonBedrockConfigBedrockProvider + +AmazonBedrockConfigBedrockProviderParam = ( + Literal["anthropic", "cohere", "ai21labs", "amazon"] + | AmazonBedrockConfigBedrockProvider +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/anthropic_config.py b/python/databricks/bundles/model_serving_endpoints/_models/anthropic_config.py index a1a191a8d5e..97ab29711b0 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/anthropic_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/anthropic_config.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AnthropicConfig: """""" @@ -90,12 +32,11 @@ class AnthropicConfig: """ @classmethod - def from_dict(cls, value: 'AnthropicConfigDict') -> 'Self': + def from_dict(cls, value: "AnthropicConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AnthropicConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AnthropicConfigDict": + return _transform_to_json_value(self) # type:ignore class AnthropicConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/api_key_auth.py b/python/databricks/bundles/model_serving_endpoints/_models/api_key_auth.py index 1fd37db68c9..a5a33f0aad5 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/api_key_auth.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/api_key_auth.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ApiKeyAuth: """""" @@ -91,12 +33,11 @@ class ApiKeyAuth: """ @classmethod - def from_dict(cls, value: 'ApiKeyAuthDict') -> 'Self': + def from_dict(cls, value: "ApiKeyAuthDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ApiKeyAuthDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ApiKeyAuthDict": + return _transform_to_json_value(self) # type:ignore class ApiKeyAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/auto_capture_config_input.py b/python/databricks/bundles/model_serving_endpoints/_models/auto_capture_config_input.py index 53c2e3b608a..b7e3a5820bb 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/auto_capture_config_input.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/auto_capture_config_input.py @@ -1,80 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AutoCaptureConfigInput: """ [DEPRECATED] Deprecated: legacy inference table configuration. Please use AI Gateway inference tables instead. See https://docs.databricks.com/aws/en/ai-gateway/inference-tables. """ + catalog_name: VariableOrOptional[str] = None """ The name of the catalog in Unity Catalog. NOTE: On update, you cannot change the catalog name if the inference table is already enabled. @@ -96,12 +39,11 @@ class AutoCaptureConfigInput: """ @classmethod - def from_dict(cls, value: 'AutoCaptureConfigInputDict') -> 'Self': + def from_dict(cls, value: "AutoCaptureConfigInputDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AutoCaptureConfigInputDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AutoCaptureConfigInputDict": + return _transform_to_json_value(self) # type:ignore class AutoCaptureConfigInputDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/bearer_token_auth.py b/python/databricks/bundles/model_serving_endpoints/_models/bearer_token_auth.py index b48cba675ab..dc05f6c53c2 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/bearer_token_auth.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/bearer_token_auth.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class BearerTokenAuth: """""" @@ -86,12 +28,11 @@ class BearerTokenAuth: """ @classmethod - def from_dict(cls, value: 'BearerTokenAuthDict') -> 'Self': + def from_dict(cls, value: "BearerTokenAuthDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'BearerTokenAuthDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "BearerTokenAuthDict": + return _transform_to_json_value(self) # type:ignore class BearerTokenAuthDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/cohere_config.py b/python/databricks/bundles/model_serving_endpoints/_models/cohere_config.py index ca8209db65b..e70c767e9c2 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/cohere_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/cohere_config.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class CohereConfig: """""" @@ -96,12 +38,11 @@ class CohereConfig: """ @classmethod - def from_dict(cls, value: 'CohereConfigDict') -> 'Self': + def from_dict(cls, value: "CohereConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CohereConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CohereConfigDict": + return _transform_to_json_value(self) # type:ignore class CohereConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/custom_provider_config.py b/python/databricks/bundles/model_serving_endpoints/_models/custom_provider_config.py index bccc6d70876..e50b61d00ff 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/custom_provider_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/custom_provider_config.py @@ -1,79 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.api_key_auth import ( + ApiKeyAuth, + ApiKeyAuthParam, +) +from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import ( + BearerTokenAuth, + BearerTokenAuthParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class CustomProviderConfig: """ Configs needed to create a custom provider model route. """ + custom_provider_url: VariableOr[str] """ This is a field to provide the URL of the custom provider API. @@ -92,12 +43,11 @@ class CustomProviderConfig: """ @classmethod - def from_dict(cls, value: 'CustomProviderConfigDict') -> 'Self': + def from_dict(cls, value: "CustomProviderConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'CustomProviderConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "CustomProviderConfigDict": + return _transform_to_json_value(self) # type:ignore class CustomProviderConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/databricks_model_serving_config.py b/python/databricks/bundles/model_serving_endpoints/_models/databricks_model_serving_config.py index 72dc9af949c..72f17e7c5de 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/databricks_model_serving_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/databricks_model_serving_config.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DatabricksModelServingConfig: """""" @@ -100,12 +42,11 @@ class DatabricksModelServingConfig: """ @classmethod - def from_dict(cls, value: 'DatabricksModelServingConfigDict') -> 'Self': + def from_dict(cls, value: "DatabricksModelServingConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DatabricksModelServingConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DatabricksModelServingConfigDict": + return _transform_to_json_value(self) # type:ignore class DatabricksModelServingConfigDict(TypedDict, total=False): @@ -138,4 +79,6 @@ class DatabricksModelServingConfigDict(TypedDict, total=False): """ -DatabricksModelServingConfigParam = DatabricksModelServingConfigDict | DatabricksModelServingConfig +DatabricksModelServingConfigParam = ( + DatabricksModelServingConfigDict | DatabricksModelServingConfig +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/email_notifications.py b/python/databricks/bundles/model_serving_endpoints/_models/email_notifications.py index a422003de36..3eaeb0a8358 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/email_notifications.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/email_notifications.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EmailNotifications: """""" @@ -84,12 +26,11 @@ class EmailNotifications: """ @classmethod - def from_dict(cls, value: 'EmailNotificationsDict') -> 'Self': + def from_dict(cls, value: "EmailNotificationsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EmailNotificationsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EmailNotificationsDict": + return _transform_to_json_value(self) # type:ignore class EmailNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/endpoint_core_config_input.py b/python/databricks/bundles/model_serving_endpoints/_models/endpoint_core_config_input.py index 62eb78c51eb..b74a53e067d 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/endpoint_core_config_input.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/endpoint_core_config_input.py @@ -1,74 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import ( + AutoCaptureConfigInput, + AutoCaptureConfigInputParam, +) +from databricks.bundles.model_serving_endpoints._models.served_entity_input import ( + ServedEntityInput, + ServedEntityInputParam, +) +from databricks.bundles.model_serving_endpoints._models.served_model_input import ( + ServedModelInput, + ServedModelInputParam, +) +from databricks.bundles.model_serving_endpoints._models.traffic_config import ( + TrafficConfig, + TrafficConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EndpointCoreConfigInput: """""" @@ -97,12 +55,11 @@ class EndpointCoreConfigInput: """ @classmethod - def from_dict(cls, value: 'EndpointCoreConfigInputDict') -> 'Self': + def from_dict(cls, value: "EndpointCoreConfigInputDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EndpointCoreConfigInputDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EndpointCoreConfigInputDict": + return _transform_to_json_value(self) # type:ignore class EndpointCoreConfigInputDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/endpoint_tag.py b/python/databricks/bundles/model_serving_endpoints/_models/endpoint_tag.py index 3ecd5c66208..ee2bc8943d1 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/endpoint_tag.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/endpoint_tag.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EndpointTag: """""" @@ -84,12 +26,11 @@ class EndpointTag: """ @classmethod - def from_dict(cls, value: 'EndpointTagDict') -> 'Self': + def from_dict(cls, value: "EndpointTagDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EndpointTagDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EndpointTagDict": + return _transform_to_json_value(self) # type:ignore class EndpointTagDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/external_model.py b/python/databricks/bundles/model_serving_endpoints/_models/external_model.py index 2fafeb97b09..0f2fe2a6224 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/external_model.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/external_model.py @@ -1,74 +1,56 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import ( + Ai21LabsConfig, + Ai21LabsConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import ( + AmazonBedrockConfig, + AmazonBedrockConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.anthropic_config import ( + AnthropicConfig, + AnthropicConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.cohere_config import ( + CohereConfig, + CohereConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.custom_provider_config import ( + CustomProviderConfig, + CustomProviderConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import ( + DatabricksModelServingConfig, + DatabricksModelServingConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.external_model_provider import ( + ExternalModelProvider, + ExternalModelProviderParam, +) +from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import ( + GoogleCloudVertexAiConfig, + GoogleCloudVertexAiConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.open_ai_config import ( + OpenAiConfig, + OpenAiConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.pa_lm_config import ( + PaLmConfig, + PaLmConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ExternalModel: """""" @@ -113,7 +95,9 @@ class ExternalModel: Custom Provider Config. Only required if the provider is 'custom'. """ - databricks_model_serving_config: VariableOrOptional[DatabricksModelServingConfig] = None + databricks_model_serving_config: VariableOrOptional[ + DatabricksModelServingConfig + ] = None """ Databricks Model Serving Config. Only required if the provider is 'databricks-model-serving'. """ @@ -134,12 +118,11 @@ class ExternalModel: """ @classmethod - def from_dict(cls, value: 'ExternalModelDict') -> 'Self': + def from_dict(cls, value: "ExternalModelDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ExternalModelDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ExternalModelDict": + return _transform_to_json_value(self) # type:ignore class ExternalModelDict(TypedDict, total=False): @@ -185,7 +168,9 @@ class ExternalModelDict(TypedDict, total=False): Custom Provider Config. Only required if the provider is 'custom'. """ - databricks_model_serving_config: VariableOrOptional[DatabricksModelServingConfigParam] + databricks_model_serving_config: VariableOrOptional[ + DatabricksModelServingConfigParam + ] """ Databricks Model Serving Config. Only required if the provider is 'databricks-model-serving'. """ diff --git a/python/databricks/bundles/model_serving_endpoints/_models/external_model_provider.py b/python/databricks/bundles/model_serving_endpoints/_models/external_model_provider.py index 706ff0f811c..ee3ef3e0c08 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/external_model_provider.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/external_model_provider.py @@ -1,68 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class ExternalModelProvider(Enum): @@ -76,4 +15,18 @@ class ExternalModelProvider(Enum): PALM = "palm" CUSTOM = "custom" -ExternalModelProviderParam = Literal["ai21labs", "anthropic", "amazon-bedrock", "cohere", "databricks-model-serving", "google-cloud-vertex-ai", "openai", "palm", "custom"] | ExternalModelProvider + +ExternalModelProviderParam = ( + Literal[ + "ai21labs", + "anthropic", + "amazon-bedrock", + "cohere", + "databricks-model-serving", + "google-cloud-vertex-ai", + "openai", + "palm", + "custom", + ] + | ExternalModelProvider +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/fallback_config.py b/python/databricks/bundles/model_serving_endpoints/_models/fallback_config.py index 591e2042919..c96493b0982 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/fallback_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/fallback_config.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FallbackConfig: """""" @@ -82,12 +24,11 @@ class FallbackConfig: """ @classmethod - def from_dict(cls, value: 'FallbackConfigDict') -> 'Self': + def from_dict(cls, value: "FallbackConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FallbackConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FallbackConfigDict": + return _transform_to_json_value(self) # type:ignore class FallbackConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/google_cloud_vertex_ai_config.py b/python/databricks/bundles/model_serving_endpoints/_models/google_cloud_vertex_ai_config.py index 5e7c526dbd9..5b8e3030f56 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/google_cloud_vertex_ai_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/google_cloud_vertex_ai_config.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GoogleCloudVertexAiConfig: """""" @@ -113,12 +55,11 @@ class GoogleCloudVertexAiConfig: """ @classmethod - def from_dict(cls, value: 'GoogleCloudVertexAiConfigDict') -> 'Self': + def from_dict(cls, value: "GoogleCloudVertexAiConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GoogleCloudVertexAiConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GoogleCloudVertexAiConfigDict": + return _transform_to_json_value(self) # type:ignore class GoogleCloudVertexAiConfigDict(TypedDict, total=False): @@ -164,4 +105,6 @@ class GoogleCloudVertexAiConfigDict(TypedDict, total=False): """ -GoogleCloudVertexAiConfigParam = GoogleCloudVertexAiConfigDict | GoogleCloudVertexAiConfig +GoogleCloudVertexAiConfigParam = ( + GoogleCloudVertexAiConfigDict | GoogleCloudVertexAiConfig +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/lifecycle.py b/python/databricks/bundles/model_serving_endpoints/_models/lifecycle.py index 6053b9dd0a1..697776a198e 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/lifecycle.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/lifecycle.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -79,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint.py b/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint.py index 3326583dda1..64ecfbd87d1 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint.py @@ -1,74 +1,53 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import ( + AiGatewayConfig, + AiGatewayConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.email_notifications import ( + EmailNotifications, + EmailNotificationsParam, +) +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import ( + EndpointCoreConfigInput, + EndpointCoreConfigInputParam, +) +from databricks.bundles.model_serving_endpoints._models.endpoint_tag import ( + EndpointTag, + EndpointTagParam, +) +from databricks.bundles.model_serving_endpoints._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ( + ModelServingEndpointPermission, + ModelServingEndpointPermissionParam, +) +from databricks.bundles.model_serving_endpoints._models.rate_limit import ( + RateLimit, + RateLimitParam, +) +from databricks.bundles.model_serving_endpoints._models.telemetry_config import ( + TelemetryConfig, + TelemetryConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelServingEndpoint(Resource): """""" @@ -106,7 +85,9 @@ class ModelServingEndpoint(Resource): Settings that control the deployment lifecycle of the resource, such as preventing it from being destroyed. """ - permissions: VariableOrList[ModelServingEndpointPermission] = field(default_factory=list) + permissions: VariableOrList[ModelServingEndpointPermission] = field( + default_factory=list + ) """ The permissions to apply to this resource. """ @@ -132,12 +113,11 @@ class ModelServingEndpoint(Resource): """ @classmethod - def from_dict(cls, value: 'ModelServingEndpointDict') -> 'Self': + def from_dict(cls, value: "ModelServingEndpointDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServingEndpointDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServingEndpointDict": + return _transform_to_json_value(self) # type:ignore class ModelServingEndpointDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint_permission.py b/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint_permission.py index 93522ed346b..82e2fdc054a 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint_permission.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/model_serving_endpoint_permission.py @@ -1,74 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ( + ServingEndpointPermissionLevel, + ServingEndpointPermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelServingEndpointPermission: """""" @@ -94,12 +40,11 @@ class ModelServingEndpointPermission: """ @classmethod - def from_dict(cls, value: 'ModelServingEndpointPermissionDict') -> 'Self': + def from_dict(cls, value: "ModelServingEndpointPermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelServingEndpointPermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelServingEndpointPermissionDict": + return _transform_to_json_value(self) # type:ignore class ModelServingEndpointPermissionDict(TypedDict, total=False): @@ -126,4 +71,6 @@ class ModelServingEndpointPermissionDict(TypedDict, total=False): """ -ModelServingEndpointPermissionParam = ModelServingEndpointPermissionDict | ModelServingEndpointPermission +ModelServingEndpointPermissionParam = ( + ModelServingEndpointPermissionDict | ModelServingEndpointPermission +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/open_ai_config.py b/python/databricks/bundles/model_serving_endpoints/_models/open_ai_config.py index a371860d43b..e19a4eac665 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/open_ai_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/open_ai_config.py @@ -1,79 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class OpenAiConfig: """ Configs needed to create an OpenAI model route. """ + microsoft_entra_client_id: VariableOrOptional[str] = None """ This field is only required for Azure AD OpenAI and is the Microsoft @@ -159,12 +102,11 @@ class OpenAiConfig: """ @classmethod - def from_dict(cls, value: 'OpenAiConfigDict') -> 'Self': + def from_dict(cls, value: "OpenAiConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'OpenAiConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "OpenAiConfigDict": + return _transform_to_json_value(self) # type:ignore class OpenAiConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/pa_lm_config.py b/python/databricks/bundles/model_serving_endpoints/_models/pa_lm_config.py index ce7e0a884d6..98e5c572687 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/pa_lm_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/pa_lm_config.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PaLmConfig: """""" @@ -90,12 +32,11 @@ class PaLmConfig: """ @classmethod - def from_dict(cls, value: 'PaLmConfigDict') -> 'Self': + def from_dict(cls, value: "PaLmConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PaLmConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PaLmConfigDict": + return _transform_to_json_value(self) # type:ignore class PaLmConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit.py b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit.py index 57c58ea9d20..5a0192df8ce 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit.py @@ -1,79 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.rate_limit_key import ( + RateLimitKey, + RateLimitKeyParam, +) +from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import ( + RateLimitRenewalPeriod, + RateLimitRenewalPeriodParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RateLimit: """ - [DEPRECATED] + [DEPRECATED] """ + calls: VariableOr[int] """ Used to specify how many calls are allowed for a key within the renewal_period. @@ -90,12 +41,11 @@ class RateLimit: """ @classmethod - def from_dict(cls, value: 'RateLimitDict') -> 'Self': + def from_dict(cls, value: "RateLimitDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RateLimitDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RateLimitDict": + return _transform_to_json_value(self) # type:ignore class RateLimitDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_key.py b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_key.py index 4c329bcc14a..b6ebe9db1c4 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_key.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_key.py @@ -1,75 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class RateLimitKey(Enum): """ - [DEPRECATED] + [DEPRECATED] """ + USER = "user" ENDPOINT = "endpoint" + RateLimitKeyParam = Literal["user", "endpoint"] | RateLimitKey diff --git a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_renewal_period.py b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_renewal_period.py index 4ec5ff8cf02..81f900bf39f 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_renewal_period.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/rate_limit_renewal_period.py @@ -1,74 +1,15 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class RateLimitRenewalPeriod(Enum): """ - [DEPRECATED] + [DEPRECATED] """ + MINUTE = "minute" + RateLimitRenewalPeriodParam = Literal["minute"] | RateLimitRenewalPeriod diff --git a/python/databricks/bundles/model_serving_endpoints/_models/route.py b/python/databricks/bundles/model_serving_endpoints/_models/route.py index 3796abc206f..eba4336b0fc 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/route.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/route.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Route: """""" @@ -86,12 +28,11 @@ class Route: """ @classmethod - def from_dict(cls, value: 'RouteDict') -> 'Self': + def from_dict(cls, value: "RouteDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RouteDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RouteDict": + return _transform_to_json_value(self) # type:ignore class RouteDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/served_entity_input.py b/python/databricks/bundles/model_serving_endpoints/_models/served_entity_input.py index 1a8208b162d..e40050ca3e6 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/served_entity_input.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/served_entity_input.py @@ -1,74 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrDict, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.external_model import ( + ExternalModel, + ExternalModelParam, +) +from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ( + ServingModelWorkloadType, + ServingModelWorkloadTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ServedEntityInput: """""" @@ -148,12 +98,11 @@ class ServedEntityInput: """ @classmethod - def from_dict(cls, value: 'ServedEntityInputDict') -> 'Self': + def from_dict(cls, value: "ServedEntityInputDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ServedEntityInputDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ServedEntityInputDict": + return _transform_to_json_value(self) # type:ignore class ServedEntityInputDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/served_model_input.py b/python/databricks/bundles/model_serving_endpoints/_models/served_model_input.py index 31a317ec704..90c98757af5 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/served_model_input.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/served_model_input.py @@ -1,74 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrOptional, +) +from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ( + ServedModelInputWorkloadType, + ServedModelInputWorkloadTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ServedModelInput: """""" @@ -140,12 +90,11 @@ class ServedModelInput: """ @classmethod - def from_dict(cls, value: 'ServedModelInputDict') -> 'Self': + def from_dict(cls, value: "ServedModelInputDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ServedModelInputDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ServedModelInputDict": + return _transform_to_json_value(self) # type:ignore class ServedModelInputDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/served_model_input_workload_type.py b/python/databricks/bundles/model_serving_endpoints/_models/served_model_input_workload_type.py index c709f6185cb..87f62c41e34 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/served_model_input_workload_type.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/served_model_input_workload_type.py @@ -1,74 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class ServedModelInputWorkloadType(Enum): """ Please keep this in sync with workload types in InferenceEndpointEntities.scala. """ + CPU = "CPU" GPU_MEDIUM = "GPU_MEDIUM" GPU_SMALL = "GPU_SMALL" @@ -79,4 +19,18 @@ class ServedModelInputWorkloadType(Enum): GPU_XLARGE = "GPU_XLARGE" CPU_MEDIUM = "CPU_MEDIUM" -ServedModelInputWorkloadTypeParam = Literal["CPU", "GPU_MEDIUM", "GPU_SMALL", "GPU_LARGE", "MULTIGPU_MEDIUM", "CPU_LARGE", "GPU_XLARGE_8", "GPU_XLARGE", "CPU_MEDIUM"] | ServedModelInputWorkloadType + +ServedModelInputWorkloadTypeParam = ( + Literal[ + "CPU", + "GPU_MEDIUM", + "GPU_SMALL", + "GPU_LARGE", + "MULTIGPU_MEDIUM", + "CPU_LARGE", + "GPU_XLARGE_8", + "GPU_XLARGE", + "CPU_MEDIUM", + ] + | ServedModelInputWorkloadType +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/serving_endpoint_permission_level.py b/python/databricks/bundles/model_serving_endpoints/_models/serving_endpoint_permission_level.py index 1cf68783544..d9d75992c54 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/serving_endpoint_permission_level.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/serving_endpoint_permission_level.py @@ -1,76 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class ServingEndpointPermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_QUERY = "CAN_QUERY" CAN_VIEW = "CAN_VIEW" -ServingEndpointPermissionLevelParam = Literal["CAN_MANAGE", "CAN_QUERY", "CAN_VIEW"] | ServingEndpointPermissionLevel + +ServingEndpointPermissionLevelParam = ( + Literal["CAN_MANAGE", "CAN_QUERY", "CAN_VIEW"] | ServingEndpointPermissionLevel +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/serving_model_workload_type.py b/python/databricks/bundles/model_serving_endpoints/_models/serving_model_workload_type.py index ded41bbf800..04b513e77a7 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/serving_model_workload_type.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/serving_model_workload_type.py @@ -1,74 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam +from typing import Literal class ServingModelWorkloadType(Enum): """ Please keep this in sync with workload types in InferenceEndpointEntities.scala. """ + CPU = "CPU" GPU_MEDIUM = "GPU_MEDIUM" GPU_SMALL = "GPU_SMALL" @@ -79,4 +19,18 @@ class ServingModelWorkloadType(Enum): GPU_XLARGE = "GPU_XLARGE" CPU_MEDIUM = "CPU_MEDIUM" -ServingModelWorkloadTypeParam = Literal["CPU", "GPU_MEDIUM", "GPU_SMALL", "GPU_LARGE", "MULTIGPU_MEDIUM", "CPU_LARGE", "GPU_XLARGE_8", "GPU_XLARGE", "CPU_MEDIUM"] | ServingModelWorkloadType + +ServingModelWorkloadTypeParam = ( + Literal[ + "CPU", + "GPU_MEDIUM", + "GPU_SMALL", + "GPU_LARGE", + "MULTIGPU_MEDIUM", + "CPU_LARGE", + "GPU_XLARGE_8", + "GPU_XLARGE", + "CPU_MEDIUM", + ] + | ServingModelWorkloadType +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_config.py b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_config.py index db4a4a68780..cffb10c2642 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_config.py @@ -1,79 +1,35 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.model_serving_endpoints._models.telemetry_feature import ( + TelemetryFeature, + TelemetryFeatureParam, +) +from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import ( + TelemetryInferenceTableConfig, + TelemetryInferenceTableConfigParam, +) +from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import ( + UnityCatalogTableNames, + UnityCatalogTableNamesParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TelemetryConfig: """""" - enabled_telemetry_features: VariableOrList[TelemetryFeature] = field(default_factory=list) + enabled_telemetry_features: VariableOrList[TelemetryFeature] = field( + default_factory=list + ) """ [Public Preview] The telemetry signals to enable for this endpoint. If empty or omitted, all signals are enabled; otherwise only the listed signals are enabled. @@ -97,12 +53,11 @@ class TelemetryConfig: """ @classmethod - def from_dict(cls, value: 'TelemetryConfigDict') -> 'Self': + def from_dict(cls, value: "TelemetryConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TelemetryConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TelemetryConfigDict": + return _transform_to_json_value(self) # type:ignore class TelemetryConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_feature.py b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_feature.py index 32629aeca9d..2d74f666c07 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_feature.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_feature.py @@ -1,68 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam +from typing import Literal class TelemetryFeature(Enum): @@ -70,9 +9,19 @@ class TelemetryFeature(Enum): A telemetry signal that a serving endpoint can export to Unity Catalog. Use these values to select which signals the endpoint exports. """ + TELEMETRY_FEATURE_LOGS = "TELEMETRY_FEATURE_LOGS" TELEMETRY_FEATURE_TRACES = "TELEMETRY_FEATURE_TRACES" TELEMETRY_FEATURE_METRICS = "TELEMETRY_FEATURE_METRICS" TELEMETRY_FEATURE_INFERENCE_TABLE = "TELEMETRY_FEATURE_INFERENCE_TABLE" -TelemetryFeatureParam = Literal["TELEMETRY_FEATURE_LOGS", "TELEMETRY_FEATURE_TRACES", "TELEMETRY_FEATURE_METRICS", "TELEMETRY_FEATURE_INFERENCE_TABLE"] | TelemetryFeature + +TelemetryFeatureParam = ( + Literal[ + "TELEMETRY_FEATURE_LOGS", + "TELEMETRY_FEATURE_TRACES", + "TELEMETRY_FEATURE_METRICS", + "TELEMETRY_FEATURE_INFERENCE_TABLE", + ] + | TelemetryFeature +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_inference_table_config.py b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_inference_table_config.py index 44dc8bee8c5..9153a4c49ac 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/telemetry_inference_table_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/telemetry_inference_table_config.py @@ -1,91 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TelemetryInferenceTableConfig: """ Inference table payload logging configuration """ + sampling_fraction: VariableOrOptional[float] = None """ [Public Preview] Fraction of requests sampled for payload logging, in the range [0.0, 1.0], where 1.0 logs all requests. """ @classmethod - def from_dict(cls, value: 'TelemetryInferenceTableConfigDict') -> 'Self': + def from_dict(cls, value: "TelemetryInferenceTableConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TelemetryInferenceTableConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TelemetryInferenceTableConfigDict": + return _transform_to_json_value(self) # type:ignore class TelemetryInferenceTableConfigDict(TypedDict, total=False): @@ -97,4 +39,6 @@ class TelemetryInferenceTableConfigDict(TypedDict, total=False): """ -TelemetryInferenceTableConfigParam = TelemetryInferenceTableConfigDict | TelemetryInferenceTableConfig +TelemetryInferenceTableConfigParam = ( + TelemetryInferenceTableConfigDict | TelemetryInferenceTableConfig +) diff --git a/python/databricks/bundles/model_serving_endpoints/_models/traffic_config.py b/python/databricks/bundles/model_serving_endpoints/_models/traffic_config.py index b4215ab2c29..d25a5c87fef 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/traffic_config.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/traffic_config.py @@ -1,74 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.unity_catalog_table_names import UnityCatalogTableNames, UnityCatalogTableNamesDict, UnityCatalogTableNamesParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.model_serving_endpoints._models.route import ( + Route, + RouteParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TrafficConfig: """""" @@ -79,12 +25,11 @@ class TrafficConfig: """ @classmethod - def from_dict(cls, value: 'TrafficConfigDict') -> 'Self': + def from_dict(cls, value: "TrafficConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TrafficConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TrafficConfigDict": + return _transform_to_json_value(self) # type:ignore class TrafficConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/model_serving_endpoints/_models/unity_catalog_table_names.py b/python/databricks/bundles/model_serving_endpoints/_models/unity_catalog_table_names.py index 3a6dcbbfb75..5cf8a32d821 100644 --- a/python/databricks/bundles/model_serving_endpoints/_models/unity_catalog_table_names.py +++ b/python/databricks/bundles/model_serving_endpoints/_models/unity_catalog_table_names.py @@ -1,74 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.model_serving_endpoints._models.ai21_labs_config import Ai21LabsConfig, Ai21LabsConfigDict, Ai21LabsConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig, AiGatewayConfigDict, AiGatewayConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_parameters import AiGatewayGuardrailParameters, AiGatewayGuardrailParametersDict, AiGatewayGuardrailParametersParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior import AiGatewayGuardrailPiiBehavior, AiGatewayGuardrailPiiBehaviorDict, AiGatewayGuardrailPiiBehaviorParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrails import AiGatewayGuardrails, AiGatewayGuardrailsDict, AiGatewayGuardrailsParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_inference_table_config import AiGatewayInferenceTableConfig, AiGatewayInferenceTableConfigDict, AiGatewayInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit import AiGatewayRateLimit, AiGatewayRateLimitDict, AiGatewayRateLimitParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_usage_tracking_config import AiGatewayUsageTrackingConfig, AiGatewayUsageTrackingConfigDict, AiGatewayUsageTrackingConfigParam -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config import AmazonBedrockConfig, AmazonBedrockConfigDict, AmazonBedrockConfigParam -from databricks.bundles.model_serving_endpoints._models.anthropic_config import AnthropicConfig, AnthropicConfigDict, AnthropicConfigParam -from databricks.bundles.model_serving_endpoints._models.api_key_auth import ApiKeyAuth, ApiKeyAuthDict, ApiKeyAuthParam -from databricks.bundles.model_serving_endpoints._models.auto_capture_config_input import AutoCaptureConfigInput, AutoCaptureConfigInputDict, AutoCaptureConfigInputParam -from databricks.bundles.model_serving_endpoints._models.bearer_token_auth import BearerTokenAuth, BearerTokenAuthDict, BearerTokenAuthParam -from databricks.bundles.model_serving_endpoints._models.cohere_config import CohereConfig, CohereConfigDict, CohereConfigParam -from databricks.bundles.model_serving_endpoints._models.custom_provider_config import CustomProviderConfig, CustomProviderConfigDict, CustomProviderConfigParam -from databricks.bundles.model_serving_endpoints._models.databricks_model_serving_config import DatabricksModelServingConfig, DatabricksModelServingConfigDict, DatabricksModelServingConfigParam -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications, EmailNotificationsDict, EmailNotificationsParam -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput, EndpointCoreConfigInputDict, EndpointCoreConfigInputParam -from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag, EndpointTagDict, EndpointTagParam -from databricks.bundles.model_serving_endpoints._models.external_model import ExternalModel, ExternalModelDict, ExternalModelParam -from databricks.bundles.model_serving_endpoints._models.fallback_config import FallbackConfig, FallbackConfigDict, FallbackConfigParam -from databricks.bundles.model_serving_endpoints._models.google_cloud_vertex_ai_config import GoogleCloudVertexAiConfig, GoogleCloudVertexAiConfigDict, GoogleCloudVertexAiConfigParam -from databricks.bundles.model_serving_endpoints._models.open_ai_config import OpenAiConfig, OpenAiConfigDict, OpenAiConfigParam -from databricks.bundles.model_serving_endpoints._models.pa_lm_config import PaLmConfig, PaLmConfigDict, PaLmConfigParam -from databricks.bundles.model_serving_endpoints._models.rate_limit import RateLimit, RateLimitDict, RateLimitParam -from databricks.bundles.model_serving_endpoints._models.route import Route, RouteDict, RouteParam -from databricks.bundles.model_serving_endpoints._models.served_entity_input import ServedEntityInput, ServedEntityInputDict, ServedEntityInputParam -from databricks.bundles.model_serving_endpoints._models.served_model_input import ServedModelInput, ServedModelInputDict, ServedModelInputParam -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig, TelemetryConfigDict, TelemetryConfigParam -from databricks.bundles.model_serving_endpoints._models.telemetry_inference_table_config import TelemetryInferenceTableConfig, TelemetryInferenceTableConfigDict, TelemetryInferenceTableConfigParam -from databricks.bundles.model_serving_endpoints._models.traffic_config import TrafficConfig, TrafficConfigDict, TrafficConfigParam -from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint, ModelServingEndpointDict, ModelServingEndpointParam -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission, ModelServingEndpointPermissionDict, ModelServingEndpointPermissionParam -from databricks.bundles.model_serving_endpoints._models.ai_gateway_guardrail_pii_behavior_behavior import AiGatewayGuardrailPiiBehaviorBehavior, AiGatewayGuardrailPiiBehaviorBehaviorParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_key import AiGatewayRateLimitKey, AiGatewayRateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.ai_gateway_rate_limit_renewal_period import AiGatewayRateLimitRenewalPeriod, AiGatewayRateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.amazon_bedrock_config_bedrock_provider import AmazonBedrockConfigBedrockProvider, AmazonBedrockConfigBedrockProviderParam - -from databricks.bundles.model_serving_endpoints._models.external_model_provider import ExternalModelProvider, ExternalModelProviderParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_key import RateLimitKey, RateLimitKeyParam - -from databricks.bundles.model_serving_endpoints._models.rate_limit_renewal_period import RateLimitRenewalPeriod, RateLimitRenewalPeriodParam - -from databricks.bundles.model_serving_endpoints._models.served_model_input_workload_type import ServedModelInputWorkloadType, ServedModelInputWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel, ServingEndpointPermissionLevelParam - -from databricks.bundles.model_serving_endpoints._models.serving_model_workload_type import ServingModelWorkloadType, ServingModelWorkloadTypeParam - -from databricks.bundles.model_serving_endpoints._models.telemetry_feature import TelemetryFeature, TelemetryFeatureParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class UnityCatalogTableNames: """""" @@ -98,12 +40,11 @@ class UnityCatalogTableNames: """ @classmethod - def from_dict(cls, value: 'UnityCatalogTableNamesDict') -> 'Self': + def from_dict(cls, value: "UnityCatalogTableNamesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'UnityCatalogTableNamesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "UnityCatalogTableNamesDict": + return _transform_to_json_value(self) # type:ignore class UnityCatalogTableNamesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/__init__.py b/python/databricks/bundles/models/__init__.py index 673f479c47a..a3469f1689e 100644 --- a/python/databricks/bundles/models/__init__.py +++ b/python/databricks/bundles/models/__init__.py @@ -1,26 +1,44 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'MlflowModel', - 'MlflowModelDict', - 'MlflowModelParam', - 'MlflowModelPermission', - 'MlflowModelPermissionDict', - 'MlflowModelPermissionParam', - 'ModelTag', - 'ModelTagDict', - 'ModelTagParam', - 'RegisteredModelPermissionLevel', - 'RegisteredModelPermissionLevelParam', + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "MlflowModel", + "MlflowModelDict", + "MlflowModelParam", + "MlflowModelPermission", + "MlflowModelPermissionDict", + "MlflowModelPermissionParam", + "ModelTag", + "ModelTagDict", + "ModelTagParam", + "RegisteredModelPermissionLevel", + "RegisteredModelPermissionLevelParam", ] -from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam -from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam -from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam -from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam - +from databricks.bundles.models._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.models._models.mlflow_model import ( + MlflowModel, + MlflowModelDict, + MlflowModelParam, +) +from databricks.bundles.models._models.mlflow_model_permission import ( + MlflowModelPermission, + MlflowModelPermissionDict, + MlflowModelPermissionParam, +) +from databricks.bundles.models._models.model_tag import ( + ModelTag, + ModelTagDict, + ModelTagParam, +) +from databricks.bundles.models._models.registered_model_permission_level import ( + RegisteredModelPermissionLevel, + RegisteredModelPermissionLevelParam, +) diff --git a/python/databricks/bundles/models/_models/lifecycle.py b/python/databricks/bundles/models/_models/lifecycle.py index c174e277c77..697776a198e 100644 --- a/python/databricks/bundles/models/_models/lifecycle.py +++ b/python/databricks/bundles/models/_models/lifecycle.py @@ -1,23 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam -from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam -from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam -from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -28,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/_models/mlflow_model.py b/python/databricks/bundles/models/_models/mlflow_model.py index e3242dbde2b..390f1155f2c 100644 --- a/python/databricks/bundles/models/_models/mlflow_model.py +++ b/python/databricks/bundles/models/_models/mlflow_model.py @@ -1,23 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam -from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam -from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.models._models.mlflow_model_permission import ( + MlflowModelPermission, + MlflowModelPermissionParam, +) +from databricks.bundles.models._models.model_tag import ModelTag, ModelTagParam if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MlflowModel(Resource): """""" @@ -48,12 +52,11 @@ class MlflowModel(Resource): """ @classmethod - def from_dict(cls, value: 'MlflowModelDict') -> 'Self': + def from_dict(cls, value: "MlflowModelDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MlflowModelDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MlflowModelDict": + return _transform_to_json_value(self) # type:ignore class MlflowModelDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/_models/mlflow_model_permission.py b/python/databricks/bundles/models/_models/mlflow_model_permission.py index 9163445ec6f..367cbfd4a3a 100644 --- a/python/databricks/bundles/models/_models/mlflow_model_permission.py +++ b/python/databricks/bundles/models/_models/mlflow_model_permission.py @@ -1,23 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam -from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam -from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.models._models.registered_model_permission_level import ( + RegisteredModelPermissionLevel, + RegisteredModelPermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MlflowModelPermission: """""" @@ -43,12 +40,11 @@ class MlflowModelPermission: """ @classmethod - def from_dict(cls, value: 'MlflowModelPermissionDict') -> 'Self': + def from_dict(cls, value: "MlflowModelPermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MlflowModelPermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MlflowModelPermissionDict": + return _transform_to_json_value(self) # type:ignore class MlflowModelPermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/_models/model_tag.py b/python/databricks/bundles/models/_models/model_tag.py index 98b9885ae7c..b434ad05552 100644 --- a/python/databricks/bundles/models/_models/model_tag.py +++ b/python/databricks/bundles/models/_models/model_tag.py @@ -1,28 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam -from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam -from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel, RegisteredModelPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ModelTag: """ Tag for a registered model """ + key: VariableOrOptional[str] = None """ The tag key. @@ -34,12 +28,11 @@ class ModelTag: """ @classmethod - def from_dict(cls, value: 'ModelTagDict') -> 'Self': + def from_dict(cls, value: "ModelTagDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ModelTagDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ModelTagDict": + return _transform_to_json_value(self) # type:ignore class ModelTagDict(TypedDict, total=False): diff --git a/python/databricks/bundles/models/_models/registered_model_permission_level.py b/python/databricks/bundles/models/_models/registered_model_permission_level.py index 56e37bb7ab9..1344b5d5e48 100644 --- a/python/databricks/bundles/models/_models/registered_model_permission_level.py +++ b/python/databricks/bundles/models/_models/registered_model_permission_level.py @@ -1,27 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.models._models.model_tag import ModelTag, ModelTagDict, ModelTagParam -from databricks.bundles.models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.models._models.mlflow_model import MlflowModel, MlflowModelDict, MlflowModelParam -from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission, MlflowModelPermissionDict, MlflowModelPermissionParam class RegisteredModelPermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" CAN_MANAGE_PRODUCTION_VERSIONS = "CAN_MANAGE_PRODUCTION_VERSIONS" CAN_MANAGE_STAGING_VERSIONS = "CAN_MANAGE_STAGING_VERSIONS" CAN_EDIT = "CAN_EDIT" CAN_READ = "CAN_READ" -RegisteredModelPermissionLevelParam = Literal["CAN_MANAGE", "CAN_MANAGE_PRODUCTION_VERSIONS", "CAN_MANAGE_STAGING_VERSIONS", "CAN_EDIT", "CAN_READ"] | RegisteredModelPermissionLevel + +RegisteredModelPermissionLevelParam = ( + Literal[ + "CAN_MANAGE", + "CAN_MANAGE_PRODUCTION_VERSIONS", + "CAN_MANAGE_STAGING_VERSIONS", + "CAN_EDIT", + "CAN_READ", + ] + | RegisteredModelPermissionLevel +) diff --git a/python/databricks/bundles/pipelines/__init__.py b/python/databricks/bundles/pipelines/__init__.py index 34861c3afac..2b66e4f795c 100644 --- a/python/databricks/bundles/pipelines/__init__.py +++ b/python/databricks/bundles/pipelines/__init__.py @@ -1,414 +1,766 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Adlsgen2Info', - 'Adlsgen2InfoDict', - 'Adlsgen2InfoParam', - 'ApiSourceConnectorConfig', - 'ApiSourceConnectorConfigDict', - 'ApiSourceConnectorConfigParam', - 'ApiSourceConnectorOptions', - 'ApiSourceConnectorOptionsDict', - 'ApiSourceConnectorOptionsParam', - 'AutoFullRefreshPolicy', - 'AutoFullRefreshPolicyDict', - 'AutoFullRefreshPolicyParam', - 'AvroTransformerOptions', - 'AvroTransformerOptionsDict', - 'AvroTransformerOptionsParam', - 'AwsAttributes', - 'AwsAttributesDict', - 'AwsAttributesParam', - 'AwsAvailability', - 'AwsAvailabilityParam', - 'AzureAttributes', - 'AzureAttributesDict', - 'AzureAttributesParam', - 'AzureAvailability', - 'AzureAvailabilityParam', - 'ClusterLogConf', - 'ClusterLogConfDict', - 'ClusterLogConfParam', - 'ConfidentialComputeType', - 'ConfidentialComputeTypeParam', - 'ConfluenceConnectorOptions', - 'ConfluenceConnectorOptionsDict', - 'ConfluenceConnectorOptionsParam', - 'ConfluentSchemaRegistryOptions', - 'ConfluentSchemaRegistryOptionsDict', - 'ConfluentSchemaRegistryOptionsParam', - 'ConnectionParameters', - 'ConnectionParametersDict', - 'ConnectionParametersParam', - 'ConnectorOptions', - 'ConnectorOptionsDict', - 'ConnectorOptionsParam', - 'ConnectorType', - 'ConnectorTypeParam', - 'DataStagingOptions', - 'DataStagingOptionsDict', - 'DataStagingOptionsParam', - 'DayOfWeek', - 'DayOfWeekParam', - 'DbfsStorageInfo', - 'DbfsStorageInfoDict', - 'DbfsStorageInfoParam', - 'EbsVolumeType', - 'EbsVolumeTypeParam', - 'EventLogSpec', - 'EventLogSpecDict', - 'EventLogSpecParam', - 'FileFilter', - 'FileFilterDict', - 'FileFilterParam', - 'FileIngestionOptions', - 'FileIngestionOptionsDict', - 'FileIngestionOptionsFileFormat', - 'FileIngestionOptionsFileFormatParam', - 'FileIngestionOptionsParam', - 'FileIngestionOptionsSchemaEvolutionMode', - 'FileIngestionOptionsSchemaEvolutionModeParam', - 'FileLibrary', - 'FileLibraryDict', - 'FileLibraryParam', - 'Filters', - 'FiltersDict', - 'FiltersParam', - 'GcpAttributes', - 'GcpAttributesDict', - 'GcpAttributesParam', - 'GcpAvailability', - 'GcpAvailabilityParam', - 'GcsStorageInfo', - 'GcsStorageInfoDict', - 'GcsStorageInfoParam', - 'GoogleAdsConfig', - 'GoogleAdsConfigDict', - 'GoogleAdsConfigParam', - 'GoogleAdsCustomReportOptions', - 'GoogleAdsCustomReportOptionsDict', - 'GoogleAdsCustomReportOptionsParam', - 'GoogleAdsOptions', - 'GoogleAdsOptionsDict', - 'GoogleAdsOptionsParam', - 'GoogleDriveOptions', - 'GoogleDriveOptionsDict', - 'GoogleDriveOptionsGoogleDriveEntityType', - 'GoogleDriveOptionsGoogleDriveEntityTypeParam', - 'GoogleDriveOptionsParam', - 'IngestionConfig', - 'IngestionConfigDict', - 'IngestionConfigParam', - 'IngestionGatewayPipelineDefinition', - 'IngestionGatewayPipelineDefinitionDict', - 'IngestionGatewayPipelineDefinitionParam', - 'IngestionPipelineDefinition', - 'IngestionPipelineDefinitionDict', - 'IngestionPipelineDefinitionFanoutOptions', - 'IngestionPipelineDefinitionFanoutOptionsDict', - 'IngestionPipelineDefinitionFanoutOptionsParam', - 'IngestionPipelineDefinitionParam', - 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig', - 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict', - 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam', - 'IngestionPipelineDefinitionWorkdayReportParameters', - 'IngestionPipelineDefinitionWorkdayReportParametersDict', - 'IngestionPipelineDefinitionWorkdayReportParametersParam', - 'InitScriptInfo', - 'InitScriptInfoDict', - 'InitScriptInfoParam', - 'JiraConnectorOptions', - 'JiraConnectorOptionsDict', - 'JiraConnectorOptionsParam', - 'JsonTransformerOptions', - 'JsonTransformerOptionsDict', - 'JsonTransformerOptionsParam', - 'KafkaOptions', - 'KafkaOptionsDict', - 'KafkaOptionsParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'LinkedInAdsOptions', - 'LinkedInAdsOptionsDict', - 'LinkedInAdsOptionsLinkedInAdsCustomReportOptions', - 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict', - 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity', - 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam', - 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder', - 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam', - 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity', - 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam', - 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam', - 'LinkedInAdsOptionsParam', - 'LocalFileInfo', - 'LocalFileInfoDict', - 'LocalFileInfoParam', - 'LogAnalyticsInfo', - 'LogAnalyticsInfoDict', - 'LogAnalyticsInfoParam', - 'MarketoOptions', - 'MarketoOptionsDict', - 'MarketoOptionsParam', - 'MavenLibrary', - 'MavenLibraryDict', - 'MavenLibraryParam', - 'MetaMarketingOptions', - 'MetaMarketingOptionsDict', - 'MetaMarketingOptionsMetaMarketingCustomReportOptions', - 'MetaMarketingOptionsMetaMarketingCustomReportOptionsDict', - 'MetaMarketingOptionsMetaMarketingCustomReportOptionsParam', - 'MetaMarketingOptionsParam', - 'NotebookLibrary', - 'NotebookLibraryDict', - 'NotebookLibraryParam', - 'Notifications', - 'NotificationsDict', - 'NotificationsParam', - 'OperationTimeWindow', - 'OperationTimeWindowDict', - 'OperationTimeWindowParam', - 'OutlookAttachmentMode', - 'OutlookAttachmentModeParam', - 'OutlookBodyFormat', - 'OutlookBodyFormatParam', - 'OutlookOptions', - 'OutlookOptionsDict', - 'OutlookOptionsParam', - 'ParseMode', - 'ParseModeParam', - 'PathPattern', - 'PathPatternDict', - 'PathPatternParam', - 'Pipeline', - 'PipelineCluster', - 'PipelineClusterAutoscale', - 'PipelineClusterAutoscaleDict', - 'PipelineClusterAutoscaleMode', - 'PipelineClusterAutoscaleModeParam', - 'PipelineClusterAutoscaleParam', - 'PipelineClusterDict', - 'PipelineClusterParam', - 'PipelineDict', - 'PipelineLibrary', - 'PipelineLibraryDict', - 'PipelineLibraryParam', - 'PipelineParam', - 'PipelinePermission', - 'PipelinePermissionDict', - 'PipelinePermissionLevel', - 'PipelinePermissionLevelParam', - 'PipelinePermissionParam', - 'PipelinesEnvironment', - 'PipelinesEnvironmentDict', - 'PipelinesEnvironmentParam', - 'PostgresCatalogConfig', - 'PostgresCatalogConfigDict', - 'PostgresCatalogConfigParam', - 'PostgresSlotConfig', - 'PostgresSlotConfigDict', - 'PostgresSlotConfigParam', - 'ProtobufTransformerOptions', - 'ProtobufTransformerOptionsDict', - 'ProtobufTransformerOptionsParam', - 'RabbitmqOptions', - 'RabbitmqOptionsDict', - 'RabbitmqOptionsParam', - 'RedditAdsOptions', - 'RedditAdsOptionsDict', - 'RedditAdsOptionsParam', - 'RedditAdsOptionsRedditAdsCustomReportOptions', - 'RedditAdsOptionsRedditAdsCustomReportOptionsDict', - 'RedditAdsOptionsRedditAdsCustomReportOptionsParam', - 'ReportSpec', - 'ReportSpecDict', - 'ReportSpecParam', - 'RestartWindow', - 'RestartWindowDict', - 'RestartWindowParam', - 'RunAs', - 'RunAsDict', - 'RunAsParam', - 'S3StorageInfo', - 'S3StorageInfoDict', - 'S3StorageInfoParam', - 'SchemaRegistryConfig', - 'SchemaRegistryConfigDict', - 'SchemaRegistryConfigParam', - 'SchemaSpec', - 'SchemaSpecDict', - 'SchemaSpecParam', - 'SharepointOptions', - 'SharepointOptionsDict', - 'SharepointOptionsParam', - 'SharepointOptionsSharepointEntityType', - 'SharepointOptionsSharepointEntityTypeParam', - 'SmartsheetOptions', - 'SmartsheetOptionsDict', - 'SmartsheetOptionsParam', - 'SourceCatalogConfig', - 'SourceCatalogConfigDict', - 'SourceCatalogConfigParam', - 'SourceConfig', - 'SourceConfigDict', - 'SourceConfigParam', - 'TableSpec', - 'TableSpecDict', - 'TableSpecParam', - 'TableSpecificConfig', - 'TableSpecificConfigDict', - 'TableSpecificConfigParam', - 'TableSpecificConfigScdType', - 'TableSpecificConfigScdTypeParam', - 'TikTokAdsOptions', - 'TikTokAdsOptionsDict', - 'TikTokAdsOptionsParam', - 'TikTokAdsOptionsTikTokAdsCustomReportOptions', - 'TikTokAdsOptionsTikTokAdsCustomReportOptionsDict', - 'TikTokAdsOptionsTikTokAdsCustomReportOptionsParam', - 'TikTokAdsOptionsTikTokDataLevel', - 'TikTokAdsOptionsTikTokDataLevelParam', - 'TikTokAdsOptionsTikTokReportType', - 'TikTokAdsOptionsTikTokReportTypeParam', - 'Transformer', - 'TransformerDict', - 'TransformerFormat', - 'TransformerFormatParam', - 'TransformerParam', - 'VolumesStorageInfo', - 'VolumesStorageInfoDict', - 'VolumesStorageInfoParam', - 'WorkspaceStorageInfo', - 'WorkspaceStorageInfoDict', - 'WorkspaceStorageInfoParam', - 'ZendeskSupportOptions', - 'ZendeskSupportOptionsDict', - 'ZendeskSupportOptionsParam', + "Adlsgen2Info", + "Adlsgen2InfoDict", + "Adlsgen2InfoParam", + "ApiSourceConnectorConfig", + "ApiSourceConnectorConfigDict", + "ApiSourceConnectorConfigParam", + "ApiSourceConnectorOptions", + "ApiSourceConnectorOptionsDict", + "ApiSourceConnectorOptionsParam", + "AutoFullRefreshPolicy", + "AutoFullRefreshPolicyDict", + "AutoFullRefreshPolicyParam", + "AvroTransformerOptions", + "AvroTransformerOptionsDict", + "AvroTransformerOptionsParam", + "AwsAttributes", + "AwsAttributesDict", + "AwsAttributesParam", + "AwsAvailability", + "AwsAvailabilityParam", + "AzureAttributes", + "AzureAttributesDict", + "AzureAttributesParam", + "AzureAvailability", + "AzureAvailabilityParam", + "ClusterLogConf", + "ClusterLogConfDict", + "ClusterLogConfParam", + "ConfidentialComputeType", + "ConfidentialComputeTypeParam", + "ConfluenceConnectorOptions", + "ConfluenceConnectorOptionsDict", + "ConfluenceConnectorOptionsParam", + "ConfluentSchemaRegistryOptions", + "ConfluentSchemaRegistryOptionsDict", + "ConfluentSchemaRegistryOptionsParam", + "ConnectionParameters", + "ConnectionParametersDict", + "ConnectionParametersParam", + "ConnectorOptions", + "ConnectorOptionsDict", + "ConnectorOptionsParam", + "ConnectorType", + "ConnectorTypeParam", + "DataStagingOptions", + "DataStagingOptionsDict", + "DataStagingOptionsParam", + "DayOfWeek", + "DayOfWeekParam", + "DbfsStorageInfo", + "DbfsStorageInfoDict", + "DbfsStorageInfoParam", + "EbsVolumeType", + "EbsVolumeTypeParam", + "EventLogSpec", + "EventLogSpecDict", + "EventLogSpecParam", + "FileFilter", + "FileFilterDict", + "FileFilterParam", + "FileIngestionOptions", + "FileIngestionOptionsDict", + "FileIngestionOptionsFileFormat", + "FileIngestionOptionsFileFormatParam", + "FileIngestionOptionsParam", + "FileIngestionOptionsSchemaEvolutionMode", + "FileIngestionOptionsSchemaEvolutionModeParam", + "FileLibrary", + "FileLibraryDict", + "FileLibraryParam", + "Filters", + "FiltersDict", + "FiltersParam", + "GcpAttributes", + "GcpAttributesDict", + "GcpAttributesParam", + "GcpAvailability", + "GcpAvailabilityParam", + "GcsStorageInfo", + "GcsStorageInfoDict", + "GcsStorageInfoParam", + "GoogleAdsConfig", + "GoogleAdsConfigDict", + "GoogleAdsConfigParam", + "GoogleAdsCustomReportOptions", + "GoogleAdsCustomReportOptionsDict", + "GoogleAdsCustomReportOptionsParam", + "GoogleAdsOptions", + "GoogleAdsOptionsDict", + "GoogleAdsOptionsParam", + "GoogleDriveOptions", + "GoogleDriveOptionsDict", + "GoogleDriveOptionsGoogleDriveEntityType", + "GoogleDriveOptionsGoogleDriveEntityTypeParam", + "GoogleDriveOptionsParam", + "IngestionConfig", + "IngestionConfigDict", + "IngestionConfigParam", + "IngestionGatewayPipelineDefinition", + "IngestionGatewayPipelineDefinitionDict", + "IngestionGatewayPipelineDefinitionParam", + "IngestionPipelineDefinition", + "IngestionPipelineDefinitionDict", + "IngestionPipelineDefinitionFanoutOptions", + "IngestionPipelineDefinitionFanoutOptionsDict", + "IngestionPipelineDefinitionFanoutOptionsParam", + "IngestionPipelineDefinitionParam", + "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig", + "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict", + "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam", + "IngestionPipelineDefinitionWorkdayReportParameters", + "IngestionPipelineDefinitionWorkdayReportParametersDict", + "IngestionPipelineDefinitionWorkdayReportParametersParam", + "InitScriptInfo", + "InitScriptInfoDict", + "InitScriptInfoParam", + "JiraConnectorOptions", + "JiraConnectorOptionsDict", + "JiraConnectorOptionsParam", + "JsonTransformerOptions", + "JsonTransformerOptionsDict", + "JsonTransformerOptionsParam", + "KafkaOptions", + "KafkaOptionsDict", + "KafkaOptionsParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "LinkedInAdsOptions", + "LinkedInAdsOptionsDict", + "LinkedInAdsOptionsLinkedInAdsCustomReportOptions", + "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict", + "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity", + "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam", + "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder", + "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam", + "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity", + "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam", + "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam", + "LinkedInAdsOptionsParam", + "LocalFileInfo", + "LocalFileInfoDict", + "LocalFileInfoParam", + "LogAnalyticsInfo", + "LogAnalyticsInfoDict", + "LogAnalyticsInfoParam", + "MarketoOptions", + "MarketoOptionsDict", + "MarketoOptionsParam", + "MavenLibrary", + "MavenLibraryDict", + "MavenLibraryParam", + "MetaMarketingOptions", + "MetaMarketingOptionsDict", + "MetaMarketingOptionsMetaMarketingCustomReportOptions", + "MetaMarketingOptionsMetaMarketingCustomReportOptionsDict", + "MetaMarketingOptionsMetaMarketingCustomReportOptionsParam", + "MetaMarketingOptionsParam", + "NotebookLibrary", + "NotebookLibraryDict", + "NotebookLibraryParam", + "Notifications", + "NotificationsDict", + "NotificationsParam", + "OperationTimeWindow", + "OperationTimeWindowDict", + "OperationTimeWindowParam", + "OutlookAttachmentMode", + "OutlookAttachmentModeParam", + "OutlookBodyFormat", + "OutlookBodyFormatParam", + "OutlookOptions", + "OutlookOptionsDict", + "OutlookOptionsParam", + "ParseMode", + "ParseModeParam", + "PathPattern", + "PathPatternDict", + "PathPatternParam", + "Pipeline", + "PipelineCluster", + "PipelineClusterAutoscale", + "PipelineClusterAutoscaleDict", + "PipelineClusterAutoscaleMode", + "PipelineClusterAutoscaleModeParam", + "PipelineClusterAutoscaleParam", + "PipelineClusterDict", + "PipelineClusterParam", + "PipelineDict", + "PipelineLibrary", + "PipelineLibraryDict", + "PipelineLibraryParam", + "PipelineParam", + "PipelinePermission", + "PipelinePermissionDict", + "PipelinePermissionLevel", + "PipelinePermissionLevelParam", + "PipelinePermissionParam", + "PipelinesEnvironment", + "PipelinesEnvironmentDict", + "PipelinesEnvironmentParam", + "PostgresCatalogConfig", + "PostgresCatalogConfigDict", + "PostgresCatalogConfigParam", + "PostgresSlotConfig", + "PostgresSlotConfigDict", + "PostgresSlotConfigParam", + "ProtobufTransformerOptions", + "ProtobufTransformerOptionsDict", + "ProtobufTransformerOptionsParam", + "RabbitmqOptions", + "RabbitmqOptionsDict", + "RabbitmqOptionsParam", + "RedditAdsOptions", + "RedditAdsOptionsDict", + "RedditAdsOptionsParam", + "RedditAdsOptionsRedditAdsCustomReportOptions", + "RedditAdsOptionsRedditAdsCustomReportOptionsDict", + "RedditAdsOptionsRedditAdsCustomReportOptionsParam", + "ReportSpec", + "ReportSpecDict", + "ReportSpecParam", + "RestartWindow", + "RestartWindowDict", + "RestartWindowParam", + "RunAs", + "RunAsDict", + "RunAsParam", + "S3StorageInfo", + "S3StorageInfoDict", + "S3StorageInfoParam", + "SchemaRegistryConfig", + "SchemaRegistryConfigDict", + "SchemaRegistryConfigParam", + "SchemaSpec", + "SchemaSpecDict", + "SchemaSpecParam", + "SharepointOptions", + "SharepointOptionsDict", + "SharepointOptionsParam", + "SharepointOptionsSharepointEntityType", + "SharepointOptionsSharepointEntityTypeParam", + "SmartsheetOptions", + "SmartsheetOptionsDict", + "SmartsheetOptionsParam", + "SourceCatalogConfig", + "SourceCatalogConfigDict", + "SourceCatalogConfigParam", + "SourceConfig", + "SourceConfigDict", + "SourceConfigParam", + "TableSpec", + "TableSpecDict", + "TableSpecParam", + "TableSpecificConfig", + "TableSpecificConfigDict", + "TableSpecificConfigParam", + "TableSpecificConfigScdType", + "TableSpecificConfigScdTypeParam", + "TikTokAdsOptions", + "TikTokAdsOptionsDict", + "TikTokAdsOptionsParam", + "TikTokAdsOptionsTikTokAdsCustomReportOptions", + "TikTokAdsOptionsTikTokAdsCustomReportOptionsDict", + "TikTokAdsOptionsTikTokAdsCustomReportOptionsParam", + "TikTokAdsOptionsTikTokDataLevel", + "TikTokAdsOptionsTikTokDataLevelParam", + "TikTokAdsOptionsTikTokReportType", + "TikTokAdsOptionsTikTokReportTypeParam", + "Transformer", + "TransformerDict", + "TransformerFormat", + "TransformerFormatParam", + "TransformerParam", + "VolumesStorageInfo", + "VolumesStorageInfoDict", + "VolumesStorageInfoParam", + "WorkspaceStorageInfo", + "WorkspaceStorageInfoDict", + "WorkspaceStorageInfoParam", + "ZendeskSupportOptions", + "ZendeskSupportOptionsDict", + "ZendeskSupportOptionsParam", ] -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - +from databricks.bundles.pipelines._models.adlsgen2_info import ( + Adlsgen2Info, + Adlsgen2InfoDict, + Adlsgen2InfoParam, +) +from databricks.bundles.pipelines._models.api_source_connector_config import ( + ApiSourceConnectorConfig, + ApiSourceConnectorConfigDict, + ApiSourceConnectorConfigParam, +) +from databricks.bundles.pipelines._models.api_source_connector_options import ( + ApiSourceConnectorOptions, + ApiSourceConnectorOptionsDict, + ApiSourceConnectorOptionsParam, +) +from databricks.bundles.pipelines._models.auto_full_refresh_policy import ( + AutoFullRefreshPolicy, + AutoFullRefreshPolicyDict, + AutoFullRefreshPolicyParam, +) +from databricks.bundles.pipelines._models.avro_transformer_options import ( + AvroTransformerOptions, + AvroTransformerOptionsDict, + AvroTransformerOptionsParam, +) +from databricks.bundles.pipelines._models.aws_attributes import ( + AwsAttributes, + AwsAttributesDict, + AwsAttributesParam, +) +from databricks.bundles.pipelines._models.aws_availability import ( + AwsAvailability, + AwsAvailabilityParam, +) +from databricks.bundles.pipelines._models.azure_attributes import ( + AzureAttributes, + AzureAttributesDict, + AzureAttributesParam, +) +from databricks.bundles.pipelines._models.azure_availability import ( + AzureAvailability, + AzureAvailabilityParam, +) +from databricks.bundles.pipelines._models.cluster_log_conf import ( + ClusterLogConf, + ClusterLogConfDict, + ClusterLogConfParam, +) +from databricks.bundles.pipelines._models.confidential_compute_type import ( + ConfidentialComputeType, + ConfidentialComputeTypeParam, +) +from databricks.bundles.pipelines._models.confluence_connector_options import ( + ConfluenceConnectorOptions, + ConfluenceConnectorOptionsDict, + ConfluenceConnectorOptionsParam, +) +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ( + ConfluentSchemaRegistryOptions, + ConfluentSchemaRegistryOptionsDict, + ConfluentSchemaRegistryOptionsParam, +) +from databricks.bundles.pipelines._models.connection_parameters import ( + ConnectionParameters, + ConnectionParametersDict, + ConnectionParametersParam, +) +from databricks.bundles.pipelines._models.connector_options import ( + ConnectorOptions, + ConnectorOptionsDict, + ConnectorOptionsParam, +) +from databricks.bundles.pipelines._models.connector_type import ( + ConnectorType, + ConnectorTypeParam, +) +from databricks.bundles.pipelines._models.data_staging_options import ( + DataStagingOptions, + DataStagingOptionsDict, + DataStagingOptionsParam, +) from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - +from databricks.bundles.pipelines._models.dbfs_storage_info import ( + DbfsStorageInfo, + DbfsStorageInfoDict, + DbfsStorageInfoParam, +) +from databricks.bundles.pipelines._models.ebs_volume_type import ( + EbsVolumeType, + EbsVolumeTypeParam, +) +from databricks.bundles.pipelines._models.event_log_spec import ( + EventLogSpec, + EventLogSpecDict, + EventLogSpecParam, +) +from databricks.bundles.pipelines._models.file_filter import ( + FileFilter, + FileFilterDict, + FileFilterParam, +) +from databricks.bundles.pipelines._models.file_ingestion_options import ( + FileIngestionOptions, + FileIngestionOptionsDict, + FileIngestionOptionsParam, +) +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import ( + FileIngestionOptionsFileFormat, + FileIngestionOptionsFileFormatParam, +) +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import ( + FileIngestionOptionsSchemaEvolutionMode, + FileIngestionOptionsSchemaEvolutionModeParam, +) +from databricks.bundles.pipelines._models.file_library import ( + FileLibrary, + FileLibraryDict, + FileLibraryParam, +) +from databricks.bundles.pipelines._models.filters import ( + Filters, + FiltersDict, + FiltersParam, +) +from databricks.bundles.pipelines._models.gcp_attributes import ( + GcpAttributes, + GcpAttributesDict, + GcpAttributesParam, +) +from databricks.bundles.pipelines._models.gcp_availability import ( + GcpAvailability, + GcpAvailabilityParam, +) +from databricks.bundles.pipelines._models.gcs_storage_info import ( + GcsStorageInfo, + GcsStorageInfoDict, + GcsStorageInfoParam, +) +from databricks.bundles.pipelines._models.google_ads_config import ( + GoogleAdsConfig, + GoogleAdsConfigDict, + GoogleAdsConfigParam, +) +from databricks.bundles.pipelines._models.google_ads_custom_report_options import ( + GoogleAdsCustomReportOptions, + GoogleAdsCustomReportOptionsDict, + GoogleAdsCustomReportOptionsParam, +) +from databricks.bundles.pipelines._models.google_ads_options import ( + GoogleAdsOptions, + GoogleAdsOptionsDict, + GoogleAdsOptionsParam, +) +from databricks.bundles.pipelines._models.google_drive_options import ( + GoogleDriveOptions, + GoogleDriveOptionsDict, + GoogleDriveOptionsParam, +) +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import ( + GoogleDriveOptionsGoogleDriveEntityType, + GoogleDriveOptionsGoogleDriveEntityTypeParam, +) +from databricks.bundles.pipelines._models.ingestion_config import ( + IngestionConfig, + IngestionConfigDict, + IngestionConfigParam, +) +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import ( + IngestionGatewayPipelineDefinition, + IngestionGatewayPipelineDefinitionDict, + IngestionGatewayPipelineDefinitionParam, +) +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import ( + IngestionPipelineDefinition, + IngestionPipelineDefinitionDict, + IngestionPipelineDefinitionParam, +) +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import ( + IngestionPipelineDefinitionFanoutOptions, + IngestionPipelineDefinitionFanoutOptionsDict, + IngestionPipelineDefinitionFanoutOptionsParam, +) +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import ( + IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, + IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, + IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam, +) +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import ( + IngestionPipelineDefinitionWorkdayReportParameters, + IngestionPipelineDefinitionWorkdayReportParametersDict, + IngestionPipelineDefinitionWorkdayReportParametersParam, +) +from databricks.bundles.pipelines._models.init_script_info import ( + InitScriptInfo, + InitScriptInfoDict, + InitScriptInfoParam, +) +from databricks.bundles.pipelines._models.jira_connector_options import ( + JiraConnectorOptions, + JiraConnectorOptionsDict, + JiraConnectorOptionsParam, +) +from databricks.bundles.pipelines._models.json_transformer_options import ( + JsonTransformerOptions, + JsonTransformerOptionsDict, + JsonTransformerOptionsParam, +) +from databricks.bundles.pipelines._models.kafka_options import ( + KafkaOptions, + KafkaOptionsDict, + KafkaOptionsParam, +) +from databricks.bundles.pipelines._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.pipelines._models.linked_in_ads_options import ( + LinkedInAdsOptions, + LinkedInAdsOptionsDict, + LinkedInAdsOptionsParam, +) +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import ( + LinkedInAdsOptionsLinkedInAdsCustomReportOptions, + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam, +) +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import ( + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam, +) +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import ( + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam, +) +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import ( + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam, +) +from databricks.bundles.pipelines._models.local_file_info import ( + LocalFileInfo, + LocalFileInfoDict, + LocalFileInfoParam, +) +from databricks.bundles.pipelines._models.log_analytics_info import ( + LogAnalyticsInfo, + LogAnalyticsInfoDict, + LogAnalyticsInfoParam, +) +from databricks.bundles.pipelines._models.marketo_options import ( + MarketoOptions, + MarketoOptionsDict, + MarketoOptionsParam, +) +from databricks.bundles.pipelines._models.maven_library import ( + MavenLibrary, + MavenLibraryDict, + MavenLibraryParam, +) +from databricks.bundles.pipelines._models.meta_marketing_options import ( + MetaMarketingOptions, + MetaMarketingOptionsDict, + MetaMarketingOptionsParam, +) +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import ( + MetaMarketingOptionsMetaMarketingCustomReportOptions, + MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, + MetaMarketingOptionsMetaMarketingCustomReportOptionsParam, +) +from databricks.bundles.pipelines._models.notebook_library import ( + NotebookLibrary, + NotebookLibraryDict, + NotebookLibraryParam, +) +from databricks.bundles.pipelines._models.notifications import ( + Notifications, + NotificationsDict, + NotificationsParam, +) +from databricks.bundles.pipelines._models.operation_time_window import ( + OperationTimeWindow, + OperationTimeWindowDict, + OperationTimeWindowParam, +) +from databricks.bundles.pipelines._models.outlook_attachment_mode import ( + OutlookAttachmentMode, + OutlookAttachmentModeParam, +) +from databricks.bundles.pipelines._models.outlook_body_format import ( + OutlookBodyFormat, + OutlookBodyFormatParam, +) +from databricks.bundles.pipelines._models.outlook_options import ( + OutlookOptions, + OutlookOptionsDict, + OutlookOptionsParam, +) from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.pipelines._models.path_pattern import ( + PathPattern, + PathPatternDict, + PathPatternParam, +) +from databricks.bundles.pipelines._models.pipeline import ( + Pipeline, + PipelineDict, + PipelineParam, +) +from databricks.bundles.pipelines._models.pipeline_cluster import ( + PipelineCluster, + PipelineClusterDict, + PipelineClusterParam, +) +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import ( + PipelineClusterAutoscale, + PipelineClusterAutoscaleDict, + PipelineClusterAutoscaleParam, +) +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import ( + PipelineClusterAutoscaleMode, + PipelineClusterAutoscaleModeParam, +) +from databricks.bundles.pipelines._models.pipeline_library import ( + PipelineLibrary, + PipelineLibraryDict, + PipelineLibraryParam, +) +from databricks.bundles.pipelines._models.pipeline_permission import ( + PipelinePermission, + PipelinePermissionDict, + PipelinePermissionParam, +) +from databricks.bundles.pipelines._models.pipeline_permission_level import ( + PipelinePermissionLevel, + PipelinePermissionLevelParam, +) +from databricks.bundles.pipelines._models.pipelines_environment import ( + PipelinesEnvironment, + PipelinesEnvironmentDict, + PipelinesEnvironmentParam, +) +from databricks.bundles.pipelines._models.postgres_catalog_config import ( + PostgresCatalogConfig, + PostgresCatalogConfigDict, + PostgresCatalogConfigParam, +) +from databricks.bundles.pipelines._models.postgres_slot_config import ( + PostgresSlotConfig, + PostgresSlotConfigDict, + PostgresSlotConfigParam, +) +from databricks.bundles.pipelines._models.protobuf_transformer_options import ( + ProtobufTransformerOptions, + ProtobufTransformerOptionsDict, + ProtobufTransformerOptionsParam, +) +from databricks.bundles.pipelines._models.rabbitmq_options import ( + RabbitmqOptions, + RabbitmqOptionsDict, + RabbitmqOptionsParam, +) +from databricks.bundles.pipelines._models.reddit_ads_options import ( + RedditAdsOptions, + RedditAdsOptionsDict, + RedditAdsOptionsParam, +) +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import ( + RedditAdsOptionsRedditAdsCustomReportOptions, + RedditAdsOptionsRedditAdsCustomReportOptionsDict, + RedditAdsOptionsRedditAdsCustomReportOptionsParam, +) +from databricks.bundles.pipelines._models.report_spec import ( + ReportSpec, + ReportSpecDict, + ReportSpecParam, +) +from databricks.bundles.pipelines._models.restart_window import ( + RestartWindow, + RestartWindowDict, + RestartWindowParam, +) +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam +from databricks.bundles.pipelines._models.s3_storage_info import ( + S3StorageInfo, + S3StorageInfoDict, + S3StorageInfoParam, +) +from databricks.bundles.pipelines._models.schema_registry_config import ( + SchemaRegistryConfig, + SchemaRegistryConfigDict, + SchemaRegistryConfigParam, +) +from databricks.bundles.pipelines._models.schema_spec import ( + SchemaSpec, + SchemaSpecDict, + SchemaSpecParam, +) +from databricks.bundles.pipelines._models.sharepoint_options import ( + SharepointOptions, + SharepointOptionsDict, + SharepointOptionsParam, +) +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import ( + SharepointOptionsSharepointEntityType, + SharepointOptionsSharepointEntityTypeParam, +) +from databricks.bundles.pipelines._models.smartsheet_options import ( + SmartsheetOptions, + SmartsheetOptionsDict, + SmartsheetOptionsParam, +) +from databricks.bundles.pipelines._models.source_catalog_config import ( + SourceCatalogConfig, + SourceCatalogConfigDict, + SourceCatalogConfigParam, +) +from databricks.bundles.pipelines._models.source_config import ( + SourceConfig, + SourceConfigDict, + SourceConfigParam, +) +from databricks.bundles.pipelines._models.table_spec import ( + TableSpec, + TableSpecDict, + TableSpecParam, +) +from databricks.bundles.pipelines._models.table_specific_config import ( + TableSpecificConfig, + TableSpecificConfigDict, + TableSpecificConfigParam, +) +from databricks.bundles.pipelines._models.table_specific_config_scd_type import ( + TableSpecificConfigScdType, + TableSpecificConfigScdTypeParam, +) +from databricks.bundles.pipelines._models.tik_tok_ads_options import ( + TikTokAdsOptions, + TikTokAdsOptionsDict, + TikTokAdsOptionsParam, +) +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import ( + TikTokAdsOptionsTikTokAdsCustomReportOptions, + TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, + TikTokAdsOptionsTikTokAdsCustomReportOptionsParam, +) +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import ( + TikTokAdsOptionsTikTokDataLevel, + TikTokAdsOptionsTikTokDataLevelParam, +) +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import ( + TikTokAdsOptionsTikTokReportType, + TikTokAdsOptionsTikTokReportTypeParam, +) +from databricks.bundles.pipelines._models.transformer import ( + Transformer, + TransformerDict, + TransformerParam, +) +from databricks.bundles.pipelines._models.transformer_format import ( + TransformerFormat, + TransformerFormatParam, +) +from databricks.bundles.pipelines._models.volumes_storage_info import ( + VolumesStorageInfo, + VolumesStorageInfoDict, + VolumesStorageInfoParam, +) +from databricks.bundles.pipelines._models.workspace_storage_info import ( + WorkspaceStorageInfo, + WorkspaceStorageInfoDict, + WorkspaceStorageInfoParam, +) +from databricks.bundles.pipelines._models.zendesk_support_options import ( + ZendeskSupportOptions, + ZendeskSupportOptionsDict, + ZendeskSupportOptionsParam, +) diff --git a/python/databricks/bundles/pipelines/_models/adlsgen2_info.py b/python/databricks/bundles/pipelines/_models/adlsgen2_info.py index 4d414ae5536..87c883adeb9 100644 --- a/python/databricks/bundles/pipelines/_models/adlsgen2_info.py +++ b/python/databricks/bundles/pipelines/_models/adlsgen2_info.py @@ -1,159 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Adlsgen2Info: """ A storage location in Adls Gen2 """ + destination: VariableOr[str] """ abfss destination, e.g. `abfss://@.dfs.core.windows.net/`. """ @classmethod - def from_dict(cls, value: 'Adlsgen2InfoDict') -> 'Self': + def from_dict(cls, value: "Adlsgen2InfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'Adlsgen2InfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "Adlsgen2InfoDict": + return _transform_to_json_value(self) # type:ignore class Adlsgen2InfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/api_source_connector_config.py b/python/databricks/bundles/pipelines/_models/api_source_connector_config.py index a7188f32a28..2ab533ab32f 100644 --- a/python/databricks/bundles/pipelines/_models/api_source_connector_config.py +++ b/python/databricks/bundles/pipelines/_models/api_source_connector_config.py @@ -1,149 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrDict if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ApiSourceConnectorConfig: """ :meta private: [EXPERIMENTAL] - + Top-level configuration for API Source connectors with arbitrary configuration. """ + configs: VariableOrDict[str] = field(default_factory=dict) """ :meta private: [EXPERIMENTAL] @@ -152,12 +27,11 @@ class ApiSourceConnectorConfig: """ @classmethod - def from_dict(cls, value: 'ApiSourceConnectorConfigDict') -> 'Self': + def from_dict(cls, value: "ApiSourceConnectorConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ApiSourceConnectorConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ApiSourceConnectorConfigDict": + return _transform_to_json_value(self) # type:ignore class ApiSourceConnectorConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/api_source_connector_options.py b/python/databricks/bundles/pipelines/_models/api_source_connector_options.py index ededf309db1..2e0f23c0f03 100644 --- a/python/databricks/bundles/pipelines/_models/api_source_connector_options.py +++ b/python/databricks/bundles/pipelines/_models/api_source_connector_options.py @@ -1,149 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrDict if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ApiSourceConnectorOptions: """ :meta private: [EXPERIMENTAL] - + Options for API Source connectors with arbitrary configuration. """ + options: VariableOrDict[str] = field(default_factory=dict) """ :meta private: [EXPERIMENTAL] @@ -152,12 +27,11 @@ class ApiSourceConnectorOptions: """ @classmethod - def from_dict(cls, value: 'ApiSourceConnectorOptionsDict') -> 'Self': + def from_dict(cls, value: "ApiSourceConnectorOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ApiSourceConnectorOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ApiSourceConnectorOptionsDict": + return _transform_to_json_value(self) # type:ignore class ApiSourceConnectorOptionsDict(TypedDict, total=False): @@ -171,4 +45,6 @@ class ApiSourceConnectorOptionsDict(TypedDict, total=False): """ -ApiSourceConnectorOptionsParam = ApiSourceConnectorOptionsDict | ApiSourceConnectorOptions +ApiSourceConnectorOptionsParam = ( + ApiSourceConnectorOptionsDict | ApiSourceConnectorOptions +) diff --git a/python/databricks/bundles/pipelines/_models/auto_full_refresh_policy.py b/python/databricks/bundles/pipelines/_models/auto_full_refresh_policy.py index dd8f87ae21f..2ee3ed2382a 100644 --- a/python/databricks/bundles/pipelines/_models/auto_full_refresh_policy.py +++ b/python/databricks/bundles/pipelines/_models/auto_full_refresh_policy.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AutoFullRefreshPolicy: """ Policy for auto full refresh. """ + enabled: VariableOr[bool] """ [Public Preview] (Required, Mutable) Whether to enable auto full refresh or not. @@ -155,12 +30,11 @@ class AutoFullRefreshPolicy: """ @classmethod - def from_dict(cls, value: 'AutoFullRefreshPolicyDict') -> 'Self': + def from_dict(cls, value: "AutoFullRefreshPolicyDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AutoFullRefreshPolicyDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AutoFullRefreshPolicyDict": + return _transform_to_json_value(self) # type:ignore class AutoFullRefreshPolicyDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/avro_transformer_options.py b/python/databricks/bundles/pipelines/_models/avro_transformer_options.py index fae21e3e9ea..3dbfd7cf72b 100644 --- a/python/databricks/bundles/pipelines/_models/avro_transformer_options.py +++ b/python/databricks/bundles/pipelines/_models/avro_transformer_options.py @@ -1,142 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - +from databricks.bundles.core._variable import VariableOrOptional from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.pipelines._models.schema_registry_config import ( + SchemaRegistryConfig, + SchemaRegistryConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AvroTransformerOptions: """""" @@ -172,12 +51,11 @@ class AvroTransformerOptions: """ @classmethod - def from_dict(cls, value: 'AvroTransformerOptionsDict') -> 'Self': + def from_dict(cls, value: "AvroTransformerOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AvroTransformerOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AvroTransformerOptionsDict": + return _transform_to_json_value(self) # type:ignore class AvroTransformerOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/aws_attributes.py b/python/databricks/bundles/pipelines/_models/aws_attributes.py index 9f63436ff27..3308f3c6e96 100644 --- a/python/databricks/bundles/pipelines/_models/aws_attributes.py +++ b/python/databricks/bundles/pipelines/_models/aws_attributes.py @@ -1,147 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.aws_availability import ( + AwsAvailability, + AwsAvailabilityParam, +) +from databricks.bundles.pipelines._models.ebs_volume_type import ( + EbsVolumeType, + EbsVolumeTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AwsAttributes: """ Attributes set during cluster creation which are related to Amazon Web Services. """ + availability: VariableOrOptional[AwsAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -240,12 +123,11 @@ class AwsAttributes: """ @classmethod - def from_dict(cls, value: 'AwsAttributesDict') -> 'Self': + def from_dict(cls, value: "AwsAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AwsAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AwsAttributesDict": + return _transform_to_json_value(self) # type:ignore class AwsAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/aws_availability.py b/python/databricks/bundles/pipelines/_models/aws_availability.py index 55cc2ec69ec..4c3810f067e 100644 --- a/python/databricks/bundles/pipelines/_models/aws_availability.py +++ b/python/databricks/bundles/pipelines/_models/aws_availability.py @@ -1,146 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class AwsAvailability(Enum): """ Availability type used for all subsequent nodes past the `first_on_demand` ones. - + Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ + SPOT = "SPOT" ON_DEMAND = "ON_DEMAND" SPOT_WITH_FALLBACK = "SPOT_WITH_FALLBACK" -AwsAvailabilityParam = Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability + +AwsAvailabilityParam = ( + Literal["SPOT", "ON_DEMAND", "SPOT_WITH_FALLBACK"] | AwsAvailability +) diff --git a/python/databricks/bundles/pipelines/_models/azure_attributes.py b/python/databricks/bundles/pipelines/_models/azure_attributes.py index d704fa90440..7caff90e481 100644 --- a/python/databricks/bundles/pipelines/_models/azure_attributes.py +++ b/python/databricks/bundles/pipelines/_models/azure_attributes.py @@ -1,147 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.azure_availability import ( + AzureAvailability, + AzureAvailabilityParam, +) +from databricks.bundles.pipelines._models.log_analytics_info import ( + LogAnalyticsInfo, + LogAnalyticsInfoParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AzureAttributes: """ Attributes set during cluster creation which are related to Microsoft Azure. """ + availability: VariableOrOptional[AzureAvailability] = None """ Availability type used for all subsequent nodes past the `first_on_demand` ones. @@ -189,12 +72,11 @@ class AzureAttributes: """ @classmethod - def from_dict(cls, value: 'AzureAttributesDict') -> 'Self': + def from_dict(cls, value: "AzureAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AzureAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AzureAttributesDict": + return _transform_to_json_value(self) # type:ignore class AzureAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/azure_availability.py b/python/databricks/bundles/pipelines/_models/azure_availability.py index bc43590fd87..a03cb1fdddf 100644 --- a/python/databricks/bundles/pipelines/_models/azure_availability.py +++ b/python/databricks/bundles/pipelines/_models/azure_availability.py @@ -1,136 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class AzureAvailability(Enum): @@ -138,8 +9,13 @@ class AzureAvailability(Enum): Availability type used for all subsequent nodes past the `first_on_demand` ones. Note: If `first_on_demand` is zero, this availability type will be used for the entire cluster. """ + SPOT_AZURE = "SPOT_AZURE" ON_DEMAND_AZURE = "ON_DEMAND_AZURE" SPOT_WITH_FALLBACK_AZURE = "SPOT_WITH_FALLBACK_AZURE" -AzureAvailabilityParam = Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] | AzureAvailability + +AzureAvailabilityParam = ( + Literal["SPOT_AZURE", "ON_DEMAND_AZURE", "SPOT_WITH_FALLBACK_AZURE"] + | AzureAvailability +) diff --git a/python/databricks/bundles/pipelines/_models/cluster_log_conf.py b/python/databricks/bundles/pipelines/_models/cluster_log_conf.py index d2153bfc0be..74c9dd37df4 100644 --- a/python/databricks/bundles/pipelines/_models/cluster_log_conf.py +++ b/python/databricks/bundles/pipelines/_models/cluster_log_conf.py @@ -1,147 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.dbfs_storage_info import ( + DbfsStorageInfo, + DbfsStorageInfoParam, +) +from databricks.bundles.pipelines._models.s3_storage_info import ( + S3StorageInfo, + S3StorageInfoParam, +) +from databricks.bundles.pipelines._models.volumes_storage_info import ( + VolumesStorageInfo, + VolumesStorageInfoParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ClusterLogConf: """ Cluster log delivery config """ + dbfs: VariableOrOptional[DbfsStorageInfo] = None """ destination needs to be provided. e.g. @@ -163,12 +50,11 @@ class ClusterLogConf: """ @classmethod - def from_dict(cls, value: 'ClusterLogConfDict') -> 'Self': + def from_dict(cls, value: "ClusterLogConfDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ClusterLogConfDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ClusterLogConfDict": + return _transform_to_json_value(self) # type:ignore class ClusterLogConfDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/confidential_compute_type.py b/python/databricks/bundles/pipelines/_models/confidential_compute_type.py index 8c579046d06..e881295bd47 100644 --- a/python/databricks/bundles/pipelines/_models/confidential_compute_type.py +++ b/python/databricks/bundles/pipelines/_models/confidential_compute_type.py @@ -1,148 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class ConfidentialComputeType(Enum): """ :meta private: [EXPERIMENTAL] - + Confidential computing technology for GCP instances. Aligns with gcloud's --confidential-compute-type flag and the REST API's confidentialInstanceConfig.confidentialInstanceType field. See: https://cloud.google.com/confidential-computing/confidential-vm/docs/create-a-confidential-vm-instance """ + CONFIDENTIAL_COMPUTE_TYPE_NONE = "CONFIDENTIAL_COMPUTE_TYPE_NONE" SEV_SNP = "SEV_SNP" -ConfidentialComputeTypeParam = Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType + +ConfidentialComputeTypeParam = ( + Literal["CONFIDENTIAL_COMPUTE_TYPE_NONE", "SEV_SNP"] | ConfidentialComputeType +) diff --git a/python/databricks/bundles/pipelines/_models/confluence_connector_options.py b/python/databricks/bundles/pipelines/_models/confluence_connector_options.py index e73f41281a1..39f414edbd2 100644 --- a/python/databricks/bundles/pipelines/_models/confluence_connector_options.py +++ b/python/databricks/bundles/pipelines/_models/confluence_connector_options.py @@ -1,159 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ConfluenceConnectorOptions: """ Confluence specific options for ingestion """ + include_confluence_spaces: VariableOrList[str] = field(default_factory=list) """ [Public Preview] (Optional) Spaces to filter Confluence data on """ @classmethod - def from_dict(cls, value: 'ConfluenceConnectorOptionsDict') -> 'Self': + def from_dict(cls, value: "ConfluenceConnectorOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ConfluenceConnectorOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ConfluenceConnectorOptionsDict": + return _transform_to_json_value(self) # type:ignore class ConfluenceConnectorOptionsDict(TypedDict, total=False): @@ -165,4 +39,6 @@ class ConfluenceConnectorOptionsDict(TypedDict, total=False): """ -ConfluenceConnectorOptionsParam = ConfluenceConnectorOptionsDict | ConfluenceConnectorOptions +ConfluenceConnectorOptionsParam = ( + ConfluenceConnectorOptionsDict | ConfluenceConnectorOptions +) diff --git a/python/databricks/bundles/pipelines/_models/confluent_schema_registry_options.py b/python/databricks/bundles/pipelines/_models/confluent_schema_registry_options.py index 2e0b57bdd7d..af82614ea7a 100644 --- a/python/databricks/bundles/pipelines/_models/confluent_schema_registry_options.py +++ b/python/databricks/bundles/pipelines/_models/confluent_schema_registry_options.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ConfluentSchemaRegistryOptions: """""" @@ -149,12 +23,11 @@ class ConfluentSchemaRegistryOptions: """ @classmethod - def from_dict(cls, value: 'ConfluentSchemaRegistryOptionsDict') -> 'Self': + def from_dict(cls, value: "ConfluentSchemaRegistryOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ConfluentSchemaRegistryOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ConfluentSchemaRegistryOptionsDict": + return _transform_to_json_value(self) # type:ignore class ConfluentSchemaRegistryOptionsDict(TypedDict, total=False): @@ -168,4 +41,6 @@ class ConfluentSchemaRegistryOptionsDict(TypedDict, total=False): """ -ConfluentSchemaRegistryOptionsParam = ConfluentSchemaRegistryOptionsDict | ConfluentSchemaRegistryOptions +ConfluentSchemaRegistryOptionsParam = ( + ConfluentSchemaRegistryOptionsDict | ConfluentSchemaRegistryOptions +) diff --git a/python/databricks/bundles/pipelines/_models/connection_parameters.py b/python/databricks/bundles/pipelines/_models/connection_parameters.py index e6057ee8909..bb13f406413 100644 --- a/python/databricks/bundles/pipelines/_models/connection_parameters.py +++ b/python/databricks/bundles/pipelines/_models/connection_parameters.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ConnectionParameters: """ :meta private: [EXPERIMENTAL] """ + source_catalog: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -153,12 +28,11 @@ class ConnectionParameters: """ @classmethod - def from_dict(cls, value: 'ConnectionParametersDict') -> 'Self': + def from_dict(cls, value: "ConnectionParametersDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ConnectionParametersDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ConnectionParametersDict": + return _transform_to_json_value(self) # type:ignore class ConnectionParametersDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/connector_options.py b/python/databricks/bundles/pipelines/_models/connector_options.py index cc98b4ad12a..945a5bab56a 100644 --- a/python/databricks/bundles/pipelines/_models/connector_options.py +++ b/python/databricks/bundles/pipelines/_models/connector_options.py @@ -1,147 +1,86 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.api_source_connector_options import ( + ApiSourceConnectorOptions, + ApiSourceConnectorOptionsParam, +) +from databricks.bundles.pipelines._models.confluence_connector_options import ( + ConfluenceConnectorOptions, + ConfluenceConnectorOptionsParam, +) +from databricks.bundles.pipelines._models.google_ads_options import ( + GoogleAdsOptions, + GoogleAdsOptionsParam, +) +from databricks.bundles.pipelines._models.google_drive_options import ( + GoogleDriveOptions, + GoogleDriveOptionsParam, +) +from databricks.bundles.pipelines._models.jira_connector_options import ( + JiraConnectorOptions, + JiraConnectorOptionsParam, +) +from databricks.bundles.pipelines._models.kafka_options import ( + KafkaOptions, + KafkaOptionsParam, +) +from databricks.bundles.pipelines._models.linked_in_ads_options import ( + LinkedInAdsOptions, + LinkedInAdsOptionsParam, +) +from databricks.bundles.pipelines._models.marketo_options import ( + MarketoOptions, + MarketoOptionsParam, +) +from databricks.bundles.pipelines._models.meta_marketing_options import ( + MetaMarketingOptions, + MetaMarketingOptionsParam, +) +from databricks.bundles.pipelines._models.outlook_options import ( + OutlookOptions, + OutlookOptionsParam, +) +from databricks.bundles.pipelines._models.rabbitmq_options import ( + RabbitmqOptions, + RabbitmqOptionsParam, +) +from databricks.bundles.pipelines._models.reddit_ads_options import ( + RedditAdsOptions, + RedditAdsOptionsParam, +) +from databricks.bundles.pipelines._models.sharepoint_options import ( + SharepointOptions, + SharepointOptionsParam, +) +from databricks.bundles.pipelines._models.smartsheet_options import ( + SmartsheetOptions, + SmartsheetOptionsParam, +) +from databricks.bundles.pipelines._models.tik_tok_ads_options import ( + TikTokAdsOptions, + TikTokAdsOptionsParam, +) +from databricks.bundles.pipelines._models.zendesk_support_options import ( + ZendeskSupportOptions, + ZendeskSupportOptionsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ConnectorOptions: """ Wrapper message for source-specific options to support multiple connector types """ + api_source_connector_options: VariableOrOptional[ApiSourceConnectorOptions] = None """ :meta private: [EXPERIMENTAL] @@ -257,12 +196,11 @@ class ConnectorOptions: """ @classmethod - def from_dict(cls, value: 'ConnectorOptionsDict') -> 'Self': + def from_dict(cls, value: "ConnectorOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ConnectorOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ConnectorOptionsDict": + return _transform_to_json_value(self) # type:ignore class ConnectorOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/connector_type.py b/python/databricks/bundles/pipelines/_models/connector_type.py index 46a3ec1b670..e8c7fb15931 100644 --- a/python/databricks/bundles/pipelines/_models/connector_type.py +++ b/python/databricks/bundles/pipelines/_models/connector_type.py @@ -1,136 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class ConnectorType(Enum): @@ -139,7 +10,9 @@ class ConnectorType(Enum): ingestion, ConnectorType can bse used to convey the type of ingestion. If connection_name is provided for database sources, we default to Query Based ingestion """ + CDC = "CDC" QUERY_BASED = "QUERY_BASED" + ConnectorTypeParam = Literal["CDC", "QUERY_BASED"] | ConnectorType diff --git a/python/databricks/bundles/pipelines/_models/data_staging_options.py b/python/databricks/bundles/pipelines/_models/data_staging_options.py index a8e833c70a4..62672108bbd 100644 --- a/python/databricks/bundles/pipelines/_models/data_staging_options.py +++ b/python/databricks/bundles/pipelines/_models/data_staging_options.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DataStagingOptions: """ Location of staged data storage """ + catalog_name: VariableOr[str] """ [Public Preview] (Required, Immutable) The name of the catalog for the connector's staging storage location. @@ -162,12 +37,11 @@ class DataStagingOptions: """ @classmethod - def from_dict(cls, value: 'DataStagingOptionsDict') -> 'Self': + def from_dict(cls, value: "DataStagingOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DataStagingOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DataStagingOptionsDict": + return _transform_to_json_value(self) # type:ignore class DataStagingOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/day_of_week.py b/python/databricks/bundles/pipelines/_models/day_of_week.py index 44674814837..8ba5a7c1ab6 100644 --- a/python/databricks/bundles/pipelines/_models/day_of_week.py +++ b/python/databricks/bundles/pipelines/_models/day_of_week.py @@ -1,136 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class DayOfWeek(Enum): @@ -138,6 +9,7 @@ class DayOfWeek(Enum): Days of week in which the window is allowed to happen. If not specified all days of the week will be used. """ + MONDAY = "MONDAY" TUESDAY = "TUESDAY" WEDNESDAY = "WEDNESDAY" @@ -146,4 +18,10 @@ class DayOfWeek(Enum): SATURDAY = "SATURDAY" SUNDAY = "SUNDAY" -DayOfWeekParam = Literal["MONDAY", "TUESDAY", "WEDNESDAY", "THURSDAY", "FRIDAY", "SATURDAY", "SUNDAY"] | DayOfWeek + +DayOfWeekParam = ( + Literal[ + "MONDAY", "TUESDAY", "WEDNESDAY", "THURSDAY", "FRIDAY", "SATURDAY", "SUNDAY" + ] + | DayOfWeek +) diff --git a/python/databricks/bundles/pipelines/_models/dbfs_storage_info.py b/python/databricks/bundles/pipelines/_models/dbfs_storage_info.py index 2a62fcd74ac..29744acf26a 100644 --- a/python/databricks/bundles/pipelines/_models/dbfs_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/dbfs_storage_info.py @@ -1,159 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DbfsStorageInfo: """ A storage location in DBFS """ + destination: VariableOr[str] """ dbfs destination, e.g. `dbfs:/my/path` """ @classmethod - def from_dict(cls, value: 'DbfsStorageInfoDict') -> 'Self': + def from_dict(cls, value: "DbfsStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DbfsStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DbfsStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class DbfsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/ebs_volume_type.py b/python/databricks/bundles/pipelines/_models/ebs_volume_type.py index a4dc426f6e9..d8f3ef13f89 100644 --- a/python/databricks/bundles/pipelines/_models/ebs_volume_type.py +++ b/python/databricks/bundles/pipelines/_models/ebs_volume_type.py @@ -1,136 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class EbsVolumeType(Enum): @@ -138,7 +9,11 @@ class EbsVolumeType(Enum): All EBS volume types that Databricks supports. See https://aws.amazon.com/ebs/details/ for details. """ + GENERAL_PURPOSE_SSD = "GENERAL_PURPOSE_SSD" THROUGHPUT_OPTIMIZED_HDD = "THROUGHPUT_OPTIMIZED_HDD" -EbsVolumeTypeParam = Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType + +EbsVolumeTypeParam = ( + Literal["GENERAL_PURPOSE_SSD", "THROUGHPUT_OPTIMIZED_HDD"] | EbsVolumeType +) diff --git a/python/databricks/bundles/pipelines/_models/event_log_spec.py b/python/databricks/bundles/pipelines/_models/event_log_spec.py index 915bc04f7e2..d415ee83246 100644 --- a/python/databricks/bundles/pipelines/_models/event_log_spec.py +++ b/python/databricks/bundles/pipelines/_models/event_log_spec.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EventLogSpec: """ Configurable event log parameters. """ + catalog: VariableOrOptional[str] = None """ The UC catalog the event log is published under. @@ -158,12 +33,11 @@ class EventLogSpec: """ @classmethod - def from_dict(cls, value: 'EventLogSpecDict') -> 'Self': + def from_dict(cls, value: "EventLogSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EventLogSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EventLogSpecDict": + return _transform_to_json_value(self) # type:ignore class EventLogSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/file_filter.py b/python/databricks/bundles/pipelines/_models/file_filter.py index 73e7b08f5a2..7e80f319c0f 100644 --- a/python/databricks/bundles/pipelines/_models/file_filter.py +++ b/python/databricks/bundles/pipelines/_models/file_filter.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FileFilter: """ :meta private: [EXPERIMENTAL] """ + modified_after: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -169,12 +44,11 @@ class FileFilter: """ @classmethod - def from_dict(cls, value: 'FileFilterDict') -> 'Self': + def from_dict(cls, value: "FileFilterDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FileFilterDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FileFilterDict": + return _transform_to_json_value(self) # type:ignore class FileFilterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/file_ingestion_options.py b/python/databricks/bundles/pipelines/_models/file_ingestion_options.py index 0f460bdb54c..0e6d82d6684 100644 --- a/python/databricks/bundles/pipelines/_models/file_ingestion_options.py +++ b/python/databricks/bundles/pipelines/_models/file_ingestion_options.py @@ -1,147 +1,35 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import ( + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterParam +from databricks.bundles.pipelines._models.file_ingestion_options_file_format import ( + FileIngestionOptionsFileFormat, + FileIngestionOptionsFileFormatParam, +) +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import ( + FileIngestionOptionsSchemaEvolutionMode, + FileIngestionOptionsSchemaEvolutionModeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FileIngestionOptions: """ :meta private: [EXPERIMENTAL] """ + corrupt_record_column: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -200,7 +88,9 @@ class FileIngestionOptions: [Private Preview] """ - schema_evolution_mode: VariableOrOptional[FileIngestionOptionsSchemaEvolutionMode] = None + schema_evolution_mode: VariableOrOptional[ + FileIngestionOptionsSchemaEvolutionMode + ] = None """ :meta private: [EXPERIMENTAL] @@ -223,12 +113,11 @@ class FileIngestionOptions: """ @classmethod - def from_dict(cls, value: 'FileIngestionOptionsDict') -> 'Self': + def from_dict(cls, value: "FileIngestionOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FileIngestionOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FileIngestionOptionsDict": + return _transform_to_json_value(self) # type:ignore class FileIngestionOptionsDict(TypedDict, total=False): @@ -292,7 +181,9 @@ class FileIngestionOptionsDict(TypedDict, total=False): [Private Preview] """ - schema_evolution_mode: VariableOrOptional[FileIngestionOptionsSchemaEvolutionModeParam] + schema_evolution_mode: VariableOrOptional[ + FileIngestionOptionsSchemaEvolutionModeParam + ] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/file_ingestion_options_file_format.py b/python/databricks/bundles/pipelines/_models/file_ingestion_options_file_format.py index f5aeae73670..25838583512 100644 --- a/python/databricks/bundles/pipelines/_models/file_ingestion_options_file_format.py +++ b/python/databricks/bundles/pipelines/_models/file_ingestion_options_file_format.py @@ -1,142 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class FileIngestionOptionsFileFormat(Enum): """ :meta private: [EXPERIMENTAL] """ + BINARYFILE = "BINARYFILE" JSON = "JSON" CSV = "CSV" @@ -146,4 +18,8 @@ class FileIngestionOptionsFileFormat(Enum): AVRO = "AVRO" ORC = "ORC" -FileIngestionOptionsFileFormatParam = Literal["BINARYFILE", "JSON", "CSV", "XML", "EXCEL", "PARQUET", "AVRO", "ORC"] | FileIngestionOptionsFileFormat + +FileIngestionOptionsFileFormatParam = ( + Literal["BINARYFILE", "JSON", "CSV", "XML", "EXCEL", "PARQUET", "AVRO", "ORC"] + | FileIngestionOptionsFileFormat +) diff --git a/python/databricks/bundles/pipelines/_models/file_ingestion_options_schema_evolution_mode.py b/python/databricks/bundles/pipelines/_models/file_ingestion_options_schema_evolution_mode.py index 6bd172b5a16..6d136255c5f 100644 --- a/python/databricks/bundles/pipelines/_models/file_ingestion_options_schema_evolution_mode.py +++ b/python/databricks/bundles/pipelines/_models/file_ingestion_options_schema_evolution_mode.py @@ -1,146 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class FileIngestionOptionsSchemaEvolutionMode(Enum): """ Based on https://docs.databricks.com/aws/en/ingestion/cloud-object-storage/auto-loader/schema#how-does-auto-loader-schema-evolution-work """ + ADD_NEW_COLUMNS_WITH_TYPE_WIDENING = "ADD_NEW_COLUMNS_WITH_TYPE_WIDENING" ADD_NEW_COLUMNS = "ADD_NEW_COLUMNS" RESCUE = "RESCUE" FAIL_ON_NEW_COLUMNS = "FAIL_ON_NEW_COLUMNS" NONE = "NONE" -FileIngestionOptionsSchemaEvolutionModeParam = Literal["ADD_NEW_COLUMNS_WITH_TYPE_WIDENING", "ADD_NEW_COLUMNS", "RESCUE", "FAIL_ON_NEW_COLUMNS", "NONE"] | FileIngestionOptionsSchemaEvolutionMode + +FileIngestionOptionsSchemaEvolutionModeParam = ( + Literal[ + "ADD_NEW_COLUMNS_WITH_TYPE_WIDENING", + "ADD_NEW_COLUMNS", + "RESCUE", + "FAIL_ON_NEW_COLUMNS", + "NONE", + ] + | FileIngestionOptionsSchemaEvolutionMode +) diff --git a/python/databricks/bundles/pipelines/_models/file_library.py b/python/databricks/bundles/pipelines/_models/file_library.py index 96e31a7749e..4fbf2a564af 100644 --- a/python/databricks/bundles/pipelines/_models/file_library.py +++ b/python/databricks/bundles/pipelines/_models/file_library.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class FileLibrary: """""" @@ -147,12 +21,11 @@ class FileLibrary: """ @classmethod - def from_dict(cls, value: 'FileLibraryDict') -> 'Self': + def from_dict(cls, value: "FileLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FileLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FileLibraryDict": + return _transform_to_json_value(self) # type:ignore class FileLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/filters.py b/python/databricks/bundles/pipelines/_models/filters.py index b5460ed2d84..c9dc61c8c0c 100644 --- a/python/databricks/bundles/pipelines/_models/filters.py +++ b/python/databricks/bundles/pipelines/_models/filters.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Filters: """""" @@ -152,12 +26,11 @@ class Filters: """ @classmethod - def from_dict(cls, value: 'FiltersDict') -> 'Self': + def from_dict(cls, value: "FiltersDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'FiltersDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "FiltersDict": + return _transform_to_json_value(self) # type:ignore class FiltersDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/gcp_attributes.py b/python/databricks/bundles/pipelines/_models/gcp_attributes.py index 5c7a0bc130b..70768dd4921 100644 --- a/python/databricks/bundles/pipelines/_models/gcp_attributes.py +++ b/python/databricks/bundles/pipelines/_models/gcp_attributes.py @@ -1,147 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.confidential_compute_type import ( + ConfidentialComputeType, + ConfidentialComputeTypeParam, +) +from databricks.bundles.pipelines._models.gcp_availability import ( + GcpAvailability, + GcpAvailabilityParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GcpAttributes: """ Attributes set during cluster creation which are related to GCP. """ + availability: VariableOrOptional[GcpAvailability] = None """ This field determines whether the spark executors will be scheduled to run on preemptible @@ -208,12 +91,11 @@ class GcpAttributes: """ @classmethod - def from_dict(cls, value: 'GcpAttributesDict') -> 'Self': + def from_dict(cls, value: "GcpAttributesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GcpAttributesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GcpAttributesDict": + return _transform_to_json_value(self) # type:ignore class GcpAttributesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/gcp_availability.py b/python/databricks/bundles/pipelines/_models/gcp_availability.py index 69a064121d9..0d391b87fe3 100644 --- a/python/databricks/bundles/pipelines/_models/gcp_availability.py +++ b/python/databricks/bundles/pipelines/_models/gcp_availability.py @@ -1,136 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class GcpAvailability(Enum): @@ -138,8 +9,13 @@ class GcpAvailability(Enum): This field determines whether the instance pool will contain preemptible VMs, on-demand VMs, or preemptible VMs with a fallback to on-demand VMs if the former is unavailable. """ + PREEMPTIBLE_GCP = "PREEMPTIBLE_GCP" ON_DEMAND_GCP = "ON_DEMAND_GCP" PREEMPTIBLE_WITH_FALLBACK_GCP = "PREEMPTIBLE_WITH_FALLBACK_GCP" -GcpAvailabilityParam = Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] | GcpAvailability + +GcpAvailabilityParam = ( + Literal["PREEMPTIBLE_GCP", "ON_DEMAND_GCP", "PREEMPTIBLE_WITH_FALLBACK_GCP"] + | GcpAvailability +) diff --git a/python/databricks/bundles/pipelines/_models/gcs_storage_info.py b/python/databricks/bundles/pipelines/_models/gcs_storage_info.py index 52828712507..e8f1059373b 100644 --- a/python/databricks/bundles/pipelines/_models/gcs_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/gcs_storage_info.py @@ -1,159 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GcsStorageInfo: """ A storage location in Google Cloud Platform's GCS """ + destination: VariableOr[str] """ GCS destination/URI, e.g. `gs://my-bucket/some-prefix` """ @classmethod - def from_dict(cls, value: 'GcsStorageInfoDict') -> 'Self': + def from_dict(cls, value: "GcsStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GcsStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GcsStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class GcsStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/google_ads_config.py b/python/databricks/bundles/pipelines/_models/google_ads_config.py index 23d17863d03..3ddb52b9667 100644 --- a/python/databricks/bundles/pipelines/_models/google_ads_config.py +++ b/python/databricks/bundles/pipelines/_models/google_ads_config.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GoogleAdsConfig: """ :meta private: [EXPERIMENTAL] """ + manager_account_id: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -154,12 +29,11 @@ class GoogleAdsConfig: """ @classmethod - def from_dict(cls, value: 'GoogleAdsConfigDict') -> 'Self': + def from_dict(cls, value: "GoogleAdsConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GoogleAdsConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GoogleAdsConfigDict": + return _transform_to_json_value(self) # type:ignore class GoogleAdsConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/google_ads_custom_report_options.py b/python/databricks/bundles/pipelines/_models/google_ads_custom_report_options.py index 096fcbf0414..441a8947342 100644 --- a/python/databricks/bundles/pipelines/_models/google_ads_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/google_ads_custom_report_options.py @@ -1,147 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GoogleAdsCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + User-defined custom report for the Google Ads connector. Mirrors the resource + fields + segments + metrics model that Google Ads GAQL exposes. The customer account this report runs against is supplied by the source @@ -151,6 +25,7 @@ class GoogleAdsCustomReportOptions: Only supported on table-type objects: a custom report requires a destination table, so it cannot be specified at the schema/source level. """ + resource: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -190,12 +65,11 @@ class GoogleAdsCustomReportOptions: """ @classmethod - def from_dict(cls, value: 'GoogleAdsCustomReportOptionsDict') -> 'Self': + def from_dict(cls, value: "GoogleAdsCustomReportOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GoogleAdsCustomReportOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GoogleAdsCustomReportOptionsDict": + return _transform_to_json_value(self) # type:ignore class GoogleAdsCustomReportOptionsDict(TypedDict, total=False): @@ -240,4 +114,6 @@ class GoogleAdsCustomReportOptionsDict(TypedDict, total=False): """ -GoogleAdsCustomReportOptionsParam = GoogleAdsCustomReportOptionsDict | GoogleAdsCustomReportOptions +GoogleAdsCustomReportOptionsParam = ( + GoogleAdsCustomReportOptionsDict | GoogleAdsCustomReportOptions +) diff --git a/python/databricks/bundles/pipelines/_models/google_ads_options.py b/python/databricks/bundles/pipelines/_models/google_ads_options.py index 3edf051c4c9..33d94383803 100644 --- a/python/databricks/bundles/pipelines/_models/google_ads_options.py +++ b/python/databricks/bundles/pipelines/_models/google_ads_options.py @@ -1,151 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.pipelines._models.google_ads_custom_report_options import ( + GoogleAdsCustomReportOptions, + GoogleAdsCustomReportOptionsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GoogleAdsOptions: """ :meta private: [EXPERIMENTAL] - + Google Ads specific options for ingestion (object-level). When set, these values override the corresponding fields in GoogleAdsConfig (source_configurations). """ + manager_account_id: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -183,12 +62,11 @@ class GoogleAdsOptions: """ @classmethod - def from_dict(cls, value: 'GoogleAdsOptionsDict') -> 'Self': + def from_dict(cls, value: "GoogleAdsOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GoogleAdsOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GoogleAdsOptionsDict": + return _transform_to_json_value(self) # type:ignore class GoogleAdsOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/google_drive_options.py b/python/databricks/bundles/pipelines/_models/google_drive_options.py index c4d3cfbfbdc..e11b5048555 100644 --- a/python/databricks/bundles/pipelines/_models/google_drive_options.py +++ b/python/databricks/bundles/pipelines/_models/google_drive_options.py @@ -1,147 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.file_ingestion_options import ( + FileIngestionOptions, + FileIngestionOptionsParam, +) +from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import ( + GoogleDriveOptionsGoogleDriveEntityType, + GoogleDriveOptionsGoogleDriveEntityTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class GoogleDriveOptions: """ :meta private: [EXPERIMENTAL] """ + entity_type: VariableOrOptional[GoogleDriveOptionsGoogleDriveEntityType] = None """ :meta private: [EXPERIMENTAL] @@ -164,12 +47,11 @@ class GoogleDriveOptions: """ @classmethod - def from_dict(cls, value: 'GoogleDriveOptionsDict') -> 'Self': + def from_dict(cls, value: "GoogleDriveOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'GoogleDriveOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "GoogleDriveOptionsDict": + return _transform_to_json_value(self) # type:ignore class GoogleDriveOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/google_drive_options_google_drive_entity_type.py b/python/databricks/bundles/pipelines/_models/google_drive_options_google_drive_entity_type.py index a6576134acd..b3b9847c550 100644 --- a/python/databricks/bundles/pipelines/_models/google_drive_options_google_drive_entity_type.py +++ b/python/databricks/bundles/pipelines/_models/google_drive_options_google_drive_entity_type.py @@ -1,144 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class GoogleDriveOptionsGoogleDriveEntityType(Enum): """ :meta private: [EXPERIMENTAL] """ + FILE = "FILE" FILE_METADATA = "FILE_METADATA" PERMISSION = "PERMISSION" -GoogleDriveOptionsGoogleDriveEntityTypeParam = Literal["FILE", "FILE_METADATA", "PERMISSION"] | GoogleDriveOptionsGoogleDriveEntityType + +GoogleDriveOptionsGoogleDriveEntityTypeParam = ( + Literal["FILE", "FILE_METADATA", "PERMISSION"] + | GoogleDriveOptionsGoogleDriveEntityType +) diff --git a/python/databricks/bundles/pipelines/_models/ingestion_config.py b/python/databricks/bundles/pipelines/_models/ingestion_config.py index e9de551c2bf..1d5a14cceef 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_config.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_config.py @@ -1,142 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecParam +from databricks.bundles.pipelines._models.schema_spec import ( + SchemaSpec, + SchemaSpecParam, +) +from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecParam if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class IngestionConfig: """""" @@ -157,12 +37,11 @@ class IngestionConfig: """ @classmethod - def from_dict(cls, value: 'IngestionConfigDict') -> 'Self': + def from_dict(cls, value: "IngestionConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'IngestionConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "IngestionConfigDict": + return _transform_to_json_value(self) # type:ignore class IngestionConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/ingestion_gateway_pipeline_definition.py b/python/databricks/bundles/pipelines/_models/ingestion_gateway_pipeline_definition.py index 55374926b31..56282a61fea 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_gateway_pipeline_definition.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_gateway_pipeline_definition.py @@ -1,147 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.pipelines._models.connection_parameters import ( + ConnectionParameters, + ConnectionParametersParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class IngestionGatewayPipelineDefinition: """ :meta private: [EXPERIMENTAL] """ + connection_name: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -180,12 +59,11 @@ class IngestionGatewayPipelineDefinition: """ @classmethod - def from_dict(cls, value: 'IngestionGatewayPipelineDefinitionDict') -> 'Self': + def from_dict(cls, value: "IngestionGatewayPipelineDefinitionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'IngestionGatewayPipelineDefinitionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "IngestionGatewayPipelineDefinitionDict": + return _transform_to_json_value(self) # type:ignore class IngestionGatewayPipelineDefinitionDict(TypedDict, total=False): @@ -229,4 +107,6 @@ class IngestionGatewayPipelineDefinitionDict(TypedDict, total=False): """ -IngestionGatewayPipelineDefinitionParam = IngestionGatewayPipelineDefinitionDict | IngestionGatewayPipelineDefinition +IngestionGatewayPipelineDefinitionParam = ( + IngestionGatewayPipelineDefinitionDict | IngestionGatewayPipelineDefinition +) diff --git a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition.py b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition.py index 382a227a495..605a119255e 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition.py @@ -1,142 +1,40 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.pipelines._models.connector_type import ( + ConnectorType, + ConnectorTypeParam, +) +from databricks.bundles.pipelines._models.data_staging_options import ( + DataStagingOptions, + DataStagingOptionsParam, +) +from databricks.bundles.pipelines._models.ingestion_config import ( + IngestionConfig, + IngestionConfigParam, +) +from databricks.bundles.pipelines._models.operation_time_window import ( + OperationTimeWindow, + OperationTimeWindowParam, +) +from databricks.bundles.pipelines._models.source_config import ( + SourceConfig, + SourceConfigParam, +) +from databricks.bundles.pipelines._models.table_specific_config import ( + TableSpecificConfig, + TableSpecificConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class IngestionPipelineDefinition: """""" @@ -212,12 +110,11 @@ class IngestionPipelineDefinition: """ @classmethod - def from_dict(cls, value: 'IngestionPipelineDefinitionDict') -> 'Self': + def from_dict(cls, value: "IngestionPipelineDefinitionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'IngestionPipelineDefinitionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "IngestionPipelineDefinitionDict": + return _transform_to_json_value(self) # type:ignore class IngestionPipelineDefinitionDict(TypedDict, total=False): @@ -294,4 +191,6 @@ class IngestionPipelineDefinitionDict(TypedDict, total=False): """ -IngestionPipelineDefinitionParam = IngestionPipelineDefinitionDict | IngestionPipelineDefinition +IngestionPipelineDefinitionParam = ( + IngestionPipelineDefinitionDict | IngestionPipelineDefinition +) diff --git a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_fanout_options.py b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_fanout_options.py index f73140a7e37..092b91581ed 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_fanout_options.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_fanout_options.py @@ -1,142 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.pipelines._models.transformer import ( + Transformer, + TransformerParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class IngestionPipelineDefinitionFanoutOptions: """ @@ -145,6 +23,7 @@ class IngestionPipelineDefinitionFanoutOptions: key derived from the record. The key value becomes the table name suffix: {destination_catalog}.{destination_schema}.{key_value}. """ + fanout_by: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -163,12 +42,11 @@ class IngestionPipelineDefinitionFanoutOptions: """ @classmethod - def from_dict(cls, value: 'IngestionPipelineDefinitionFanoutOptionsDict') -> 'Self': + def from_dict(cls, value: "IngestionPipelineDefinitionFanoutOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'IngestionPipelineDefinitionFanoutOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "IngestionPipelineDefinitionFanoutOptionsDict": + return _transform_to_json_value(self) # type:ignore class IngestionPipelineDefinitionFanoutOptionsDict(TypedDict, total=False): @@ -192,4 +70,7 @@ class IngestionPipelineDefinitionFanoutOptionsDict(TypedDict, total=False): """ -IngestionPipelineDefinitionFanoutOptionsParam = IngestionPipelineDefinitionFanoutOptionsDict | IngestionPipelineDefinitionFanoutOptions +IngestionPipelineDefinitionFanoutOptionsParam = ( + IngestionPipelineDefinitionFanoutOptionsDict + | IngestionPipelineDefinitionFanoutOptions +) diff --git a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_table_specific_config_query_based_connector_config.py b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_table_specific_config_query_based_connector_config.py index 183a0e7e6b7..a784bb6e8b4 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_table_specific_config_query_based_connector_config.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_table_specific_config_query_based_connector_config.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig: """ Configurations that are only applicable for query-based ingestion connectors. """ + cursor_columns: VariableOrList[str] = field(default_factory=list) """ [Public Preview] The names of the monotonically increasing columns in the source table that are used to enable @@ -178,15 +53,21 @@ class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig: """ @classmethod - def from_dict(cls, value: 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict') -> 'Self': + def from_dict( + cls, + value: "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict", + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict( + self, + ) -> "IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict": + return _transform_to_json_value(self) # type:ignore -class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict(TypedDict, total=False): +class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict( + TypedDict, total=False +): """""" cursor_columns: VariableOrList[str] @@ -225,4 +106,7 @@ class IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDic """ -IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam = IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict | IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig +IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam = ( + IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict + | IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig +) diff --git a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_workday_report_parameters.py b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_workday_report_parameters.py index 4452dc0d37b..a819adcf236 100644 --- a/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_workday_report_parameters.py +++ b/python/databricks/bundles/pipelines/_models/ingestion_pipeline_definition_workday_report_parameters.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrDict if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class IngestionPipelineDefinitionWorkdayReportParameters: """ :meta private: [EXPERIMENTAL] """ + parameters: VariableOrDict[str] = field(default_factory=dict) """ :meta private: [EXPERIMENTAL] @@ -156,12 +31,13 @@ class IngestionPipelineDefinitionWorkdayReportParameters: """ @classmethod - def from_dict(cls, value: 'IngestionPipelineDefinitionWorkdayReportParametersDict') -> 'Self': + def from_dict( + cls, value: "IngestionPipelineDefinitionWorkdayReportParametersDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'IngestionPipelineDefinitionWorkdayReportParametersDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "IngestionPipelineDefinitionWorkdayReportParametersDict": + return _transform_to_json_value(self) # type:ignore class IngestionPipelineDefinitionWorkdayReportParametersDict(TypedDict, total=False): @@ -181,4 +57,7 @@ class IngestionPipelineDefinitionWorkdayReportParametersDict(TypedDict, total=Fa """ -IngestionPipelineDefinitionWorkdayReportParametersParam = IngestionPipelineDefinitionWorkdayReportParametersDict | IngestionPipelineDefinitionWorkdayReportParameters +IngestionPipelineDefinitionWorkdayReportParametersParam = ( + IngestionPipelineDefinitionWorkdayReportParametersDict + | IngestionPipelineDefinitionWorkdayReportParameters +) diff --git a/python/databricks/bundles/pipelines/_models/init_script_info.py b/python/databricks/bundles/pipelines/_models/init_script_info.py index 0d8eda0cc7e..f6c696d11bd 100644 --- a/python/databricks/bundles/pipelines/_models/init_script_info.py +++ b/python/databricks/bundles/pipelines/_models/init_script_info.py @@ -1,147 +1,50 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.adlsgen2_info import ( + Adlsgen2Info, + Adlsgen2InfoParam, +) +from databricks.bundles.pipelines._models.dbfs_storage_info import ( + DbfsStorageInfo, + DbfsStorageInfoParam, +) +from databricks.bundles.pipelines._models.gcs_storage_info import ( + GcsStorageInfo, + GcsStorageInfoParam, +) +from databricks.bundles.pipelines._models.local_file_info import ( + LocalFileInfo, + LocalFileInfoParam, +) +from databricks.bundles.pipelines._models.s3_storage_info import ( + S3StorageInfo, + S3StorageInfoParam, +) +from databricks.bundles.pipelines._models.volumes_storage_info import ( + VolumesStorageInfo, + VolumesStorageInfoParam, +) +from databricks.bundles.pipelines._models.workspace_storage_info import ( + WorkspaceStorageInfo, + WorkspaceStorageInfoParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class InitScriptInfo: """ Config for an individual init script """ + abfss: VariableOrOptional[Adlsgen2Info] = None """ Contains the Azure Data Lake Storage destination path @@ -186,12 +89,11 @@ class InitScriptInfo: """ @classmethod - def from_dict(cls, value: 'InitScriptInfoDict') -> 'Self': + def from_dict(cls, value: "InitScriptInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'InitScriptInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "InitScriptInfoDict": + return _transform_to_json_value(self) # type:ignore class InitScriptInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/jira_connector_options.py b/python/databricks/bundles/pipelines/_models/jira_connector_options.py index 5eff3ae1c58..ef7ba677587 100644 --- a/python/databricks/bundles/pipelines/_models/jira_connector_options.py +++ b/python/databricks/bundles/pipelines/_models/jira_connector_options.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JiraConnectorOptions: """ Jira specific options for ingestion """ + include_jira_spaces: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -150,12 +25,11 @@ class JiraConnectorOptions: """ @classmethod - def from_dict(cls, value: 'JiraConnectorOptionsDict') -> 'Self': + def from_dict(cls, value: "JiraConnectorOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JiraConnectorOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JiraConnectorOptionsDict": + return _transform_to_json_value(self) # type:ignore class JiraConnectorOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/json_transformer_options.py b/python/databricks/bundles/pipelines/_models/json_transformer_options.py index aec0bd3b209..8359f564eb2 100644 --- a/python/databricks/bundles/pipelines/_models/json_transformer_options.py +++ b/python/databricks/bundles/pipelines/_models/json_transformer_options.py @@ -1,142 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import ( + FileIngestionOptionsSchemaEvolutionMode, + FileIngestionOptionsSchemaEvolutionModeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class JsonTransformerOptions: """""" @@ -155,7 +33,9 @@ class JsonTransformerOptions: [Beta] Inline schema string for JSON parsing (Spark DDL format). """ - schema_evolution_mode: VariableOrOptional[FileIngestionOptionsSchemaEvolutionMode] = None + schema_evolution_mode: VariableOrOptional[ + FileIngestionOptionsSchemaEvolutionMode + ] = None """ :meta private: [EXPERIMENTAL] @@ -177,12 +57,11 @@ class JsonTransformerOptions: """ @classmethod - def from_dict(cls, value: 'JsonTransformerOptionsDict') -> 'Self': + def from_dict(cls, value: "JsonTransformerOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'JsonTransformerOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "JsonTransformerOptionsDict": + return _transform_to_json_value(self) # type:ignore class JsonTransformerOptionsDict(TypedDict, total=False): @@ -202,7 +81,9 @@ class JsonTransformerOptionsDict(TypedDict, total=False): [Beta] Inline schema string for JSON parsing (Spark DDL format). """ - schema_evolution_mode: VariableOrOptional[FileIngestionOptionsSchemaEvolutionModeParam] + schema_evolution_mode: VariableOrOptional[ + FileIngestionOptionsSchemaEvolutionModeParam + ] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/kafka_options.py b/python/databricks/bundles/pipelines/_models/kafka_options.py index 30c4d017e3a..d42b585624d 100644 --- a/python/databricks/bundles/pipelines/_models/kafka_options.py +++ b/python/databricks/bundles/pipelines/_models/kafka_options.py @@ -1,142 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import ( + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.pipelines._models.transformer import ( + Transformer, + TransformerParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class KafkaOptions: """""" @@ -198,12 +80,11 @@ class KafkaOptions: """ @classmethod - def from_dict(cls, value: 'KafkaOptionsDict') -> 'Self': + def from_dict(cls, value: "KafkaOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'KafkaOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "KafkaOptionsDict": + return _transform_to_json_value(self) # type:ignore class KafkaOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/lifecycle.py b/python/databricks/bundles/pipelines/_models/lifecycle.py index de0d6e7ddfc..697776a198e 100644 --- a/python/databricks/bundles/pipelines/_models/lifecycle.py +++ b/python/databricks/bundles/pipelines/_models/lifecycle.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -147,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options.py index e012d648407..1d997abc919 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options.py @@ -1,153 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import ( + LinkedInAdsOptionsLinkedInAdsCustomReportOptions, + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LinkedInAdsOptions: """ :meta private: [EXPERIMENTAL] - + LinkedIn Ads specific options for ingestion. sync_start_date and lookback_window_days apply to both the prebuilt analytics tables and custom reports. custom_report_options defines a custom (user-defined) adAnalytics report and is only valid on a table object. """ - custom_report_options: VariableOrOptional[LinkedInAdsOptionsLinkedInAdsCustomReportOptions] = None + + custom_report_options: VariableOrOptional[ + LinkedInAdsOptionsLinkedInAdsCustomReportOptions + ] = None """ :meta private: [EXPERIMENTAL] @@ -177,18 +58,19 @@ class LinkedInAdsOptions: """ @classmethod - def from_dict(cls, value: 'LinkedInAdsOptionsDict') -> 'Self': + def from_dict(cls, value: "LinkedInAdsOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LinkedInAdsOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LinkedInAdsOptionsDict": + return _transform_to_json_value(self) # type:ignore class LinkedInAdsOptionsDict(TypedDict, total=False): """""" - custom_report_options: VariableOrOptional[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam] + custom_report_options: VariableOrOptional[ + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam + ] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options.py index 5abb7047fb2..1420eb50d50 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options.py @@ -1,160 +1,55 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import ( + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam, +) +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import ( + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam, +) +from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import ( + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LinkedInAdsOptionsLinkedInAdsCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + User-defined custom report for the LinkedIn Ads connector. The destination table name comes from the enclosing TableSpec.destination_table, the start date from the enclosing LinkedInAdsOptions.sync_start_date, and the account it runs against from the source schema (namespace) -- none are repeated here. """ - finder: VariableOr[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder] + + finder: VariableOr[ + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder + ] """ :meta private: [EXPERIMENTAL] [Private Preview] (Required) adAnalytics finder. See LinkedInAdsFinder. """ - entity_granularity: VariableOrList[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity] = field(default_factory=list) + entity_granularity: VariableOrList[ + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity + ] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -174,7 +69,9 @@ class LinkedInAdsOptionsLinkedInAdsCustomReportOptions: RevenueAttributionMetrics struct). """ - time_granularity: VariableOrOptional[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity] = None + time_granularity: VariableOrOptional[ + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity + ] = None """ :meta private: [EXPERIMENTAL] @@ -183,25 +80,30 @@ class LinkedInAdsOptionsLinkedInAdsCustomReportOptions: """ @classmethod - def from_dict(cls, value: 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict') -> 'Self': + def from_dict( + cls, value: "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict": + return _transform_to_json_value(self) # type:ignore class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict(TypedDict, total=False): """""" - finder: VariableOr[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam] + finder: VariableOr[ + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam + ] """ :meta private: [EXPERIMENTAL] [Private Preview] (Required) adAnalytics finder. See LinkedInAdsFinder. """ - entity_granularity: VariableOrList[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam] + entity_granularity: VariableOrList[ + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam + ] """ :meta private: [EXPERIMENTAL] @@ -221,7 +123,9 @@ class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict(TypedDict, total=Fals RevenueAttributionMetrics struct). """ - time_granularity: VariableOrOptional[LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam] + time_granularity: VariableOrOptional[ + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam + ] """ :meta private: [EXPERIMENTAL] @@ -230,4 +134,7 @@ class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict(TypedDict, total=Fals """ -LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam = LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict | LinkedInAdsOptionsLinkedInAdsCustomReportOptions +LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam = ( + LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict + | LinkedInAdsOptionsLinkedInAdsCustomReportOptions +) diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity.py index e737ceaaa7f..ae32b0b83a4 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity.py @@ -1,146 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - - -class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity(Enum): +class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity( + Enum +): """ :meta private: [EXPERIMENTAL] - + Entity pivot to group by. """ + CAMPAIGN = "CAMPAIGN" CREATIVE = "CREATIVE" CAMPAIGN_GROUP = "CAMPAIGN_GROUP" -LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam = Literal["CAMPAIGN", "CREATIVE", "CAMPAIGN_GROUP"] | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity + +LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam = ( + Literal["CAMPAIGN", "CREATIVE", "CAMPAIGN_GROUP"] + | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity +) diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder.py index 239eee96e80..9abf3e9d7be 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder.py @@ -1,147 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder(Enum): """ :meta private: [EXPERIMENTAL] - + adAnalytics finder. Determines call shape, valid pivots, and metric requirements. """ + ANALYTICS = "ANALYTICS" STATISTICS = "STATISTICS" ATTRIBUTED_REVENUE_METRICS = "ATTRIBUTED_REVENUE_METRICS" -LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam = Literal["ANALYTICS", "STATISTICS", "ATTRIBUTED_REVENUE_METRICS"] | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder + +LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam = ( + Literal["ANALYTICS", "STATISTICS", "ATTRIBUTED_REVENUE_METRICS"] + | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder +) diff --git a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity.py b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity.py index 5ef329b2449..256974843ea 100644 --- a/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity.py +++ b/python/databricks/bundles/pipelines/_models/linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity.py @@ -1,148 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity(Enum): """ :meta private: [EXPERIMENTAL] - + Time aggregation. Used by analytics/statistics; ignored for attributedRevenueMetrics. Defaults to DAILY when unspecified. """ + ALL = "ALL" DAILY = "DAILY" MONTHLY = "MONTHLY" YEARLY = "YEARLY" -LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam = Literal["ALL", "DAILY", "MONTHLY", "YEARLY"] | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity + +LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam = ( + Literal["ALL", "DAILY", "MONTHLY", "YEARLY"] + | LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity +) diff --git a/python/databricks/bundles/pipelines/_models/local_file_info.py b/python/databricks/bundles/pipelines/_models/local_file_info.py index 09965373829..875f2fe8353 100644 --- a/python/databricks/bundles/pipelines/_models/local_file_info.py +++ b/python/databricks/bundles/pipelines/_models/local_file_info.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LocalFileInfo: """""" @@ -147,12 +21,11 @@ class LocalFileInfo: """ @classmethod - def from_dict(cls, value: 'LocalFileInfoDict') -> 'Self': + def from_dict(cls, value: "LocalFileInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LocalFileInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LocalFileInfoDict": + return _transform_to_json_value(self) # type:ignore class LocalFileInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/log_analytics_info.py b/python/databricks/bundles/pipelines/_models/log_analytics_info.py index e4ab2bcbb48..67cdf85b2b9 100644 --- a/python/databricks/bundles/pipelines/_models/log_analytics_info.py +++ b/python/databricks/bundles/pipelines/_models/log_analytics_info.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LogAnalyticsInfo: """""" @@ -152,12 +26,11 @@ class LogAnalyticsInfo: """ @classmethod - def from_dict(cls, value: 'LogAnalyticsInfoDict') -> 'Self': + def from_dict(cls, value: "LogAnalyticsInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LogAnalyticsInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LogAnalyticsInfoDict": + return _transform_to_json_value(self) # type:ignore class LogAnalyticsInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/marketo_options.py b/python/databricks/bundles/pipelines/_models/marketo_options.py index 14776f340fb..943dba4ff7b 100644 --- a/python/databricks/bundles/pipelines/_models/marketo_options.py +++ b/python/databricks/bundles/pipelines/_models/marketo_options.py @@ -1,149 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MarketoOptions: """ :meta private: [EXPERIMENTAL] - + Marketo specific options for ingestion """ + sync_start_date: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -154,12 +29,11 @@ class MarketoOptions: """ @classmethod - def from_dict(cls, value: 'MarketoOptionsDict') -> 'Self': + def from_dict(cls, value: "MarketoOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MarketoOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MarketoOptionsDict": + return _transform_to_json_value(self) # type:ignore class MarketoOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/maven_library.py b/python/databricks/bundles/pipelines/_models/maven_library.py index 97368254302..25be1599c28 100644 --- a/python/databricks/bundles/pipelines/_models/maven_library.py +++ b/python/databricks/bundles/pipelines/_models/maven_library.py @@ -1,142 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MavenLibrary: """""" @@ -161,12 +39,11 @@ class MavenLibrary: """ @classmethod - def from_dict(cls, value: 'MavenLibraryDict') -> 'Self': + def from_dict(cls, value: "MavenLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MavenLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MavenLibraryDict": + return _transform_to_json_value(self) # type:ignore class MavenLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/meta_marketing_options.py b/python/databricks/bundles/pipelines/_models/meta_marketing_options.py index e87a1e8165c..9a9f657bf5f 100644 --- a/python/databricks/bundles/pipelines/_models/meta_marketing_options.py +++ b/python/databricks/bundles/pipelines/_models/meta_marketing_options.py @@ -1,147 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import ( + MetaMarketingOptionsMetaMarketingCustomReportOptions, + MetaMarketingOptionsMetaMarketingCustomReportOptionsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MetaMarketingOptions: """ Meta Marketing (Meta Ads) specific options for ingestion """ + action_attribution_windows: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -180,7 +59,9 @@ class MetaMarketingOptions: updated conversion data from the API, shared by prebuilt and custom reports. """ - custom_report_options: VariableOrOptional[MetaMarketingOptionsMetaMarketingCustomReportOptions] = None + custom_report_options: VariableOrOptional[ + MetaMarketingOptionsMetaMarketingCustomReportOptions + ] = None """ :meta private: [EXPERIMENTAL] @@ -214,12 +95,11 @@ class MetaMarketingOptions: """ @classmethod - def from_dict(cls, value: 'MetaMarketingOptionsDict') -> 'Self': + def from_dict(cls, value: "MetaMarketingOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MetaMarketingOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MetaMarketingOptionsDict": + return _transform_to_json_value(self) # type:ignore class MetaMarketingOptionsDict(TypedDict, total=False): @@ -263,7 +143,9 @@ class MetaMarketingOptionsDict(TypedDict, total=False): updated conversion data from the API, shared by prebuilt and custom reports. """ - custom_report_options: VariableOrOptional[MetaMarketingOptionsMetaMarketingCustomReportOptionsParam] + custom_report_options: VariableOrOptional[ + MetaMarketingOptionsMetaMarketingCustomReportOptionsParam + ] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/meta_marketing_options_meta_marketing_custom_report_options.py b/python/databricks/bundles/pipelines/_models/meta_marketing_options_meta_marketing_custom_report_options.py index 9d9d0fda95b..a01a24ec45b 100644 --- a/python/databricks/bundles/pipelines/_models/meta_marketing_options_meta_marketing_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/meta_marketing_options_meta_marketing_custom_report_options.py @@ -1,151 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MetaMarketingOptionsMetaMarketingCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + Defines the shape of a single Meta Ads custom report (one /insights call shape). start_date, custom_insights_lookback_window live on MetaMarketingOptions, not here. Metrics are not customer-selectable; the connector returns a fixed standard metric set. """ + action_attribution_windows: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -189,12 +64,13 @@ class MetaMarketingOptionsMetaMarketingCustomReportOptions: """ @classmethod - def from_dict(cls, value: 'MetaMarketingOptionsMetaMarketingCustomReportOptionsDict') -> 'Self': + def from_dict( + cls, value: "MetaMarketingOptionsMetaMarketingCustomReportOptionsDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MetaMarketingOptionsMetaMarketingCustomReportOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MetaMarketingOptionsMetaMarketingCustomReportOptionsDict": + return _transform_to_json_value(self) # type:ignore class MetaMarketingOptionsMetaMarketingCustomReportOptionsDict(TypedDict, total=False): @@ -243,4 +119,7 @@ class MetaMarketingOptionsMetaMarketingCustomReportOptionsDict(TypedDict, total= """ -MetaMarketingOptionsMetaMarketingCustomReportOptionsParam = MetaMarketingOptionsMetaMarketingCustomReportOptionsDict | MetaMarketingOptionsMetaMarketingCustomReportOptions +MetaMarketingOptionsMetaMarketingCustomReportOptionsParam = ( + MetaMarketingOptionsMetaMarketingCustomReportOptionsDict + | MetaMarketingOptionsMetaMarketingCustomReportOptions +) diff --git a/python/databricks/bundles/pipelines/_models/notebook_library.py b/python/databricks/bundles/pipelines/_models/notebook_library.py index 503014c0709..72ded094151 100644 --- a/python/databricks/bundles/pipelines/_models/notebook_library.py +++ b/python/databricks/bundles/pipelines/_models/notebook_library.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class NotebookLibrary: """""" @@ -147,12 +21,11 @@ class NotebookLibrary: """ @classmethod - def from_dict(cls, value: 'NotebookLibraryDict') -> 'Self': + def from_dict(cls, value: "NotebookLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'NotebookLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "NotebookLibraryDict": + return _transform_to_json_value(self) # type:ignore class NotebookLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/notifications.py b/python/databricks/bundles/pipelines/_models/notifications.py index 47b12875b76..271bb4c1fdb 100644 --- a/python/databricks/bundles/pipelines/_models/notifications.py +++ b/python/databricks/bundles/pipelines/_models/notifications.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Notifications: """""" @@ -158,12 +32,11 @@ class Notifications: """ @classmethod - def from_dict(cls, value: 'NotificationsDict') -> 'Self': + def from_dict(cls, value: "NotificationsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'NotificationsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "NotificationsDict": + return _transform_to_json_value(self) # type:ignore class NotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/operation_time_window.py b/python/databricks/bundles/pipelines/_models/operation_time_window.py index a7dd7dce6d1..cc3c315d6de 100644 --- a/python/databricks/bundles/pipelines/_models/operation_time_window.py +++ b/python/databricks/bundles/pipelines/_models/operation_time_window.py @@ -1,147 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class OperationTimeWindow: """ Proto representing a window """ + start_hour: VariableOr[int] """ [Public Preview] An integer between 0 and 23 denoting the start hour for the window in the 24-hour day. @@ -160,12 +40,11 @@ class OperationTimeWindow: """ @classmethod - def from_dict(cls, value: 'OperationTimeWindowDict') -> 'Self': + def from_dict(cls, value: "OperationTimeWindowDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'OperationTimeWindowDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "OperationTimeWindowDict": + return _transform_to_json_value(self) # type:ignore class OperationTimeWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/outlook_attachment_mode.py b/python/databricks/bundles/pipelines/_models/outlook_attachment_mode.py index 98a0b153d21..5d3c5fa042f 100644 --- a/python/databricks/bundles/pipelines/_models/outlook_attachment_mode.py +++ b/python/databricks/bundles/pipelines/_models/outlook_attachment_mode.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class OutlookAttachmentMode(Enum): """ :meta private: [EXPERIMENTAL] - + Attachment behavior mode for Outlook ingestion """ + ALL = "ALL" NON_INLINE_ONLY = "NON_INLINE_ONLY" INLINE_ONLY = "INLINE_ONLY" NONE = "NONE" -OutlookAttachmentModeParam = Literal["ALL", "NON_INLINE_ONLY", "INLINE_ONLY", "NONE"] | OutlookAttachmentMode + +OutlookAttachmentModeParam = ( + Literal["ALL", "NON_INLINE_ONLY", "INLINE_ONLY", "NONE"] | OutlookAttachmentMode +) diff --git a/python/databricks/bundles/pipelines/_models/outlook_body_format.py b/python/databricks/bundles/pipelines/_models/outlook_body_format.py index c94bcf6e14e..e5f409e0412 100644 --- a/python/databricks/bundles/pipelines/_models/outlook_body_format.py +++ b/python/databricks/bundles/pipelines/_models/outlook_body_format.py @@ -1,145 +1,18 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class OutlookBodyFormat(Enum): """ :meta private: [EXPERIMENTAL] - + Body format for Outlook email content """ + TEXT_HTML = "TEXT_HTML" TEXT_PLAIN = "TEXT_PLAIN" + OutlookBodyFormatParam = Literal["TEXT_HTML", "TEXT_PLAIN"] | OutlookBodyFormat diff --git a/python/databricks/bundles/pipelines/_models/outlook_options.py b/python/databricks/bundles/pipelines/_models/outlook_options.py index 63247123aa9..98e93a16a69 100644 --- a/python/databricks/bundles/pipelines/_models/outlook_options.py +++ b/python/databricks/bundles/pipelines/_models/outlook_options.py @@ -1,149 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.pipelines._models.outlook_attachment_mode import ( + OutlookAttachmentMode, + OutlookAttachmentModeParam, +) +from databricks.bundles.pipelines._models.outlook_body_format import ( + OutlookBodyFormat, + OutlookBodyFormatParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class OutlookOptions: """ :meta private: [EXPERIMENTAL] - + Outlook specific options for ingestion """ + attachment_mode: VariableOrOptional[OutlookAttachmentMode] = None """ :meta private: [EXPERIMENTAL] @@ -212,12 +95,11 @@ class OutlookOptions: """ @classmethod - def from_dict(cls, value: 'OutlookOptionsDict') -> 'Self': + def from_dict(cls, value: "OutlookOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'OutlookOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "OutlookOptionsDict": + return _transform_to_json_value(self) # type:ignore class OutlookOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/parse_mode.py b/python/databricks/bundles/pipelines/_models/parse_mode.py index 73726d967a7..2e55f31226d 100644 --- a/python/databricks/bundles/pipelines/_models/parse_mode.py +++ b/python/databricks/bundles/pipelines/_models/parse_mode.py @@ -1,143 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class ParseMode(Enum): """ Determines how errors encountered while deserializing records are handled. """ + FAILFAST = "FAILFAST" PERMISSIVE = "PERMISSIVE" + ParseModeParam = Literal["FAILFAST", "PERMISSIVE"] | ParseMode diff --git a/python/databricks/bundles/pipelines/_models/path_pattern.py b/python/databricks/bundles/pipelines/_models/path_pattern.py index 579e10436ec..6df8c4c5a3f 100644 --- a/python/databricks/bundles/pipelines/_models/path_pattern.py +++ b/python/databricks/bundles/pipelines/_models/path_pattern.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PathPattern: """""" @@ -147,12 +21,11 @@ class PathPattern: """ @classmethod - def from_dict(cls, value: 'PathPatternDict') -> 'Self': + def from_dict(cls, value: "PathPatternDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PathPatternDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PathPatternDict": + return _transform_to_json_value(self) # type:ignore class PathPatternDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline.py b/python/databricks/bundles/pipelines/_models/pipeline.py index 676a4f511fb..5e4b04bb726 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline.py +++ b/python/databricks/bundles/pipelines/_models/pipeline.py @@ -1,142 +1,63 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import ( + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.pipelines._models.event_log_spec import ( + EventLogSpec, + EventLogSpecParam, +) +from databricks.bundles.pipelines._models.filters import ( + Filters, + FiltersParam, +) +from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import ( + IngestionGatewayPipelineDefinition, + IngestionGatewayPipelineDefinitionParam, +) +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import ( + IngestionPipelineDefinition, + IngestionPipelineDefinitionParam, +) +from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.pipelines._models.notifications import ( + Notifications, + NotificationsParam, +) +from databricks.bundles.pipelines._models.pipeline_cluster import ( + PipelineCluster, + PipelineClusterParam, +) +from databricks.bundles.pipelines._models.pipeline_library import ( + PipelineLibrary, + PipelineLibraryParam, +) +from databricks.bundles.pipelines._models.pipeline_permission import ( + PipelinePermission, + PipelinePermissionParam, +) +from databricks.bundles.pipelines._models.pipelines_environment import ( + PipelinesEnvironment, + PipelinesEnvironmentParam, +) +from databricks.bundles.pipelines._models.restart_window import ( + RestartWindow, + RestartWindowParam, +) +from databricks.bundles.pipelines._models.run_as import RunAs, RunAsParam if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Pipeline(Resource): """""" @@ -344,12 +265,11 @@ class Pipeline(Resource): """ @classmethod - def from_dict(cls, value: 'PipelineDict') -> 'Self': + def from_dict(cls, value: "PipelineDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PipelineDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PipelineDict": + return _transform_to_json_value(self) # type:ignore class PipelineDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_cluster.py b/python/databricks/bundles/pipelines/_models/pipeline_cluster.py index 18d91182a27..6eda9e865a0 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_cluster.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_cluster.py @@ -1,142 +1,44 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import ( + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.pipelines._models.aws_attributes import ( + AwsAttributes, + AwsAttributesParam, +) +from databricks.bundles.pipelines._models.azure_attributes import ( + AzureAttributes, + AzureAttributesParam, +) +from databricks.bundles.pipelines._models.cluster_log_conf import ( + ClusterLogConf, + ClusterLogConfParam, +) +from databricks.bundles.pipelines._models.gcp_attributes import ( + GcpAttributes, + GcpAttributesParam, +) +from databricks.bundles.pipelines._models.init_script_info import ( + InitScriptInfo, + InitScriptInfoParam, +) +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import ( + PipelineClusterAutoscale, + PipelineClusterAutoscaleParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PipelineCluster: """""" @@ -277,12 +179,11 @@ class PipelineCluster: """ @classmethod - def from_dict(cls, value: 'PipelineClusterDict') -> 'Self': + def from_dict(cls, value: "PipelineClusterDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PipelineClusterDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PipelineClusterDict": + return _transform_to_json_value(self) # type:ignore class PipelineClusterDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale.py b/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale.py index 7ab741637fe..d5ba9da3665 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale.py @@ -1,142 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import ( + PipelineClusterAutoscaleMode, + PipelineClusterAutoscaleModeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PipelineClusterAutoscale: """""" @@ -162,12 +40,11 @@ class PipelineClusterAutoscale: """ @classmethod - def from_dict(cls, value: 'PipelineClusterAutoscaleDict') -> 'Self': + def from_dict(cls, value: "PipelineClusterAutoscaleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PipelineClusterAutoscaleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PipelineClusterAutoscaleDict": + return _transform_to_json_value(self) # type:ignore class PipelineClusterAutoscaleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale_mode.py b/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale_mode.py index 09bf9e1c806..52afa7d09d1 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale_mode.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_cluster_autoscale_mode.py @@ -1,136 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class PipelineClusterAutoscaleMode(Enum): @@ -141,7 +12,11 @@ class PipelineClusterAutoscaleMode(Enum): for `updates` clusters only. The legacy autoscaling feature is used for `maintenance` clusters. """ + ENHANCED = "ENHANCED" LEGACY = "LEGACY" -PipelineClusterAutoscaleModeParam = Literal["ENHANCED", "LEGACY"] | PipelineClusterAutoscaleMode + +PipelineClusterAutoscaleModeParam = ( + Literal["ENHANCED", "LEGACY"] | PipelineClusterAutoscaleMode +) diff --git a/python/databricks/bundles/pipelines/_models/pipeline_library.py b/python/databricks/bundles/pipelines/_models/pipeline_library.py index 4f5fb7d4787..5639f57fb2f 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_library.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_library.py @@ -1,142 +1,32 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.file_library import ( + FileLibrary, + FileLibraryParam, +) +from databricks.bundles.pipelines._models.maven_library import ( + MavenLibrary, + MavenLibraryParam, +) +from databricks.bundles.pipelines._models.notebook_library import ( + NotebookLibrary, + NotebookLibraryParam, +) +from databricks.bundles.pipelines._models.path_pattern import ( + PathPattern, + PathPatternParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PipelineLibrary: """""" @@ -173,12 +63,11 @@ class PipelineLibrary: """ @classmethod - def from_dict(cls, value: 'PipelineLibraryDict') -> 'Self': + def from_dict(cls, value: "PipelineLibraryDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PipelineLibraryDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PipelineLibraryDict": + return _transform_to_json_value(self) # type:ignore class PipelineLibraryDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_permission.py b/python/databricks/bundles/pipelines/_models/pipeline_permission.py index ea2809a4879..e7b9c90f94e 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_permission.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_permission.py @@ -1,142 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.pipelines._models.pipeline_permission_level import ( + PipelinePermissionLevel, + PipelinePermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PipelinePermission: """""" @@ -162,12 +40,11 @@ class PipelinePermission: """ @classmethod - def from_dict(cls, value: 'PipelinePermissionDict') -> 'Self': + def from_dict(cls, value: "PipelinePermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PipelinePermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PipelinePermissionDict": + return _transform_to_json_value(self) # type:ignore class PipelinePermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/pipeline_permission_level.py b/python/databricks/bundles/pipelines/_models/pipeline_permission_level.py index 02e1a4df38e..5c2d68bfe57 100644 --- a/python/databricks/bundles/pipelines/_models/pipeline_permission_level.py +++ b/python/databricks/bundles/pipelines/_models/pipeline_permission_level.py @@ -1,145 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class PipelinePermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" IS_OWNER = "IS_OWNER" CAN_RUN = "CAN_RUN" CAN_VIEW = "CAN_VIEW" -PipelinePermissionLevelParam = Literal["CAN_MANAGE", "IS_OWNER", "CAN_RUN", "CAN_VIEW"] | PipelinePermissionLevel + +PipelinePermissionLevelParam = ( + Literal["CAN_MANAGE", "IS_OWNER", "CAN_RUN", "CAN_VIEW"] | PipelinePermissionLevel +) diff --git a/python/databricks/bundles/pipelines/_models/pipelines_environment.py b/python/databricks/bundles/pipelines/_models/pipelines_environment.py index 14db0afc3dc..0493c5d4fa7 100644 --- a/python/databricks/bundles/pipelines/_models/pipelines_environment.py +++ b/python/databricks/bundles/pipelines/_models/pipelines_environment.py @@ -1,148 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PipelinesEnvironment: """ The environment entity used to preserve serverless environment side panel, jobs' environment for non-notebook task, and SDP's environment for classic and serverless pipelines. In this minimal environment spec, only pip dependencies are supported. """ + dependencies: VariableOrList[str] = field(default_factory=list) """ [Public Preview] List of pip dependencies, as supported by the version of pip in this environment. @@ -165,12 +40,11 @@ class PipelinesEnvironment: """ @classmethod - def from_dict(cls, value: 'PipelinesEnvironmentDict') -> 'Self': + def from_dict(cls, value: "PipelinesEnvironmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PipelinesEnvironmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PipelinesEnvironmentDict": + return _transform_to_json_value(self) # type:ignore class PipelinesEnvironmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/postgres_catalog_config.py b/python/databricks/bundles/pipelines/_models/postgres_catalog_config.py index ef3e7f7b64c..0bc642e8a5a 100644 --- a/python/databricks/bundles/pipelines/_models/postgres_catalog_config.py +++ b/python/databricks/bundles/pipelines/_models/postgres_catalog_config.py @@ -1,159 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.postgres_slot_config import ( + PostgresSlotConfig, + PostgresSlotConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PostgresCatalogConfig: """ PG-specific catalog-level configuration parameters """ + slot_config: VariableOrOptional[PostgresSlotConfig] = None """ [Public Preview] Optional. The Postgres slot configuration to use for logical replication """ @classmethod - def from_dict(cls, value: 'PostgresCatalogConfigDict') -> 'Self': + def from_dict(cls, value: "PostgresCatalogConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PostgresCatalogConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PostgresCatalogConfigDict": + return _transform_to_json_value(self) # type:ignore class PostgresCatalogConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/postgres_slot_config.py b/python/databricks/bundles/pipelines/_models/postgres_slot_config.py index a629f1905c5..d6eaeacbb08 100644 --- a/python/databricks/bundles/pipelines/_models/postgres_slot_config.py +++ b/python/databricks/bundles/pipelines/_models/postgres_slot_config.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PostgresSlotConfig: """ PostgresSlotConfig contains the configuration for a Postgres logical replication slot """ + publication_name: VariableOrOptional[str] = None """ [Public Preview] The name of the publication to use for the Postgres source @@ -153,12 +28,11 @@ class PostgresSlotConfig: """ @classmethod - def from_dict(cls, value: 'PostgresSlotConfigDict') -> 'Self': + def from_dict(cls, value: "PostgresSlotConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PostgresSlotConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PostgresSlotConfigDict": + return _transform_to_json_value(self) # type:ignore class PostgresSlotConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/protobuf_transformer_options.py b/python/databricks/bundles/pipelines/_models/protobuf_transformer_options.py index 3eb7538c530..cace688c4c9 100644 --- a/python/databricks/bundles/pipelines/_models/protobuf_transformer_options.py +++ b/python/databricks/bundles/pipelines/_models/protobuf_transformer_options.py @@ -1,142 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - +from databricks.bundles.core._variable import VariableOrOptional from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.pipelines._models.schema_registry_config import ( + SchemaRegistryConfig, + SchemaRegistryConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ProtobufTransformerOptions: """""" @@ -182,12 +61,11 @@ class ProtobufTransformerOptions: """ @classmethod - def from_dict(cls, value: 'ProtobufTransformerOptionsDict') -> 'Self': + def from_dict(cls, value: "ProtobufTransformerOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ProtobufTransformerOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ProtobufTransformerOptionsDict": + return _transform_to_json_value(self) # type:ignore class ProtobufTransformerOptionsDict(TypedDict, total=False): @@ -234,4 +112,6 @@ class ProtobufTransformerOptionsDict(TypedDict, total=False): """ -ProtobufTransformerOptionsParam = ProtobufTransformerOptionsDict | ProtobufTransformerOptions +ProtobufTransformerOptionsParam = ( + ProtobufTransformerOptionsDict | ProtobufTransformerOptions +) diff --git a/python/databricks/bundles/pipelines/_models/rabbitmq_options.py b/python/databricks/bundles/pipelines/_models/rabbitmq_options.py index f2c31512535..a08a4a0613a 100644 --- a/python/databricks/bundles/pipelines/_models/rabbitmq_options.py +++ b/python/databricks/bundles/pipelines/_models/rabbitmq_options.py @@ -1,142 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RabbitmqOptions: """ @@ -145,6 +19,7 @@ class RabbitmqOptions: are intentionally not exposed in the public API. The managed connector uses sensible defaults internally. These can be added later if user demand arises. """ + queue: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -153,12 +28,11 @@ class RabbitmqOptions: """ @classmethod - def from_dict(cls, value: 'RabbitmqOptionsDict') -> 'Self': + def from_dict(cls, value: "RabbitmqOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RabbitmqOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RabbitmqOptionsDict": + return _transform_to_json_value(self) # type:ignore class RabbitmqOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/reddit_ads_options.py b/python/databricks/bundles/pipelines/_models/reddit_ads_options.py index 77a3afb6b63..3e39958b026 100644 --- a/python/databricks/bundles/pipelines/_models/reddit_ads_options.py +++ b/python/databricks/bundles/pipelines/_models/reddit_ads_options.py @@ -1,150 +1,31 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import ( + RedditAdsOptionsRedditAdsCustomReportOptions, + RedditAdsOptionsRedditAdsCustomReportOptionsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RedditAdsOptions: """ :meta private: [EXPERIMENTAL] - + Reddit Ads specific options for ingestion """ - custom_report_options: VariableOrOptional[RedditAdsOptionsRedditAdsCustomReportOptions] = None + + custom_report_options: VariableOrOptional[ + RedditAdsOptionsRedditAdsCustomReportOptions + ] = None """ :meta private: [EXPERIMENTAL] @@ -172,18 +53,19 @@ class RedditAdsOptions: """ @classmethod - def from_dict(cls, value: 'RedditAdsOptionsDict') -> 'Self': + def from_dict(cls, value: "RedditAdsOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RedditAdsOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RedditAdsOptionsDict": + return _transform_to_json_value(self) # type:ignore class RedditAdsOptionsDict(TypedDict, total=False): """""" - custom_report_options: VariableOrOptional[RedditAdsOptionsRedditAdsCustomReportOptionsParam] + custom_report_options: VariableOrOptional[ + RedditAdsOptionsRedditAdsCustomReportOptionsParam + ] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/reddit_ads_options_reddit_ads_custom_report_options.py b/python/databricks/bundles/pipelines/_models/reddit_ads_options_reddit_ads_custom_report_options.py index 32c383eb5f6..78063a7dcaf 100644 --- a/python/databricks/bundles/pipelines/_models/reddit_ads_options_reddit_ads_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/reddit_ads_options_reddit_ads_custom_report_options.py @@ -1,150 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RedditAdsOptionsRedditAdsCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + User-defined custom report for the Reddit Ads connector. Applies only to the custom_report table — prebuilt tables ignore this. """ + breakdowns: VariableOrList[str] = field(default_factory=list) """ :meta private: [EXPERIMENTAL] @@ -163,12 +38,13 @@ class RedditAdsOptionsRedditAdsCustomReportOptions: """ @classmethod - def from_dict(cls, value: 'RedditAdsOptionsRedditAdsCustomReportOptionsDict') -> 'Self': + def from_dict( + cls, value: "RedditAdsOptionsRedditAdsCustomReportOptionsDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RedditAdsOptionsRedditAdsCustomReportOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RedditAdsOptionsRedditAdsCustomReportOptionsDict": + return _transform_to_json_value(self) # type:ignore class RedditAdsOptionsRedditAdsCustomReportOptionsDict(TypedDict, total=False): @@ -192,4 +68,7 @@ class RedditAdsOptionsRedditAdsCustomReportOptionsDict(TypedDict, total=False): """ -RedditAdsOptionsRedditAdsCustomReportOptionsParam = RedditAdsOptionsRedditAdsCustomReportOptionsDict | RedditAdsOptionsRedditAdsCustomReportOptions +RedditAdsOptionsRedditAdsCustomReportOptionsParam = ( + RedditAdsOptionsRedditAdsCustomReportOptionsDict + | RedditAdsOptionsRedditAdsCustomReportOptions +) diff --git a/python/databricks/bundles/pipelines/_models/report_spec.py b/python/databricks/bundles/pipelines/_models/report_spec.py index 90785f46cd8..c1ae97d014b 100644 --- a/python/databricks/bundles/pipelines/_models/report_spec.py +++ b/python/databricks/bundles/pipelines/_models/report_spec.py @@ -1,142 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.pipelines._models.table_specific_config import ( + TableSpecificConfig, + TableSpecificConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ReportSpec: """""" @@ -167,12 +45,11 @@ class ReportSpec: """ @classmethod - def from_dict(cls, value: 'ReportSpecDict') -> 'Self': + def from_dict(cls, value: "ReportSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ReportSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ReportSpecDict": + return _transform_to_json_value(self) # type:ignore class ReportSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/restart_window.py b/python/databricks/bundles/pipelines/_models/restart_window.py index ebd9b3a6672..9987bbe8ec7 100644 --- a/python/databricks/bundles/pipelines/_models/restart_window.py +++ b/python/databricks/bundles/pipelines/_models/restart_window.py @@ -1,147 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RestartWindow: """ :meta private: [EXPERIMENTAL] """ + start_hour: VariableOr[int] """ :meta private: [EXPERIMENTAL] @@ -167,12 +47,11 @@ class RestartWindow: """ @classmethod - def from_dict(cls, value: 'RestartWindowDict') -> 'Self': + def from_dict(cls, value: "RestartWindowDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RestartWindowDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RestartWindowDict": + return _transform_to_json_value(self) # type:ignore class RestartWindowDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/run_as.py b/python/databricks/bundles/pipelines/_models/run_as.py index 59512af8d53..50967e4e1e5 100644 --- a/python/databricks/bundles/pipelines/_models/run_as.py +++ b/python/databricks/bundles/pipelines/_models/run_as.py @@ -1,149 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RunAs: """ Write-only setting, available only in Create/Update calls. Specifies the user or service principal that the pipeline runs as. If not specified, the pipeline runs as the user who created the pipeline. - + Only `user_name` or `service_principal_name` can be specified. If both are specified, an error is thrown. """ + group_name: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -162,12 +37,11 @@ class RunAs: """ @classmethod - def from_dict(cls, value: 'RunAsDict') -> 'Self': + def from_dict(cls, value: "RunAsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RunAsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RunAsDict": + return _transform_to_json_value(self) # type:ignore class RunAsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/s3_storage_info.py b/python/databricks/bundles/pipelines/_models/s3_storage_info.py index be08638dc6c..071abd5b6f3 100644 --- a/python/databricks/bundles/pipelines/_models/s3_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/s3_storage_info.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class S3StorageInfo: """ A storage location in Amazon S3 """ + destination: VariableOr[str] """ S3 destination, e.g. `s3://my-bucket/some-prefix` Note that logs will be delivered using @@ -189,12 +64,11 @@ class S3StorageInfo: """ @classmethod - def from_dict(cls, value: 'S3StorageInfoDict') -> 'Self': + def from_dict(cls, value: "S3StorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'S3StorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "S3StorageInfoDict": + return _transform_to_json_value(self) # type:ignore class S3StorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/schema_registry_config.py b/python/databricks/bundles/pipelines/_models/schema_registry_config.py index ae8401fa98c..d2617560d7a 100644 --- a/python/databricks/bundles/pipelines/_models/schema_registry_config.py +++ b/python/databricks/bundles/pipelines/_models/schema_registry_config.py @@ -1,142 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.confluent_schema_registry_options import ( + ConfluentSchemaRegistryOptions, + ConfluentSchemaRegistryOptionsParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SchemaRegistryConfig: """""" @@ -166,12 +44,11 @@ class SchemaRegistryConfig: """ @classmethod - def from_dict(cls, value: 'SchemaRegistryConfigDict') -> 'Self': + def from_dict(cls, value: "SchemaRegistryConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SchemaRegistryConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SchemaRegistryConfigDict": + return _transform_to_json_value(self) # type:ignore class SchemaRegistryConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/schema_spec.py b/python/databricks/bundles/pipelines/_models/schema_spec.py index af5ac81b184..d40e976ddcd 100644 --- a/python/databricks/bundles/pipelines/_models/schema_spec.py +++ b/python/databricks/bundles/pipelines/_models/schema_spec.py @@ -1,142 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.pipelines._models.connector_options import ( + ConnectorOptions, + ConnectorOptionsParam, +) +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import ( + IngestionPipelineDefinitionFanoutOptions, + IngestionPipelineDefinitionFanoutOptionsParam, +) +from databricks.bundles.pipelines._models.table_specific_config import ( + TableSpecificConfig, + TableSpecificConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SchemaSpec: """""" @@ -184,12 +70,11 @@ class SchemaSpec: """ @classmethod - def from_dict(cls, value: 'SchemaSpecDict') -> 'Self': + def from_dict(cls, value: "SchemaSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SchemaSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SchemaSpecDict": + return _transform_to_json_value(self) # type:ignore class SchemaSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/sharepoint_options.py b/python/databricks/bundles/pipelines/_models/sharepoint_options.py index 1178fd2877a..4da2754ee21 100644 --- a/python/databricks/bundles/pipelines/_models/sharepoint_options.py +++ b/python/databricks/bundles/pipelines/_models/sharepoint_options.py @@ -1,147 +1,30 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.file_ingestion_options import ( + FileIngestionOptions, + FileIngestionOptionsParam, +) +from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import ( + SharepointOptionsSharepointEntityType, + SharepointOptionsSharepointEntityTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SharepointOptions: """ :meta private: [EXPERIMENTAL] """ + entity_type: VariableOrOptional[SharepointOptionsSharepointEntityType] = None """ :meta private: [EXPERIMENTAL] @@ -165,12 +48,11 @@ class SharepointOptions: """ @classmethod - def from_dict(cls, value: 'SharepointOptionsDict') -> 'Self': + def from_dict(cls, value: "SharepointOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SharepointOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SharepointOptionsDict": + return _transform_to_json_value(self) # type:ignore class SharepointOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/sharepoint_options_sharepoint_entity_type.py b/python/databricks/bundles/pipelines/_models/sharepoint_options_sharepoint_entity_type.py index 280b5e277b8..3b0f8bf6403 100644 --- a/python/databricks/bundles/pipelines/_models/sharepoint_options_sharepoint_entity_type.py +++ b/python/databricks/bundles/pipelines/_models/sharepoint_options_sharepoint_entity_type.py @@ -1,145 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class SharepointOptionsSharepointEntityType(Enum): """ :meta private: [EXPERIMENTAL] """ + FILE = "FILE" FILE_METADATA = "FILE_METADATA" PERMISSION = "PERMISSION" LIST = "LIST" -SharepointOptionsSharepointEntityTypeParam = Literal["FILE", "FILE_METADATA", "PERMISSION", "LIST"] | SharepointOptionsSharepointEntityType + +SharepointOptionsSharepointEntityTypeParam = ( + Literal["FILE", "FILE_METADATA", "PERMISSION", "LIST"] + | SharepointOptionsSharepointEntityType +) diff --git a/python/databricks/bundles/pipelines/_models/smartsheet_options.py b/python/databricks/bundles/pipelines/_models/smartsheet_options.py index e12f7bf4026..73691be256b 100644 --- a/python/databricks/bundles/pipelines/_models/smartsheet_options.py +++ b/python/databricks/bundles/pipelines/_models/smartsheet_options.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SmartsheetOptions: """ Smartsheet specific options for ingestion """ + enforce_schema: VariableOrOptional[bool] = None """ [Public Preview] (Optional) When true, maps each column to its Smartsheet-declared type (Text/Number/Date/ @@ -152,12 +27,11 @@ class SmartsheetOptions: """ @classmethod - def from_dict(cls, value: 'SmartsheetOptionsDict') -> 'Self': + def from_dict(cls, value: "SmartsheetOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SmartsheetOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SmartsheetOptionsDict": + return _transform_to_json_value(self) # type:ignore class SmartsheetOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/source_catalog_config.py b/python/databricks/bundles/pipelines/_models/source_catalog_config.py index dd90b20e46f..096c7e971d7 100644 --- a/python/databricks/bundles/pipelines/_models/source_catalog_config.py +++ b/python/databricks/bundles/pipelines/_models/source_catalog_config.py @@ -1,147 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.postgres_catalog_config import ( + PostgresCatalogConfig, + PostgresCatalogConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SourceCatalogConfig: """ SourceCatalogConfig contains catalog-level custom configuration parameters for each source """ + postgres: VariableOrOptional[PostgresCatalogConfig] = None """ [Public Preview] Postgres-specific catalog-level configuration parameters @@ -153,12 +32,11 @@ class SourceCatalogConfig: """ @classmethod - def from_dict(cls, value: 'SourceCatalogConfigDict') -> 'Self': + def from_dict(cls, value: "SourceCatalogConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SourceCatalogConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SourceCatalogConfigDict": + return _transform_to_json_value(self) # type:ignore class SourceCatalogConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/source_config.py b/python/databricks/bundles/pipelines/_models/source_config.py index 33351d05d0b..4695b153b40 100644 --- a/python/databricks/bundles/pipelines/_models/source_config.py +++ b/python/databricks/bundles/pipelines/_models/source_config.py @@ -1,142 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.api_source_connector_config import ( + ApiSourceConnectorConfig, + ApiSourceConnectorConfigParam, +) +from databricks.bundles.pipelines._models.google_ads_config import ( + GoogleAdsConfig, + GoogleAdsConfigParam, +) +from databricks.bundles.pipelines._models.source_catalog_config import ( + SourceCatalogConfig, + SourceCatalogConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SourceConfig: """""" @@ -161,12 +47,11 @@ class SourceConfig: """ @classmethod - def from_dict(cls, value: 'SourceConfigDict') -> 'Self': + def from_dict(cls, value: "SourceConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SourceConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SourceConfigDict": + return _transform_to_json_value(self) # type:ignore class SourceConfigDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/table_spec.py b/python/databricks/bundles/pipelines/_models/table_spec.py index 86c3c8445d1..dc75d872c71 100644 --- a/python/databricks/bundles/pipelines/_models/table_spec.py +++ b/python/databricks/bundles/pipelines/_models/table_spec.py @@ -1,142 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.pipelines._models.connector_options import ( + ConnectorOptions, + ConnectorOptionsParam, +) +from databricks.bundles.pipelines._models.table_specific_config import ( + TableSpecificConfig, + TableSpecificConfigParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TableSpec: """""" @@ -184,12 +66,11 @@ class TableSpec: """ @classmethod - def from_dict(cls, value: 'TableSpecDict') -> 'Self': + def from_dict(cls, value: "TableSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TableSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TableSpecDict": + return _transform_to_json_value(self) # type:ignore class TableSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/table_specific_config.py b/python/databricks/bundles/pipelines/_models/table_specific_config.py index fb29c052283..42773ae759e 100644 --- a/python/databricks/bundles/pipelines/_models/table_specific_config.py +++ b/python/databricks/bundles/pipelines/_models/table_specific_config.py @@ -1,142 +1,36 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import ( + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.pipelines._models.auto_full_refresh_policy import ( + AutoFullRefreshPolicy, + AutoFullRefreshPolicyParam, +) +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import ( + IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, + IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam, +) +from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import ( + IngestionPipelineDefinitionWorkdayReportParameters, + IngestionPipelineDefinitionWorkdayReportParametersParam, +) +from databricks.bundles.pipelines._models.table_specific_config_scd_type import ( + TableSpecificConfigScdType, + TableSpecificConfigScdTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TableSpecificConfig: """""" @@ -202,7 +96,9 @@ class TableSpecificConfig: [Public Preview] The primary key of the table used to apply changes. """ - query_based_connector_config: VariableOrOptional[IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig] = None + query_based_connector_config: VariableOrOptional[ + IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig + ] = None """ [Public Preview] Configurations that are only applicable for query-based ingestion connectors. """ @@ -249,7 +145,9 @@ class TableSpecificConfig: Note: table_properties in table specific configuration will override the table_properties of the pipeline definition. """ - workday_report_parameters: VariableOrOptional[IngestionPipelineDefinitionWorkdayReportParameters] = None + workday_report_parameters: VariableOrOptional[ + IngestionPipelineDefinitionWorkdayReportParameters + ] = None """ :meta private: [EXPERIMENTAL] @@ -257,12 +155,11 @@ class TableSpecificConfig: """ @classmethod - def from_dict(cls, value: 'TableSpecificConfigDict') -> 'Self': + def from_dict(cls, value: "TableSpecificConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TableSpecificConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TableSpecificConfigDict": + return _transform_to_json_value(self) # type:ignore class TableSpecificConfigDict(TypedDict, total=False): @@ -329,7 +226,9 @@ class TableSpecificConfigDict(TypedDict, total=False): [Public Preview] The primary key of the table used to apply changes. """ - query_based_connector_config: VariableOrOptional[IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam] + query_based_connector_config: VariableOrOptional[ + IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam + ] """ [Public Preview] Configurations that are only applicable for query-based ingestion connectors. """ @@ -376,7 +275,9 @@ class TableSpecificConfigDict(TypedDict, total=False): Note: table_properties in table specific configuration will override the table_properties of the pipeline definition. """ - workday_report_parameters: VariableOrOptional[IngestionPipelineDefinitionWorkdayReportParametersParam] + workday_report_parameters: VariableOrOptional[ + IngestionPipelineDefinitionWorkdayReportParametersParam + ] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/table_specific_config_scd_type.py b/python/databricks/bundles/pipelines/_models/table_specific_config_scd_type.py index fe3e10d7cbc..2d650e17111 100644 --- a/python/databricks/bundles/pipelines/_models/table_specific_config_scd_type.py +++ b/python/databricks/bundles/pipelines/_models/table_specific_config_scd_type.py @@ -1,144 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class TableSpecificConfigScdType(Enum): """ The SCD type to use to ingest the table. """ + SCD_TYPE_1 = "SCD_TYPE_1" SCD_TYPE_2 = "SCD_TYPE_2" APPEND_ONLY = "APPEND_ONLY" -TableSpecificConfigScdTypeParam = Literal["SCD_TYPE_1", "SCD_TYPE_2", "APPEND_ONLY"] | TableSpecificConfigScdType + +TableSpecificConfigScdTypeParam = ( + Literal["SCD_TYPE_1", "SCD_TYPE_2", "APPEND_ONLY"] | TableSpecificConfigScdType +) diff --git a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options.py b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options.py index 1a5330b3e0c..531f06cd434 100644 --- a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options.py +++ b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options.py @@ -1,148 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import ( + TikTokAdsOptionsTikTokAdsCustomReportOptions, + TikTokAdsOptionsTikTokAdsCustomReportOptionsParam, +) +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import ( + TikTokAdsOptionsTikTokDataLevel, + TikTokAdsOptionsTikTokDataLevelParam, +) +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import ( + TikTokAdsOptionsTikTokReportType, + TikTokAdsOptionsTikTokReportTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TikTokAdsOptions: """ TikTok Ads specific options for ingestion """ - custom_report_options: VariableOrOptional[TikTokAdsOptionsTikTokAdsCustomReportOptions] = None + + custom_report_options: VariableOrOptional[ + TikTokAdsOptionsTikTokAdsCustomReportOptions + ] = None """ :meta private: [EXPERIMENTAL] @@ -205,18 +94,19 @@ class TikTokAdsOptions: """ @classmethod - def from_dict(cls, value: 'TikTokAdsOptionsDict') -> 'Self': + def from_dict(cls, value: "TikTokAdsOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TikTokAdsOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TikTokAdsOptionsDict": + return _transform_to_json_value(self) # type:ignore class TikTokAdsOptionsDict(TypedDict, total=False): """""" - custom_report_options: VariableOrOptional[TikTokAdsOptionsTikTokAdsCustomReportOptionsParam] + custom_report_options: VariableOrOptional[ + TikTokAdsOptionsTikTokAdsCustomReportOptionsParam + ] """ :meta private: [EXPERIMENTAL] diff --git a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_ads_custom_report_options.py b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_ads_custom_report_options.py index 4586a8359bd..6c2820a3bc2 100644 --- a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_ads_custom_report_options.py +++ b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_ads_custom_report_options.py @@ -1,151 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import ( + TikTokAdsOptionsTikTokDataLevel, + TikTokAdsOptionsTikTokDataLevelParam, +) +from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import ( + TikTokAdsOptionsTikTokReportType, + TikTokAdsOptionsTikTokReportTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class TikTokAdsOptionsTikTokAdsCustomReportOptions: """ :meta private: [EXPERIMENTAL] - + User-defined custom report for the TikTok Ads connector. Groups the dimensions + metrics + report type + data level that define a TikTok Ads custom report request. """ + data_level: VariableOrOptional[TikTokAdsOptionsTikTokDataLevel] = None """ :meta private: [EXPERIMENTAL] @@ -188,12 +71,13 @@ class TikTokAdsOptionsTikTokAdsCustomReportOptions: """ @classmethod - def from_dict(cls, value: 'TikTokAdsOptionsTikTokAdsCustomReportOptionsDict') -> 'Self': + def from_dict( + cls, value: "TikTokAdsOptionsTikTokAdsCustomReportOptionsDict" + ) -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TikTokAdsOptionsTikTokAdsCustomReportOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TikTokAdsOptionsTikTokAdsCustomReportOptionsDict": + return _transform_to_json_value(self) # type:ignore class TikTokAdsOptionsTikTokAdsCustomReportOptionsDict(TypedDict, total=False): @@ -241,4 +125,7 @@ class TikTokAdsOptionsTikTokAdsCustomReportOptionsDict(TypedDict, total=False): """ -TikTokAdsOptionsTikTokAdsCustomReportOptionsParam = TikTokAdsOptionsTikTokAdsCustomReportOptionsDict | TikTokAdsOptionsTikTokAdsCustomReportOptions +TikTokAdsOptionsTikTokAdsCustomReportOptionsParam = ( + TikTokAdsOptionsTikTokAdsCustomReportOptionsDict + | TikTokAdsOptionsTikTokAdsCustomReportOptions +) diff --git a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_data_level.py b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_data_level.py index ddbe90f1ab9..7795cd821b6 100644 --- a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_data_level.py +++ b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_data_level.py @@ -1,145 +1,21 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class TikTokAdsOptionsTikTokDataLevel(Enum): """ Data level for TikTok Ads report aggregation. """ + AUCTION_ADVERTISER = "AUCTION_ADVERTISER" AUCTION_CAMPAIGN = "AUCTION_CAMPAIGN" AUCTION_ADGROUP = "AUCTION_ADGROUP" AUCTION_AD = "AUCTION_AD" -TikTokAdsOptionsTikTokDataLevelParam = Literal["AUCTION_ADVERTISER", "AUCTION_CAMPAIGN", "AUCTION_ADGROUP", "AUCTION_AD"] | TikTokAdsOptionsTikTokDataLevel + +TikTokAdsOptionsTikTokDataLevelParam = ( + Literal["AUCTION_ADVERTISER", "AUCTION_CAMPAIGN", "AUCTION_ADGROUP", "AUCTION_AD"] + | TikTokAdsOptionsTikTokDataLevel +) diff --git a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_report_type.py b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_report_type.py index 1bc844137e9..c97e2ee93a5 100644 --- a/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_report_type.py +++ b/python/databricks/bundles/pipelines/_models/tik_tok_ads_options_tik_tok_report_type.py @@ -1,142 +1,14 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam +from typing import Literal class TikTokAdsOptionsTikTokReportType(Enum): """ Report type for TikTok Ads API. """ + BASIC = "BASIC" AUDIENCE = "AUDIENCE" PLAYABLE_AD = "PLAYABLE_AD" @@ -144,4 +16,8 @@ class TikTokAdsOptionsTikTokReportType(Enum): BUSINESS_CENTER = "BUSINESS_CENTER" GMV_MAX = "GMV_MAX" -TikTokAdsOptionsTikTokReportTypeParam = Literal["BASIC", "AUDIENCE", "PLAYABLE_AD", "DSA", "BUSINESS_CENTER", "GMV_MAX"] | TikTokAdsOptionsTikTokReportType + +TikTokAdsOptionsTikTokReportTypeParam = ( + Literal["BASIC", "AUDIENCE", "PLAYABLE_AD", "DSA", "BUSINESS_CENTER", "GMV_MAX"] + | TikTokAdsOptionsTikTokReportType +) diff --git a/python/databricks/bundles/pipelines/_models/transformer.py b/python/databricks/bundles/pipelines/_models/transformer.py index 3c26cdc5ee9..b42fcef66f9 100644 --- a/python/databricks/bundles/pipelines/_models/transformer.py +++ b/python/databricks/bundles/pipelines/_models/transformer.py @@ -1,147 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.pipelines._models.avro_transformer_options import ( + AvroTransformerOptions, + AvroTransformerOptionsParam, +) +from databricks.bundles.pipelines._models.json_transformer_options import ( + JsonTransformerOptions, + JsonTransformerOptionsParam, +) +from databricks.bundles.pipelines._models.protobuf_transformer_options import ( + ProtobufTransformerOptions, + ProtobufTransformerOptionsParam, +) +from databricks.bundles.pipelines._models.transformer_format import ( + TransformerFormat, + TransformerFormatParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Transformer: """ Specifies how to transform binary data into structured data. """ + avro_options: VariableOrOptional[AvroTransformerOptions] = None """ :meta private: [EXPERIMENTAL] @@ -187,12 +78,11 @@ class Transformer: """ @classmethod - def from_dict(cls, value: 'TransformerDict') -> 'Self': + def from_dict(cls, value: "TransformerDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'TransformerDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "TransformerDict": + return _transform_to_json_value(self) # type:ignore class TransformerDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/transformer_format.py b/python/databricks/bundles/pipelines/_models/transformer_format.py index 0b49977dc3f..006855705ca 100644 --- a/python/databricks/bundles/pipelines/_models/transformer_format.py +++ b/python/databricks/bundles/pipelines/_models/transformer_format.py @@ -1,136 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam +from typing import Literal class TransformerFormat(Enum): @@ -139,4 +10,7 @@ class TransformerFormat(Enum): AVRO = "AVRO" PROTOBUF = "PROTOBUF" -TransformerFormatParam = Literal["STRING", "JSON", "AVRO", "PROTOBUF"] | TransformerFormat + +TransformerFormatParam = ( + Literal["STRING", "JSON", "AVRO", "PROTOBUF"] | TransformerFormat +) diff --git a/python/databricks/bundles/pipelines/_models/volumes_storage_info.py b/python/databricks/bundles/pipelines/_models/volumes_storage_info.py index 1e0f05dbcf2..1eb4fff7bc2 100644 --- a/python/databricks/bundles/pipelines/_models/volumes_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/volumes_storage_info.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class VolumesStorageInfo: """ A storage location back by UC Volumes. """ + destination: VariableOr[str] """ UC Volumes destination, e.g. `/Volumes/catalog/schema/vol1/init-scripts/setup-datadog.sh` @@ -149,12 +24,11 @@ class VolumesStorageInfo: """ @classmethod - def from_dict(cls, value: 'VolumesStorageInfoDict') -> 'Self': + def from_dict(cls, value: "VolumesStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'VolumesStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "VolumesStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class VolumesStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/workspace_storage_info.py b/python/databricks/bundles/pipelines/_models/workspace_storage_info.py index f222fd1f7ea..49aef9256c8 100644 --- a/python/databricks/bundles/pipelines/_models/workspace_storage_info.py +++ b/python/databricks/bundles/pipelines/_models/workspace_storage_info.py @@ -1,159 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.zendesk_support_options import ZendeskSupportOptions, ZendeskSupportOptionsDict, ZendeskSupportOptionsParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class WorkspaceStorageInfo: """ A storage location in Workspace Filesystem (WSFS) """ + destination: VariableOr[str] """ wsfs destination, e.g. `workspace:/cluster-init-scripts/setup-datadog.sh` """ @classmethod - def from_dict(cls, value: 'WorkspaceStorageInfoDict') -> 'Self': + def from_dict(cls, value: "WorkspaceStorageInfoDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'WorkspaceStorageInfoDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "WorkspaceStorageInfoDict": + return _transform_to_json_value(self) # type:ignore class WorkspaceStorageInfoDict(TypedDict, total=False): diff --git a/python/databricks/bundles/pipelines/_models/zendesk_support_options.py b/python/databricks/bundles/pipelines/_models/zendesk_support_options.py index 38b25e3c76f..23a03e8eab8 100644 --- a/python/databricks/bundles/pipelines/_models/zendesk_support_options.py +++ b/python/databricks/bundles/pipelines/_models/zendesk_support_options.py @@ -1,147 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.pipelines._models.adlsgen2_info import Adlsgen2Info, Adlsgen2InfoDict, Adlsgen2InfoParam -from databricks.bundles.pipelines._models.aws_attributes import AwsAttributes, AwsAttributesDict, AwsAttributesParam -from databricks.bundles.pipelines._models.azure_attributes import AzureAttributes, AzureAttributesDict, AzureAttributesParam -from databricks.bundles.pipelines._models.cluster_log_conf import ClusterLogConf, ClusterLogConfDict, ClusterLogConfParam -from databricks.bundles.pipelines._models.dbfs_storage_info import DbfsStorageInfo, DbfsStorageInfoDict, DbfsStorageInfoParam -from databricks.bundles.pipelines._models.gcp_attributes import GcpAttributes, GcpAttributesDict, GcpAttributesParam -from databricks.bundles.pipelines._models.gcs_storage_info import GcsStorageInfo, GcsStorageInfoDict, GcsStorageInfoParam -from databricks.bundles.pipelines._models.init_script_info import InitScriptInfo, InitScriptInfoDict, InitScriptInfoParam -from databricks.bundles.pipelines._models.local_file_info import LocalFileInfo, LocalFileInfoDict, LocalFileInfoParam -from databricks.bundles.pipelines._models.log_analytics_info import LogAnalyticsInfo, LogAnalyticsInfoDict, LogAnalyticsInfoParam -from databricks.bundles.pipelines._models.maven_library import MavenLibrary, MavenLibraryDict, MavenLibraryParam -from databricks.bundles.pipelines._models.s3_storage_info import S3StorageInfo, S3StorageInfoDict, S3StorageInfoParam -from databricks.bundles.pipelines._models.volumes_storage_info import VolumesStorageInfo, VolumesStorageInfoDict, VolumesStorageInfoParam -from databricks.bundles.pipelines._models.workspace_storage_info import WorkspaceStorageInfo, WorkspaceStorageInfoDict, WorkspaceStorageInfoParam -from databricks.bundles.pipelines._models.api_source_connector_config import ApiSourceConnectorConfig, ApiSourceConnectorConfigDict, ApiSourceConnectorConfigParam -from databricks.bundles.pipelines._models.api_source_connector_options import ApiSourceConnectorOptions, ApiSourceConnectorOptionsDict, ApiSourceConnectorOptionsParam -from databricks.bundles.pipelines._models.auto_full_refresh_policy import AutoFullRefreshPolicy, AutoFullRefreshPolicyDict, AutoFullRefreshPolicyParam -from databricks.bundles.pipelines._models.avro_transformer_options import AvroTransformerOptions, AvroTransformerOptionsDict, AvroTransformerOptionsParam -from databricks.bundles.pipelines._models.confluence_connector_options import ConfluenceConnectorOptions, ConfluenceConnectorOptionsDict, ConfluenceConnectorOptionsParam -from databricks.bundles.pipelines._models.confluent_schema_registry_options import ConfluentSchemaRegistryOptions, ConfluentSchemaRegistryOptionsDict, ConfluentSchemaRegistryOptionsParam -from databricks.bundles.pipelines._models.connection_parameters import ConnectionParameters, ConnectionParametersDict, ConnectionParametersParam -from databricks.bundles.pipelines._models.connector_options import ConnectorOptions, ConnectorOptionsDict, ConnectorOptionsParam -from databricks.bundles.pipelines._models.data_staging_options import DataStagingOptions, DataStagingOptionsDict, DataStagingOptionsParam -from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec, EventLogSpecDict, EventLogSpecParam -from databricks.bundles.pipelines._models.file_filter import FileFilter, FileFilterDict, FileFilterParam -from databricks.bundles.pipelines._models.file_ingestion_options import FileIngestionOptions, FileIngestionOptionsDict, FileIngestionOptionsParam -from databricks.bundles.pipelines._models.file_library import FileLibrary, FileLibraryDict, FileLibraryParam -from databricks.bundles.pipelines._models.filters import Filters, FiltersDict, FiltersParam -from databricks.bundles.pipelines._models.google_ads_config import GoogleAdsConfig, GoogleAdsConfigDict, GoogleAdsConfigParam -from databricks.bundles.pipelines._models.google_ads_custom_report_options import GoogleAdsCustomReportOptions, GoogleAdsCustomReportOptionsDict, GoogleAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.google_ads_options import GoogleAdsOptions, GoogleAdsOptionsDict, GoogleAdsOptionsParam -from databricks.bundles.pipelines._models.google_drive_options import GoogleDriveOptions, GoogleDriveOptionsDict, GoogleDriveOptionsParam -from databricks.bundles.pipelines._models.ingestion_config import IngestionConfig, IngestionConfigDict, IngestionConfigParam -from databricks.bundles.pipelines._models.ingestion_gateway_pipeline_definition import IngestionGatewayPipelineDefinition, IngestionGatewayPipelineDefinitionDict, IngestionGatewayPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition, IngestionPipelineDefinitionDict, IngestionPipelineDefinitionParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_fanout_options import IngestionPipelineDefinitionFanoutOptions, IngestionPipelineDefinitionFanoutOptionsDict, IngestionPipelineDefinitionFanoutOptionsParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_table_specific_config_query_based_connector_config import IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfig, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigDict, IngestionPipelineDefinitionTableSpecificConfigQueryBasedConnectorConfigParam -from databricks.bundles.pipelines._models.ingestion_pipeline_definition_workday_report_parameters import IngestionPipelineDefinitionWorkdayReportParameters, IngestionPipelineDefinitionWorkdayReportParametersDict, IngestionPipelineDefinitionWorkdayReportParametersParam -from databricks.bundles.pipelines._models.jira_connector_options import JiraConnectorOptions, JiraConnectorOptionsDict, JiraConnectorOptionsParam -from databricks.bundles.pipelines._models.json_transformer_options import JsonTransformerOptions, JsonTransformerOptionsDict, JsonTransformerOptionsParam -from databricks.bundles.pipelines._models.kafka_options import KafkaOptions, KafkaOptionsDict, KafkaOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options import LinkedInAdsOptions, LinkedInAdsOptionsDict, LinkedInAdsOptionsParam -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options import LinkedInAdsOptionsLinkedInAdsCustomReportOptions, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsDict, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.marketo_options import MarketoOptions, MarketoOptionsDict, MarketoOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options import MetaMarketingOptions, MetaMarketingOptionsDict, MetaMarketingOptionsParam -from databricks.bundles.pipelines._models.meta_marketing_options_meta_marketing_custom_report_options import MetaMarketingOptionsMetaMarketingCustomReportOptions, MetaMarketingOptionsMetaMarketingCustomReportOptionsDict, MetaMarketingOptionsMetaMarketingCustomReportOptionsParam -from databricks.bundles.pipelines._models.notebook_library import NotebookLibrary, NotebookLibraryDict, NotebookLibraryParam -from databricks.bundles.pipelines._models.notifications import Notifications, NotificationsDict, NotificationsParam -from databricks.bundles.pipelines._models.operation_time_window import OperationTimeWindow, OperationTimeWindowDict, OperationTimeWindowParam -from databricks.bundles.pipelines._models.outlook_options import OutlookOptions, OutlookOptionsDict, OutlookOptionsParam -from databricks.bundles.pipelines._models.path_pattern import PathPattern, PathPatternDict, PathPatternParam -from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster, PipelineClusterDict, PipelineClusterParam -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale import PipelineClusterAutoscale, PipelineClusterAutoscaleDict, PipelineClusterAutoscaleParam -from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary, PipelineLibraryDict, PipelineLibraryParam -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment, PipelinesEnvironmentDict, PipelinesEnvironmentParam -from databricks.bundles.pipelines._models.postgres_catalog_config import PostgresCatalogConfig, PostgresCatalogConfigDict, PostgresCatalogConfigParam -from databricks.bundles.pipelines._models.postgres_slot_config import PostgresSlotConfig, PostgresSlotConfigDict, PostgresSlotConfigParam -from databricks.bundles.pipelines._models.protobuf_transformer_options import ProtobufTransformerOptions, ProtobufTransformerOptionsDict, ProtobufTransformerOptionsParam -from databricks.bundles.pipelines._models.rabbitmq_options import RabbitmqOptions, RabbitmqOptionsDict, RabbitmqOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options import RedditAdsOptions, RedditAdsOptionsDict, RedditAdsOptionsParam -from databricks.bundles.pipelines._models.reddit_ads_options_reddit_ads_custom_report_options import RedditAdsOptionsRedditAdsCustomReportOptions, RedditAdsOptionsRedditAdsCustomReportOptionsDict, RedditAdsOptionsRedditAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.report_spec import ReportSpec, ReportSpecDict, ReportSpecParam -from databricks.bundles.pipelines._models.restart_window import RestartWindow, RestartWindowDict, RestartWindowParam -from databricks.bundles.pipelines._models.run_as import RunAs, RunAsDict, RunAsParam -from databricks.bundles.pipelines._models.schema_registry_config import SchemaRegistryConfig, SchemaRegistryConfigDict, SchemaRegistryConfigParam -from databricks.bundles.pipelines._models.schema_spec import SchemaSpec, SchemaSpecDict, SchemaSpecParam -from databricks.bundles.pipelines._models.sharepoint_options import SharepointOptions, SharepointOptionsDict, SharepointOptionsParam -from databricks.bundles.pipelines._models.smartsheet_options import SmartsheetOptions, SmartsheetOptionsDict, SmartsheetOptionsParam -from databricks.bundles.pipelines._models.source_catalog_config import SourceCatalogConfig, SourceCatalogConfigDict, SourceCatalogConfigParam -from databricks.bundles.pipelines._models.source_config import SourceConfig, SourceConfigDict, SourceConfigParam -from databricks.bundles.pipelines._models.table_spec import TableSpec, TableSpecDict, TableSpecParam -from databricks.bundles.pipelines._models.table_specific_config import TableSpecificConfig, TableSpecificConfigDict, TableSpecificConfigParam -from databricks.bundles.pipelines._models.tik_tok_ads_options import TikTokAdsOptions, TikTokAdsOptionsDict, TikTokAdsOptionsParam -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_ads_custom_report_options import TikTokAdsOptionsTikTokAdsCustomReportOptions, TikTokAdsOptionsTikTokAdsCustomReportOptionsDict, TikTokAdsOptionsTikTokAdsCustomReportOptionsParam -from databricks.bundles.pipelines._models.transformer import Transformer, TransformerDict, TransformerParam -from databricks.bundles.pipelines._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.pipelines._models.pipeline import Pipeline, PipelineDict, PipelineParam -from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission, PipelinePermissionDict, PipelinePermissionParam -from databricks.bundles.pipelines._models.aws_availability import AwsAvailability, AwsAvailabilityParam - -from databricks.bundles.pipelines._models.azure_availability import AzureAvailability, AzureAvailabilityParam - -from databricks.bundles.pipelines._models.confidential_compute_type import ConfidentialComputeType, ConfidentialComputeTypeParam - -from databricks.bundles.pipelines._models.ebs_volume_type import EbsVolumeType, EbsVolumeTypeParam - -from databricks.bundles.pipelines._models.gcp_availability import GcpAvailability, GcpAvailabilityParam - -from databricks.bundles.pipelines._models.connector_type import ConnectorType, ConnectorTypeParam - -from databricks.bundles.pipelines._models.day_of_week import DayOfWeek, DayOfWeekParam - -from databricks.bundles.pipelines._models.file_ingestion_options_file_format import FileIngestionOptionsFileFormat, FileIngestionOptionsFileFormatParam - -from databricks.bundles.pipelines._models.file_ingestion_options_schema_evolution_mode import FileIngestionOptionsSchemaEvolutionMode, FileIngestionOptionsSchemaEvolutionModeParam - -from databricks.bundles.pipelines._models.google_drive_options_google_drive_entity_type import GoogleDriveOptionsGoogleDriveEntityType, GoogleDriveOptionsGoogleDriveEntityTypeParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_entity_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsEntityGranularityParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_finder import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinder, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsFinderParam - -from databricks.bundles.pipelines._models.linked_in_ads_options_linked_in_ads_custom_report_options_linked_in_ads_time_granularity import LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularity, LinkedInAdsOptionsLinkedInAdsCustomReportOptionsLinkedInAdsTimeGranularityParam - -from databricks.bundles.pipelines._models.outlook_attachment_mode import OutlookAttachmentMode, OutlookAttachmentModeParam - -from databricks.bundles.pipelines._models.outlook_body_format import OutlookBodyFormat, OutlookBodyFormatParam - -from databricks.bundles.pipelines._models.parse_mode import ParseMode, ParseModeParam - -from databricks.bundles.pipelines._models.pipeline_cluster_autoscale_mode import PipelineClusterAutoscaleMode, PipelineClusterAutoscaleModeParam - -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel, PipelinePermissionLevelParam - -from databricks.bundles.pipelines._models.sharepoint_options_sharepoint_entity_type import SharepointOptionsSharepointEntityType, SharepointOptionsSharepointEntityTypeParam - -from databricks.bundles.pipelines._models.table_specific_config_scd_type import TableSpecificConfigScdType, TableSpecificConfigScdTypeParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_data_level import TikTokAdsOptionsTikTokDataLevel, TikTokAdsOptionsTikTokDataLevelParam - -from databricks.bundles.pipelines._models.tik_tok_ads_options_tik_tok_report_type import TikTokAdsOptionsTikTokReportType, TikTokAdsOptionsTikTokReportTypeParam - -from databricks.bundles.pipelines._models.transformer_format import TransformerFormat, TransformerFormatParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class ZendeskSupportOptions: """ Zendesk Support specific options for ingestion """ + start_date: VariableOrOptional[str] = None """ [Public Preview] (Optional) Start date in YYYY-MM-DD format for the initial sync. @@ -149,12 +24,11 @@ class ZendeskSupportOptions: """ @classmethod - def from_dict(cls, value: 'ZendeskSupportOptionsDict') -> 'Self': + def from_dict(cls, value: "ZendeskSupportOptionsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ZendeskSupportOptionsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ZendeskSupportOptionsDict": + return _transform_to_json_value(self) # type:ignore class ZendeskSupportOptionsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/__init__.py b/python/databricks/bundles/quality_monitors/__init__.py index 5c4e3fed940..ad0902bfcf6 100644 --- a/python/databricks/bundles/quality_monitors/__init__.py +++ b/python/databricks/bundles/quality_monitors/__init__.py @@ -1,58 +1,104 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'MonitorCronSchedule', - 'MonitorCronScheduleDict', - 'MonitorCronScheduleParam', - 'MonitorCronSchedulePauseStatus', - 'MonitorCronSchedulePauseStatusParam', - 'MonitorDataClassificationConfig', - 'MonitorDataClassificationConfigDict', - 'MonitorDataClassificationConfigParam', - 'MonitorDestination', - 'MonitorDestinationDict', - 'MonitorDestinationParam', - 'MonitorInferenceLog', - 'MonitorInferenceLogDict', - 'MonitorInferenceLogParam', - 'MonitorInferenceLogProblemType', - 'MonitorInferenceLogProblemTypeParam', - 'MonitorMetric', - 'MonitorMetricDict', - 'MonitorMetricParam', - 'MonitorMetricType', - 'MonitorMetricTypeParam', - 'MonitorNotifications', - 'MonitorNotificationsDict', - 'MonitorNotificationsParam', - 'MonitorSnapshot', - 'MonitorSnapshotDict', - 'MonitorSnapshotParam', - 'MonitorTimeSeries', - 'MonitorTimeSeriesDict', - 'MonitorTimeSeriesParam', - 'QualityMonitor', - 'QualityMonitorDict', - 'QualityMonitorParam', + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "MonitorCronSchedule", + "MonitorCronScheduleDict", + "MonitorCronScheduleParam", + "MonitorCronSchedulePauseStatus", + "MonitorCronSchedulePauseStatusParam", + "MonitorDataClassificationConfig", + "MonitorDataClassificationConfigDict", + "MonitorDataClassificationConfigParam", + "MonitorDestination", + "MonitorDestinationDict", + "MonitorDestinationParam", + "MonitorInferenceLog", + "MonitorInferenceLogDict", + "MonitorInferenceLogParam", + "MonitorInferenceLogProblemType", + "MonitorInferenceLogProblemTypeParam", + "MonitorMetric", + "MonitorMetricDict", + "MonitorMetricParam", + "MonitorMetricType", + "MonitorMetricTypeParam", + "MonitorNotifications", + "MonitorNotificationsDict", + "MonitorNotificationsParam", + "MonitorSnapshot", + "MonitorSnapshotDict", + "MonitorSnapshotParam", + "MonitorTimeSeries", + "MonitorTimeSeriesDict", + "MonitorTimeSeriesParam", + "QualityMonitor", + "QualityMonitorDict", + "QualityMonitorParam", ] -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.quality_monitors._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import ( + MonitorCronSchedule, + MonitorCronScheduleDict, + MonitorCronScheduleParam, +) +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import ( + MonitorCronSchedulePauseStatus, + MonitorCronSchedulePauseStatusParam, +) +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import ( + MonitorDataClassificationConfig, + MonitorDataClassificationConfigDict, + MonitorDataClassificationConfigParam, +) +from databricks.bundles.quality_monitors._models.monitor_destination import ( + MonitorDestination, + MonitorDestinationDict, + MonitorDestinationParam, +) +from databricks.bundles.quality_monitors._models.monitor_inference_log import ( + MonitorInferenceLog, + MonitorInferenceLogDict, + MonitorInferenceLogParam, +) +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import ( + MonitorInferenceLogProblemType, + MonitorInferenceLogProblemTypeParam, +) +from databricks.bundles.quality_monitors._models.monitor_metric import ( + MonitorMetric, + MonitorMetricDict, + MonitorMetricParam, +) +from databricks.bundles.quality_monitors._models.monitor_metric_type import ( + MonitorMetricType, + MonitorMetricTypeParam, +) +from databricks.bundles.quality_monitors._models.monitor_notifications import ( + MonitorNotifications, + MonitorNotificationsDict, + MonitorNotificationsParam, +) +from databricks.bundles.quality_monitors._models.monitor_snapshot import ( + MonitorSnapshot, + MonitorSnapshotDict, + MonitorSnapshotParam, +) +from databricks.bundles.quality_monitors._models.monitor_time_series import ( + MonitorTimeSeries, + MonitorTimeSeriesDict, + MonitorTimeSeriesParam, +) +from databricks.bundles.quality_monitors._models.quality_monitor import ( + QualityMonitor, + QualityMonitorDict, + QualityMonitorParam, +) diff --git a/python/databricks/bundles/quality_monitors/_models/lifecycle.py b/python/databricks/bundles/quality_monitors/_models/lifecycle.py index 0de2fca97eb..697776a198e 100644 --- a/python/databricks/bundles/quality_monitors/_models/lifecycle.py +++ b/python/databricks/bundles/quality_monitors/_models/lifecycle.py @@ -1,33 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -38,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule.py b/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule.py index ff2c7d94a65..02a8de80077 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule.py @@ -1,33 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import ( + MonitorCronSchedulePauseStatus, + MonitorCronSchedulePauseStatusParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MonitorCronSchedule: """""" @@ -48,12 +35,11 @@ class MonitorCronSchedule: """ @classmethod - def from_dict(cls, value: 'MonitorCronScheduleDict') -> 'Self': + def from_dict(cls, value: "MonitorCronScheduleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MonitorCronScheduleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MonitorCronScheduleDict": + return _transform_to_json_value(self) # type:ignore class MonitorCronScheduleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule_pause_status.py b/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule_pause_status.py index 6994b0fca8d..69b5dd3666b 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule_pause_status.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_cron_schedule_pause_status.py @@ -1,27 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam +from typing import Literal class MonitorCronSchedulePauseStatus(Enum): @@ -29,8 +9,12 @@ class MonitorCronSchedulePauseStatus(Enum): Source link: https://src.dev.databricks.com/databricks/universe/-/blob/elastic-spark-common/api/messages/schedule.proto Monitoring workflow schedule pause status. """ + UNSPECIFIED = "UNSPECIFIED" UNPAUSED = "UNPAUSED" PAUSED = "PAUSED" -MonitorCronSchedulePauseStatusParam = Literal["UNSPECIFIED", "UNPAUSED", "PAUSED"] | MonitorCronSchedulePauseStatus + +MonitorCronSchedulePauseStatusParam = ( + Literal["UNSPECIFIED", "UNPAUSED", "PAUSED"] | MonitorCronSchedulePauseStatus +) diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_data_classification_config.py b/python/databricks/bundles/quality_monitors/_models/monitor_data_classification_config.py index be3a2f2d29a..cda4e643683 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_data_classification_config.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_data_classification_config.py @@ -1,40 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MonitorDataClassificationConfig: """ :meta private: [EXPERIMENTAL] - + Data classification related configuration. """ + enabled: VariableOrOptional[bool] = None """ :meta private: [EXPERIMENTAL] @@ -43,12 +27,11 @@ class MonitorDataClassificationConfig: """ @classmethod - def from_dict(cls, value: 'MonitorDataClassificationConfigDict') -> 'Self': + def from_dict(cls, value: "MonitorDataClassificationConfigDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MonitorDataClassificationConfigDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MonitorDataClassificationConfigDict": + return _transform_to_json_value(self) # type:ignore class MonitorDataClassificationConfigDict(TypedDict, total=False): @@ -62,4 +45,6 @@ class MonitorDataClassificationConfigDict(TypedDict, total=False): """ -MonitorDataClassificationConfigParam = MonitorDataClassificationConfigDict | MonitorDataClassificationConfig +MonitorDataClassificationConfigParam = ( + MonitorDataClassificationConfigDict | MonitorDataClassificationConfig +) diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_destination.py b/python/databricks/bundles/quality_monitors/_models/monitor_destination.py index ca6decc05f7..2f4f5b1d37f 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_destination.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_destination.py @@ -1,33 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.core._variable import VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MonitorDestination: """""" @@ -38,12 +21,11 @@ class MonitorDestination: """ @classmethod - def from_dict(cls, value: 'MonitorDestinationDict') -> 'Self': + def from_dict(cls, value: "MonitorDestinationDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MonitorDestinationDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MonitorDestinationDict": + return _transform_to_json_value(self) # type:ignore class MonitorDestinationDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_inference_log.py b/python/databricks/bundles/quality_monitors/_models/monitor_inference_log.py index 4804b87def3..24b6ea70e87 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_inference_log.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_inference_log.py @@ -1,33 +1,24 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import ( + MonitorInferenceLogProblemType, + MonitorInferenceLogProblemTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MonitorInferenceLog: """""" @@ -68,12 +59,11 @@ class MonitorInferenceLog: """ @classmethod - def from_dict(cls, value: 'MonitorInferenceLogDict') -> 'Self': + def from_dict(cls, value: "MonitorInferenceLogDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MonitorInferenceLogDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MonitorInferenceLogDict": + return _transform_to_json_value(self) # type:ignore class MonitorInferenceLogDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_inference_log_problem_type.py b/python/databricks/bundles/quality_monitors/_models/monitor_inference_log_problem_type.py index de722188952..42d8569619b 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_inference_log_problem_type.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_inference_log_problem_type.py @@ -1,31 +1,15 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam +from typing import Literal class MonitorInferenceLogProblemType(Enum): PROBLEM_TYPE_CLASSIFICATION = "PROBLEM_TYPE_CLASSIFICATION" PROBLEM_TYPE_REGRESSION = "PROBLEM_TYPE_REGRESSION" -MonitorInferenceLogProblemTypeParam = Literal["PROBLEM_TYPE_CLASSIFICATION", "PROBLEM_TYPE_REGRESSION"] | MonitorInferenceLogProblemType + +MonitorInferenceLogProblemTypeParam = ( + Literal["PROBLEM_TYPE_CLASSIFICATION", "PROBLEM_TYPE_REGRESSION"] + | MonitorInferenceLogProblemType +) diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_metric.py b/python/databricks/bundles/quality_monitors/_models/monitor_metric.py index a22566cc818..cb7f876dae8 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_metric.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_metric.py @@ -1,38 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrList +from databricks.bundles.quality_monitors._models.monitor_metric_type import ( + MonitorMetricType, + MonitorMetricTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MonitorMetric: """ Custom metric definition. """ + definition: VariableOr[str] """ Jinja template for a SQL expression that specifies how to compute the metric. See [create metric definition](https://docs.databricks.com/en/lakehouse-monitoring/custom-metrics.html#create-definition). @@ -66,12 +54,11 @@ class MonitorMetric: """ @classmethod - def from_dict(cls, value: 'MonitorMetricDict') -> 'Self': + def from_dict(cls, value: "MonitorMetricDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MonitorMetricDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MonitorMetricDict": + return _transform_to_json_value(self) # type:ignore class MonitorMetricDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_metric_type.py b/python/databricks/bundles/quality_monitors/_models/monitor_metric_type.py index abe31acfa9a..4c70c089a0f 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_metric_type.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_metric_type.py @@ -1,27 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam +from typing import Literal class MonitorMetricType(Enum): @@ -34,8 +14,17 @@ class MonitorMetricType(Enum): - CUSTOM_METRIC_TYPE_DERIVED: depend on previously computed aggregate metrics - CUSTOM_METRIC_TYPE_DRIFT: depend on previously computed aggregate or derived metrics """ + CUSTOM_METRIC_TYPE_AGGREGATE = "CUSTOM_METRIC_TYPE_AGGREGATE" CUSTOM_METRIC_TYPE_DERIVED = "CUSTOM_METRIC_TYPE_DERIVED" CUSTOM_METRIC_TYPE_DRIFT = "CUSTOM_METRIC_TYPE_DRIFT" -MonitorMetricTypeParam = Literal["CUSTOM_METRIC_TYPE_AGGREGATE", "CUSTOM_METRIC_TYPE_DERIVED", "CUSTOM_METRIC_TYPE_DRIFT"] | MonitorMetricType + +MonitorMetricTypeParam = ( + Literal[ + "CUSTOM_METRIC_TYPE_AGGREGATE", + "CUSTOM_METRIC_TYPE_DERIVED", + "CUSTOM_METRIC_TYPE_DRIFT", + ] + | MonitorMetricType +) diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_notifications.py b/python/databricks/bundles/quality_monitors/_models/monitor_notifications.py index 3801a91aafb..007b3ae64e4 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_notifications.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_notifications.py @@ -1,33 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.quality_monitors._models.monitor_destination import ( + MonitorDestination, + MonitorDestinationParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MonitorNotifications: """""" @@ -45,12 +32,11 @@ class MonitorNotifications: """ @classmethod - def from_dict(cls, value: 'MonitorNotificationsDict') -> 'Self': + def from_dict(cls, value: "MonitorNotificationsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MonitorNotificationsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MonitorNotificationsDict": + return _transform_to_json_value(self) # type:ignore class MonitorNotificationsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_snapshot.py b/python/databricks/bundles/quality_monitors/_models/monitor_snapshot.py index e07bf168f4c..4cc39dcb9d0 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_snapshot.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_snapshot.py @@ -1,45 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MonitorSnapshot: """ Snapshot analysis configuration """ + @classmethod - def from_dict(cls, value: 'MonitorSnapshotDict') -> 'Self': + def from_dict(cls, value: "MonitorSnapshotDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MonitorSnapshotDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MonitorSnapshotDict": + return _transform_to_json_value(self) # type:ignore class MonitorSnapshotDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/monitor_time_series.py b/python/databricks/bundles/quality_monitors/_models/monitor_time_series.py index f5bc7302386..29f3b950a0e 100644 --- a/python/databricks/bundles/quality_monitors/_models/monitor_time_series.py +++ b/python/databricks/bundles/quality_monitors/_models/monitor_time_series.py @@ -1,38 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor, QualityMonitorDict, QualityMonitorParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrList if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class MonitorTimeSeries: """ Time series analysis configuration. """ + timestamp_col: VariableOr[str] """ Column for the timestamp. @@ -44,12 +28,11 @@ class MonitorTimeSeries: """ @classmethod - def from_dict(cls, value: 'MonitorTimeSeriesDict') -> 'Self': + def from_dict(cls, value: "MonitorTimeSeriesDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'MonitorTimeSeriesDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "MonitorTimeSeriesDict": + return _transform_to_json_value(self) # type:ignore class MonitorTimeSeriesDict(TypedDict, total=False): diff --git a/python/databricks/bundles/quality_monitors/_models/quality_monitor.py b/python/databricks/bundles/quality_monitors/_models/quality_monitor.py index 7bd52cea946..d2c22518503 100644 --- a/python/databricks/bundles/quality_monitors/_models/quality_monitor.py +++ b/python/databricks/bundles/quality_monitors/_models/quality_monitor.py @@ -1,33 +1,53 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule, MonitorCronScheduleDict, MonitorCronScheduleParam -from databricks.bundles.quality_monitors._models.monitor_data_classification_config import MonitorDataClassificationConfig, MonitorDataClassificationConfigDict, MonitorDataClassificationConfigParam -from databricks.bundles.quality_monitors._models.monitor_destination import MonitorDestination, MonitorDestinationDict, MonitorDestinationParam -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog, MonitorInferenceLogDict, MonitorInferenceLogParam -from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric, MonitorMetricDict, MonitorMetricParam -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications, MonitorNotificationsDict, MonitorNotificationsParam -from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot, MonitorSnapshotDict, MonitorSnapshotParam -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries, MonitorTimeSeriesDict, MonitorTimeSeriesParam -from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.quality_monitors._models.monitor_cron_schedule_pause_status import MonitorCronSchedulePauseStatus, MonitorCronSchedulePauseStatusParam - -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType, MonitorInferenceLogProblemTypeParam - -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType, MonitorMetricTypeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.quality_monitors._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import ( + MonitorCronSchedule, + MonitorCronScheduleParam, +) +from databricks.bundles.quality_monitors._models.monitor_data_classification_config import ( + MonitorDataClassificationConfig, + MonitorDataClassificationConfigParam, +) +from databricks.bundles.quality_monitors._models.monitor_inference_log import ( + MonitorInferenceLog, + MonitorInferenceLogParam, +) +from databricks.bundles.quality_monitors._models.monitor_metric import ( + MonitorMetric, + MonitorMetricParam, +) +from databricks.bundles.quality_monitors._models.monitor_notifications import ( + MonitorNotifications, + MonitorNotificationsParam, +) +from databricks.bundles.quality_monitors._models.monitor_snapshot import ( + MonitorSnapshot, + MonitorSnapshotParam, +) +from databricks.bundles.quality_monitors._models.monitor_time_series import ( + MonitorTimeSeries, + MonitorTimeSeriesParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class QualityMonitor(Resource): """""" @@ -57,7 +77,9 @@ class QualityMonitor(Resource): [Create:OPT Update:OPT] Custom metrics. """ - data_classification_config: VariableOrOptional[MonitorDataClassificationConfig] = None + data_classification_config: VariableOrOptional[MonitorDataClassificationConfig] = ( + None + ) """ :meta private: [EXPERIMENTAL] @@ -118,12 +140,11 @@ class QualityMonitor(Resource): """ @classmethod - def from_dict(cls, value: 'QualityMonitorDict') -> 'Self': + def from_dict(cls, value: "QualityMonitorDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'QualityMonitorDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "QualityMonitorDict": + return _transform_to_json_value(self) # type:ignore class QualityMonitorDict(TypedDict, total=False): diff --git a/python/databricks/bundles/registered_models/__init__.py b/python/databricks/bundles/registered_models/__init__.py index cd283ff2cfa..a7ae3ba9ef3 100644 --- a/python/databricks/bundles/registered_models/__init__.py +++ b/python/databricks/bundles/registered_models/__init__.py @@ -1,26 +1,44 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'RegisteredModel', - 'RegisteredModelAlias', - 'RegisteredModelAliasDict', - 'RegisteredModelAliasParam', - 'RegisteredModelDict', - 'RegisteredModelParam', + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", + "RegisteredModel", + "RegisteredModelAlias", + "RegisteredModelAliasDict", + "RegisteredModelAliasParam", + "RegisteredModelDict", + "RegisteredModelParam", ] -from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam -from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam -from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.registered_models._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.registered_models._models.privilege import ( + Privilege, + PrivilegeParam, +) +from databricks.bundles.registered_models._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) +from databricks.bundles.registered_models._models.registered_model import ( + RegisteredModel, + RegisteredModelDict, + RegisteredModelParam, +) +from databricks.bundles.registered_models._models.registered_model_alias import ( + RegisteredModelAlias, + RegisteredModelAliasDict, + RegisteredModelAliasParam, +) diff --git a/python/databricks/bundles/registered_models/_models/lifecycle.py b/python/databricks/bundles/registered_models/_models/lifecycle.py index c360ba26bfc..697776a198e 100644 --- a/python/databricks/bundles/registered_models/_models/lifecycle.py +++ b/python/databricks/bundles/registered_models/_models/lifecycle.py @@ -1,23 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam -from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam -from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -28,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/registered_models/_models/privilege.py b/python/databricks/bundles/registered_models/_models/privilege.py index 6b316c45dbc..df45211a76a 100644 --- a/python/databricks/bundles/registered_models/_models/privilege.py +++ b/python/databricks/bundles/registered_models/_models/privilege.py @@ -1,18 +1,8 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam -from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam class Privilege(Enum): SELECT = "SELECT" @@ -68,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/registered_models/_models/privilege_assignment.py b/python/databricks/bundles/registered_models/_models/privilege_assignment.py index a7b57d701c4..10ef2d094f8 100644 --- a/python/databricks/bundles/registered_models/_models/privilege_assignment.py +++ b/python/databricks/bundles/registered_models/_models/privilege_assignment.py @@ -1,23 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam -from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam -from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.registered_models._models.privilege import ( + Privilege, + PrivilegeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -34,12 +31,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/registered_models/_models/registered_model.py b/python/databricks/bundles/registered_models/_models/registered_model.py index e680d9f21de..08cf2ea45ae 100644 --- a/python/databricks/bundles/registered_models/_models/registered_model.py +++ b/python/databricks/bundles/registered_models/_models/registered_model.py @@ -1,23 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias, RegisteredModelAliasDict, RegisteredModelAliasParam -from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.registered_models._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.registered_models._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) +from databricks.bundles.registered_models._models.registered_model_alias import ( + RegisteredModelAlias, + RegisteredModelAliasParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RegisteredModel(Resource): """""" @@ -98,12 +104,11 @@ class RegisteredModel(Resource): """ @classmethod - def from_dict(cls, value: 'RegisteredModelDict') -> 'Self': + def from_dict(cls, value: "RegisteredModelDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RegisteredModelDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RegisteredModelDict": + return _transform_to_json_value(self) # type:ignore class RegisteredModelDict(TypedDict, total=False): diff --git a/python/databricks/bundles/registered_models/_models/registered_model_alias.py b/python/databricks/bundles/registered_models/_models/registered_model_alias.py index 4842cf5d944..f4ea570a9a6 100644 --- a/python/databricks/bundles/registered_models/_models/registered_model_alias.py +++ b/python/databricks/bundles/registered_models/_models/registered_model_alias.py @@ -1,23 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.registered_models._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.registered_models._models.registered_model import RegisteredModel, RegisteredModelDict, RegisteredModelParam -from databricks.bundles.registered_models._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class RegisteredModelAlias: """""" @@ -53,12 +46,11 @@ class RegisteredModelAlias: """ @classmethod - def from_dict(cls, value: 'RegisteredModelAliasDict') -> 'Self': + def from_dict(cls, value: "RegisteredModelAliasDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'RegisteredModelAliasDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "RegisteredModelAliasDict": + return _transform_to_json_value(self) # type:ignore class RegisteredModelAliasDict(TypedDict, total=False): diff --git a/python/databricks/bundles/schemas/__init__.py b/python/databricks/bundles/schemas/__init__.py index 9f9373136dd..e891669012c 100644 --- a/python/databricks/bundles/schemas/__init__.py +++ b/python/databricks/bundles/schemas/__init__.py @@ -1,30 +1,37 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'Schema', - 'SchemaDict', - 'SchemaParam', - 'SchemaGrant', - 'SchemaGrantDict', - 'SchemaGrantParam', - 'SchemaGrantPrivilege', - 'SchemaGrantPrivilegeParam', + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", + "Schema", + "SchemaDict", + "SchemaGrant", + "SchemaGrantDict", + "SchemaGrantParam", + "SchemaGrantPrivilege", + "SchemaGrantPrivilegeParam", + "SchemaParam", ] -from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam +from databricks.bundles.schemas._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) from databricks.bundles.schemas._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.schemas._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) +from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam SchemaGrant = PrivilegeAssignment SchemaGrantDict = PrivilegeAssignmentDict diff --git a/python/databricks/bundles/schemas/_models/lifecycle.py b/python/databricks/bundles/schemas/_models/lifecycle.py index c5c99d0e4bf..697776a198e 100644 --- a/python/databricks/bundles/schemas/_models/lifecycle.py +++ b/python/databricks/bundles/schemas/_models/lifecycle.py @@ -1,22 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam -from databricks.bundles.schemas._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -27,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/schemas/_models/privilege.py b/python/databricks/bundles/schemas/_models/privilege.py index cdd1f5348ac..df45211a76a 100644 --- a/python/databricks/bundles/schemas/_models/privilege.py +++ b/python/databricks/bundles/schemas/_models/privilege.py @@ -1,17 +1,8 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam class Privilege(Enum): SELECT = "SELECT" @@ -67,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/schemas/_models/privilege_assignment.py b/python/databricks/bundles/schemas/_models/privilege_assignment.py index 1155f70c633..012fe5b3b59 100644 --- a/python/databricks/bundles/schemas/_models/privilege_assignment.py +++ b/python/databricks/bundles/schemas/_models/privilege_assignment.py @@ -1,22 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.schemas._models.schema import Schema, SchemaDict, SchemaParam +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional from databricks.bundles.schemas._models.privilege import Privilege, PrivilegeParam - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -33,12 +28,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/schemas/_models/schema.py b/python/databricks/bundles/schemas/_models/schema.py index fa5bd691fd8..32ff22c8227 100644 --- a/python/databricks/bundles/schemas/_models/schema.py +++ b/python/databricks/bundles/schemas/_models/schema.py @@ -1,22 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.schemas._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrDict, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.schemas._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.schemas._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Schema(Resource): """""" @@ -62,12 +67,11 @@ class Schema(Resource): """ @classmethod - def from_dict(cls, value: 'SchemaDict') -> 'Self': + def from_dict(cls, value: "SchemaDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SchemaDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SchemaDict": + return _transform_to_json_value(self) # type:ignore class SchemaDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secret_scopes/__init__.py b/python/databricks/bundles/secret_scopes/__init__.py index c5882bdb65f..1f3d9c9555d 100644 --- a/python/databricks/bundles/secret_scopes/__init__.py +++ b/python/databricks/bundles/secret_scopes/__init__.py @@ -1,30 +1,50 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'AzureKeyVaultSecretScopeMetadata', - 'AzureKeyVaultSecretScopeMetadataDict', - 'AzureKeyVaultSecretScopeMetadataParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'ScopeBackendType', - 'ScopeBackendTypeParam', - 'SecretScope', - 'SecretScopeDict', - 'SecretScopeParam', - 'SecretScopePermission', - 'SecretScopePermissionDict', - 'SecretScopePermissionLevel', - 'SecretScopePermissionLevelParam', - 'SecretScopePermissionParam', + "AzureKeyVaultSecretScopeMetadata", + "AzureKeyVaultSecretScopeMetadataDict", + "AzureKeyVaultSecretScopeMetadataParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "ScopeBackendType", + "ScopeBackendTypeParam", + "SecretScope", + "SecretScopeDict", + "SecretScopeParam", + "SecretScopePermission", + "SecretScopePermissionDict", + "SecretScopePermissionLevel", + "SecretScopePermissionLevelParam", + "SecretScopePermissionParam", ] -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam -from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam -from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam -from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam - -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam - +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import ( + AzureKeyVaultSecretScopeMetadata, + AzureKeyVaultSecretScopeMetadataDict, + AzureKeyVaultSecretScopeMetadataParam, +) +from databricks.bundles.secret_scopes._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.secret_scopes._models.scope_backend_type import ( + ScopeBackendType, + ScopeBackendTypeParam, +) +from databricks.bundles.secret_scopes._models.secret_scope import ( + SecretScope, + SecretScopeDict, + SecretScopeParam, +) +from databricks.bundles.secret_scopes._models.secret_scope_permission import ( + SecretScopePermission, + SecretScopePermissionDict, + SecretScopePermissionParam, +) +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import ( + SecretScopePermissionLevel, + SecretScopePermissionLevelParam, +) diff --git a/python/databricks/bundles/secret_scopes/_models/azure_key_vault_secret_scope_metadata.py b/python/databricks/bundles/secret_scopes/_models/azure_key_vault_secret_scope_metadata.py index 8731cab043a..ceedb8b2ec4 100644 --- a/python/databricks/bundles/secret_scopes/_models/azure_key_vault_secret_scope_metadata.py +++ b/python/databricks/bundles/secret_scopes/_models/azure_key_vault_secret_scope_metadata.py @@ -1,30 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam -from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam -from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam - -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam - +from databricks.bundles.core._variable import VariableOr if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class AzureKeyVaultSecretScopeMetadata: """ The metadata of the Azure KeyVault for a secret scope of type `AZURE_KEYVAULT` """ + dns_name: VariableOr[str] """ The DNS of the KeyVault @@ -36,12 +28,11 @@ class AzureKeyVaultSecretScopeMetadata: """ @classmethod - def from_dict(cls, value: 'AzureKeyVaultSecretScopeMetadataDict') -> 'Self': + def from_dict(cls, value: "AzureKeyVaultSecretScopeMetadataDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'AzureKeyVaultSecretScopeMetadataDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "AzureKeyVaultSecretScopeMetadataDict": + return _transform_to_json_value(self) # type:ignore class AzureKeyVaultSecretScopeMetadataDict(TypedDict, total=False): @@ -58,4 +49,6 @@ class AzureKeyVaultSecretScopeMetadataDict(TypedDict, total=False): """ -AzureKeyVaultSecretScopeMetadataParam = AzureKeyVaultSecretScopeMetadataDict | AzureKeyVaultSecretScopeMetadata +AzureKeyVaultSecretScopeMetadataParam = ( + AzureKeyVaultSecretScopeMetadataDict | AzureKeyVaultSecretScopeMetadata +) diff --git a/python/databricks/bundles/secret_scopes/_models/lifecycle.py b/python/databricks/bundles/secret_scopes/_models/lifecycle.py index 16e88694a25..697776a198e 100644 --- a/python/databricks/bundles/secret_scopes/_models/lifecycle.py +++ b/python/databricks/bundles/secret_scopes/_models/lifecycle.py @@ -1,25 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam -from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam -from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam -from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam - -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -30,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secret_scopes/_models/scope_backend_type.py b/python/databricks/bundles/secret_scopes/_models/scope_backend_type.py index 9bc901f7b48..d95f7022d33 100644 --- a/python/databricks/bundles/secret_scopes/_models/scope_backend_type.py +++ b/python/databricks/bundles/secret_scopes/_models/scope_backend_type.py @@ -1,19 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam -from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam -from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam +from typing import Literal class ScopeBackendType(Enum): @@ -21,7 +9,9 @@ class ScopeBackendType(Enum): The types of secret scope backends in the Secret Manager. Azure KeyVault backed secret scopes will be supported in a later release. """ + DATABRICKS = "DATABRICKS" AZURE_KEYVAULT = "AZURE_KEYVAULT" + ScopeBackendTypeParam = Literal["DATABRICKS", "AZURE_KEYVAULT"] | ScopeBackendType diff --git a/python/databricks/bundles/secret_scopes/_models/secret_scope.py b/python/databricks/bundles/secret_scopes/_models/secret_scope.py index fa36c384708..5e436f1d247 100644 --- a/python/databricks/bundles/secret_scopes/_models/secret_scope.py +++ b/python/databricks/bundles/secret_scopes/_models/secret_scope.py @@ -1,25 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam -from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam -from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam - -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import ( + AzureKeyVaultSecretScopeMetadata, + AzureKeyVaultSecretScopeMetadataParam, +) +from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.secret_scopes._models.scope_backend_type import ( + ScopeBackendType, + ScopeBackendTypeParam, +) +from databricks.bundles.secret_scopes._models.secret_scope_permission import ( + SecretScopePermission, + SecretScopePermissionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SecretScope(Resource): """""" @@ -50,12 +59,11 @@ class SecretScope(Resource): """ @classmethod - def from_dict(cls, value: 'SecretScopeDict') -> 'Self': + def from_dict(cls, value: "SecretScopeDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SecretScopeDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SecretScopeDict": + return _transform_to_json_value(self) # type:ignore class SecretScopeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secret_scopes/_models/secret_scope_permission.py b/python/databricks/bundles/secret_scopes/_models/secret_scope_permission.py index 47726469652..3715039c411 100644 --- a/python/databricks/bundles/secret_scopes/_models/secret_scope_permission.py +++ b/python/databricks/bundles/secret_scopes/_models/secret_scope_permission.py @@ -1,25 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam -from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam -from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam - -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel, SecretScopePermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import ( + SecretScopePermissionLevel, + SecretScopePermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SecretScopePermission: """""" @@ -45,12 +40,11 @@ class SecretScopePermission: """ @classmethod - def from_dict(cls, value: 'SecretScopePermissionDict') -> 'Self': + def from_dict(cls, value: "SecretScopePermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SecretScopePermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SecretScopePermissionDict": + return _transform_to_json_value(self) # type:ignore class SecretScopePermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secret_scopes/_models/secret_scope_permission_level.py b/python/databricks/bundles/secret_scopes/_models/secret_scope_permission_level.py index 373403cb375..d7d84cd5854 100644 --- a/python/databricks/bundles/secret_scopes/_models/secret_scope_permission_level.py +++ b/python/databricks/bundles/secret_scopes/_models/secret_scope_permission_level.py @@ -1,19 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata, AzureKeyVaultSecretScopeMetadataDict, AzureKeyVaultSecretScopeMetadataParam -from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secret_scopes._models.secret_scope import SecretScope, SecretScopeDict, SecretScopeParam -from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission, SecretScopePermissionDict, SecretScopePermissionParam -from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType, ScopeBackendTypeParam +from typing import Literal class SecretScopePermissionLevel(Enum): @@ -21,4 +9,7 @@ class SecretScopePermissionLevel(Enum): WRITE = "WRITE" MANAGE = "MANAGE" -SecretScopePermissionLevelParam = Literal["READ", "WRITE", "MANAGE"] | SecretScopePermissionLevel + +SecretScopePermissionLevelParam = ( + Literal["READ", "WRITE", "MANAGE"] | SecretScopePermissionLevel +) diff --git a/python/databricks/bundles/secrets/__init__.py b/python/databricks/bundles/secrets/__init__.py index feb447b97fc..6bc14c2a19d 100644 --- a/python/databricks/bundles/secrets/__init__.py +++ b/python/databricks/bundles/secrets/__init__.py @@ -1,22 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'Secret', - 'SecretDict', - 'SecretParam', + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", + "Secret", + "SecretDict", + "SecretParam", ] -from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secrets._models.secret import Secret, SecretDict, SecretParam +from databricks.bundles.secrets._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) from databricks.bundles.secrets._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.secrets._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) +from databricks.bundles.secrets._models.secret import Secret, SecretDict, SecretParam diff --git a/python/databricks/bundles/secrets/_models/lifecycle.py b/python/databricks/bundles/secrets/_models/lifecycle.py index 81ddf91d6b7..697776a198e 100644 --- a/python/databricks/bundles/secrets/_models/lifecycle.py +++ b/python/databricks/bundles/secrets/_models/lifecycle.py @@ -1,22 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.secrets._models.secret import Secret, SecretDict, SecretParam -from databricks.bundles.secrets._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -27,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secrets/_models/privilege.py b/python/databricks/bundles/secrets/_models/privilege.py index 543526c1089..df45211a76a 100644 --- a/python/databricks/bundles/secrets/_models/privilege.py +++ b/python/databricks/bundles/secrets/_models/privilege.py @@ -1,17 +1,8 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field +from typing import Literal -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secrets._models.secret import Secret, SecretDict, SecretParam class Privilege(Enum): SELECT = "SELECT" @@ -67,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/secrets/_models/privilege_assignment.py b/python/databricks/bundles/secrets/_models/privilege_assignment.py index 4c2c7500501..443fa0120c7 100644 --- a/python/databricks/bundles/secrets/_models/privilege_assignment.py +++ b/python/databricks/bundles/secrets/_models/privilege_assignment.py @@ -1,22 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secrets._models.secret import Secret, SecretDict, SecretParam +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional from databricks.bundles.secrets._models.privilege import Privilege, PrivilegeParam - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -33,12 +28,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/secrets/_models/secret.py b/python/databricks/bundles/secrets/_models/secret.py index 4bfbcbcf219..b0db4c03ada 100644 --- a/python/databricks/bundles/secrets/_models/secret.py +++ b/python/databricks/bundles/secrets/_models/secret.py @@ -1,22 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.secrets._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.secrets._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.secrets._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Secret(Resource): """ @@ -24,6 +28,7 @@ class Secret(Resource): (catalog.schema.secret) that securely store sensitive credential data such as passwords, tokens, and keys. """ + catalog_name: VariableOr[str] """ The name of the catalog where the schema and the secret reside. @@ -71,12 +76,11 @@ class Secret(Resource): """ @classmethod - def from_dict(cls, value: 'SecretDict') -> 'Self': + def from_dict(cls, value: "SecretDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SecretDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SecretDict": + return _transform_to_json_value(self) # type:ignore class SecretDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/__init__.py b/python/databricks/bundles/sql_warehouses/__init__.py index 5a397048264..4866d2ca86e 100644 --- a/python/databricks/bundles/sql_warehouses/__init__.py +++ b/python/databricks/bundles/sql_warehouses/__init__.py @@ -1,46 +1,78 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Channel', - 'ChannelDict', - 'ChannelName', - 'ChannelNameParam', - 'ChannelParam', - 'CreateWarehouseRequestWarehouseType', - 'CreateWarehouseRequestWarehouseTypeParam', - 'EndpointTagPair', - 'EndpointTagPairDict', - 'EndpointTagPairParam', - 'EndpointTags', - 'EndpointTagsDict', - 'EndpointTagsParam', - 'LifecycleWithStarted', - 'LifecycleWithStartedDict', - 'LifecycleWithStartedParam', - 'SpotInstancePolicy', - 'SpotInstancePolicyParam', - 'SqlWarehouse', - 'SqlWarehouseDict', - 'SqlWarehouseParam', - 'SqlWarehousePermission', - 'SqlWarehousePermissionDict', - 'SqlWarehousePermissionParam', - 'WarehousePermissionLevel', - 'WarehousePermissionLevelParam', + "Channel", + "ChannelDict", + "ChannelName", + "ChannelNameParam", + "ChannelParam", + "CreateWarehouseRequestWarehouseType", + "CreateWarehouseRequestWarehouseTypeParam", + "EndpointTagPair", + "EndpointTagPairDict", + "EndpointTagPairParam", + "EndpointTags", + "EndpointTagsDict", + "EndpointTagsParam", + "LifecycleWithStarted", + "LifecycleWithStartedDict", + "LifecycleWithStartedParam", + "SpotInstancePolicy", + "SpotInstancePolicyParam", + "SqlWarehouse", + "SqlWarehouseDict", + "SqlWarehouseParam", + "SqlWarehousePermission", + "SqlWarehousePermissionDict", + "SqlWarehousePermissionParam", + "WarehousePermissionLevel", + "WarehousePermissionLevelParam", ] -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam - +from databricks.bundles.sql_warehouses._models.channel import ( + Channel, + ChannelDict, + ChannelParam, +) +from databricks.bundles.sql_warehouses._models.channel_name import ( + ChannelName, + ChannelNameParam, +) +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import ( + CreateWarehouseRequestWarehouseType, + CreateWarehouseRequestWarehouseTypeParam, +) +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import ( + EndpointTagPair, + EndpointTagPairDict, + EndpointTagPairParam, +) +from databricks.bundles.sql_warehouses._models.endpoint_tags import ( + EndpointTags, + EndpointTagsDict, + EndpointTagsParam, +) +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import ( + LifecycleWithStarted, + LifecycleWithStartedDict, + LifecycleWithStartedParam, +) +from databricks.bundles.sql_warehouses._models.spot_instance_policy import ( + SpotInstancePolicy, + SpotInstancePolicyParam, +) +from databricks.bundles.sql_warehouses._models.sql_warehouse import ( + SqlWarehouse, + SqlWarehouseDict, + SqlWarehouseParam, +) +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import ( + SqlWarehousePermission, + SqlWarehousePermissionDict, + SqlWarehousePermissionParam, +) +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import ( + WarehousePermissionLevel, + WarehousePermissionLevelParam, +) diff --git a/python/databricks/bundles/sql_warehouses/_models/channel.py b/python/databricks/bundles/sql_warehouses/_models/channel.py index 367ed75dd55..f5893d7273f 100644 --- a/python/databricks/bundles/sql_warehouses/_models/channel.py +++ b/python/databricks/bundles/sql_warehouses/_models/channel.py @@ -1,47 +1,36 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional +from databricks.bundles.sql_warehouses._models.channel_name import ( + ChannelName, + ChannelNameParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Channel: """ Configures the channel name and DBSQL version of the warehouse. CHANNEL_NAME_CUSTOM should be chosen only when `dbsql_version` is specified. """ + dbsql_version: VariableOrOptional[str] = None name: VariableOrOptional[ChannelName] = None @classmethod - def from_dict(cls, value: 'ChannelDict') -> 'Self': + def from_dict(cls, value: "ChannelDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'ChannelDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "ChannelDict": + return _transform_to_json_value(self) # type:ignore class ChannelDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/channel_name.py b/python/databricks/bundles/sql_warehouses/_models/channel_name.py index 8d1f8fef0c3..a47e1c890f4 100644 --- a/python/databricks/bundles/sql_warehouses/_models/channel_name.py +++ b/python/databricks/bundles/sql_warehouses/_models/channel_name.py @@ -1,25 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam +from typing import Literal class ChannelName(Enum): @@ -28,4 +10,13 @@ class ChannelName(Enum): CHANNEL_NAME_PREVIOUS = "CHANNEL_NAME_PREVIOUS" CHANNEL_NAME_CUSTOM = "CHANNEL_NAME_CUSTOM" -ChannelNameParam = Literal["CHANNEL_NAME_PREVIEW", "CHANNEL_NAME_CURRENT", "CHANNEL_NAME_PREVIOUS", "CHANNEL_NAME_CUSTOM"] | ChannelName + +ChannelNameParam = ( + Literal[ + "CHANNEL_NAME_PREVIEW", + "CHANNEL_NAME_CURRENT", + "CHANNEL_NAME_PREVIOUS", + "CHANNEL_NAME_CUSTOM", + ] + | ChannelName +) diff --git a/python/databricks/bundles/sql_warehouses/_models/create_warehouse_request_warehouse_type.py b/python/databricks/bundles/sql_warehouses/_models/create_warehouse_request_warehouse_type.py index d4f26614cfa..feb20fbbbee 100644 --- a/python/databricks/bundles/sql_warehouses/_models/create_warehouse_request_warehouse_type.py +++ b/python/databricks/bundles/sql_warehouses/_models/create_warehouse_request_warehouse_type.py @@ -1,25 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam +from typing import Literal class CreateWarehouseRequestWarehouseType(Enum): @@ -28,4 +10,8 @@ class CreateWarehouseRequestWarehouseType(Enum): PRO = "PRO" REALTIME = "REALTIME" -CreateWarehouseRequestWarehouseTypeParam = Literal["TYPE_UNSPECIFIED", "CLASSIC", "PRO", "REALTIME"] | CreateWarehouseRequestWarehouseType + +CreateWarehouseRequestWarehouseTypeParam = ( + Literal["TYPE_UNSPECIFIED", "CLASSIC", "PRO", "REALTIME"] + | CreateWarehouseRequestWarehouseType +) diff --git a/python/databricks/bundles/sql_warehouses/_models/endpoint_tag_pair.py b/python/databricks/bundles/sql_warehouses/_models/endpoint_tag_pair.py index b2ae77f8606..45f4d904d1e 100644 --- a/python/databricks/bundles/sql_warehouses/_models/endpoint_tag_pair.py +++ b/python/databricks/bundles/sql_warehouses/_models/endpoint_tag_pair.py @@ -1,31 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EndpointTagPair: """""" @@ -35,12 +20,11 @@ class EndpointTagPair: value: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: 'EndpointTagPairDict') -> 'Self': + def from_dict(cls, value: "EndpointTagPairDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EndpointTagPairDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EndpointTagPairDict": + return _transform_to_json_value(self) # type:ignore class EndpointTagPairDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/endpoint_tags.py b/python/databricks/bundles/sql_warehouses/_models/endpoint_tags.py index 0d74becb5fd..7d70c86199d 100644 --- a/python/databricks/bundles/sql_warehouses/_models/endpoint_tags.py +++ b/python/databricks/bundles/sql_warehouses/_models/endpoint_tags.py @@ -1,31 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam - +from databricks.bundles.core._variable import VariableOrList +from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import ( + EndpointTagPair, + EndpointTagPairParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EndpointTags: """""" @@ -33,12 +22,11 @@ class EndpointTags: custom_tags: VariableOrList[EndpointTagPair] = field(default_factory=list) @classmethod - def from_dict(cls, value: 'EndpointTagsDict') -> 'Self': + def from_dict(cls, value: "EndpointTagsDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EndpointTagsDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EndpointTagsDict": + return _transform_to_json_value(self) # type:ignore class EndpointTagsDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/lifecycle_with_started.py b/python/databricks/bundles/sql_warehouses/_models/lifecycle_with_started.py index 3092b166ce0..f9cb9b8d49c 100644 --- a/python/databricks/bundles/sql_warehouses/_models/lifecycle_with_started.py +++ b/python/databricks/bundles/sql_warehouses/_models/lifecycle_with_started.py @@ -1,31 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class LifecycleWithStarted: """""" @@ -41,12 +26,11 @@ class LifecycleWithStarted: """ @classmethod - def from_dict(cls, value: 'LifecycleWithStartedDict') -> 'Self': + def from_dict(cls, value: "LifecycleWithStartedDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleWithStartedDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleWithStartedDict": + return _transform_to_json_value(self) # type:ignore class LifecycleWithStartedDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/spot_instance_policy.py b/python/databricks/bundles/sql_warehouses/_models/spot_instance_policy.py index 6216f63086d..e0d5105a95f 100644 --- a/python/databricks/bundles/sql_warehouses/_models/spot_instance_policy.py +++ b/python/databricks/bundles/sql_warehouses/_models/spot_instance_policy.py @@ -1,42 +1,29 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam +from typing import Literal class SpotInstancePolicy(Enum): """ EndpointSpotInstancePolicy configures whether the endpoint should use spot instances. - + The breakdown of how the EndpointSpotInstancePolicy converts to per cloud configurations is: - + - AWS, COST_OPTIMIZED: On Demand Driver with Spot Executors. - AWS, RELIABILITY_OPTIMIZED: On Demand Driver and Executors. - AZURE, COST_OPTIMIZED: On Demand Driver and Executors. - AZURE, RELIABILITY_OPTIMIZED: On Demand Driver and Executors. """ + POLICY_UNSPECIFIED = "POLICY_UNSPECIFIED" COST_OPTIMIZED = "COST_OPTIMIZED" RELIABILITY_OPTIMIZED = "RELIABILITY_OPTIMIZED" -SpotInstancePolicyParam = Literal["POLICY_UNSPECIFIED", "COST_OPTIMIZED", "RELIABILITY_OPTIMIZED"] | SpotInstancePolicy + +SpotInstancePolicyParam = ( + Literal["POLICY_UNSPECIFIED", "COST_OPTIMIZED", "RELIABILITY_OPTIMIZED"] + | SpotInstancePolicy +) diff --git a/python/databricks/bundles/sql_warehouses/_models/sql_warehouse.py b/python/databricks/bundles/sql_warehouses/_models/sql_warehouse.py index caf96bb8b0e..f3033f894ab 100644 --- a/python/databricks/bundles/sql_warehouses/_models/sql_warehouse.py +++ b/python/databricks/bundles/sql_warehouses/_models/sql_warehouse.py @@ -1,36 +1,44 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelParam +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import ( + CreateWarehouseRequestWarehouseType, + CreateWarehouseRequestWarehouseTypeParam, +) +from databricks.bundles.sql_warehouses._models.endpoint_tags import ( + EndpointTags, + EndpointTagsParam, +) +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import ( + LifecycleWithStarted, + LifecycleWithStartedParam, +) +from databricks.bundles.sql_warehouses._models.spot_instance_policy import ( + SpotInstancePolicy, + SpotInstancePolicyParam, +) +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import ( + SqlWarehousePermission, + SqlWarehousePermissionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SqlWarehouse(Resource): """ Creates a new SQL warehouse. """ + auto_stop_mins: VariableOrOptional[int] = None """ The amount of time in minutes that a SQL warehouse must be idle (i.e., no @@ -188,12 +196,11 @@ class SqlWarehouse(Resource): """ @classmethod - def from_dict(cls, value: 'SqlWarehouseDict') -> 'Self': + def from_dict(cls, value: "SqlWarehouseDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SqlWarehouseDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SqlWarehouseDict": + return _transform_to_json_value(self) # type:ignore class SqlWarehouseDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/sql_warehouse_permission.py b/python/databricks/bundles/sql_warehouses/_models/sql_warehouse_permission.py index 559b5b9acf8..2c9f284a2c0 100644 --- a/python/databricks/bundles/sql_warehouses/_models/sql_warehouse_permission.py +++ b/python/databricks/bundles/sql_warehouses/_models/sql_warehouse_permission.py @@ -1,31 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam - -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel, WarehousePermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import ( + WarehousePermissionLevel, + WarehousePermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SqlWarehousePermission: """""" @@ -51,12 +40,11 @@ class SqlWarehousePermission: """ @classmethod - def from_dict(cls, value: 'SqlWarehousePermissionDict') -> 'Self': + def from_dict(cls, value: "SqlWarehousePermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SqlWarehousePermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SqlWarehousePermissionDict": + return _transform_to_json_value(self) # type:ignore class SqlWarehousePermissionDict(TypedDict, total=False): diff --git a/python/databricks/bundles/sql_warehouses/_models/warehouse_permission_level.py b/python/databricks/bundles/sql_warehouses/_models/warehouse_permission_level.py index 385a19e3ff2..5728daa0d12 100644 --- a/python/databricks/bundles/sql_warehouses/_models/warehouse_permission_level.py +++ b/python/databricks/bundles/sql_warehouses/_models/warehouse_permission_level.py @@ -1,35 +1,22 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.sql_warehouses._models.channel import Channel, ChannelDict, ChannelParam -from databricks.bundles.sql_warehouses._models.endpoint_tag_pair import EndpointTagPair, EndpointTagPairDict, EndpointTagPairParam -from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags, EndpointTagsDict, EndpointTagsParam -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted, LifecycleWithStartedDict, LifecycleWithStartedParam -from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse, SqlWarehouseDict, SqlWarehouseParam -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission, SqlWarehousePermissionDict, SqlWarehousePermissionParam -from databricks.bundles.sql_warehouses._models.channel_name import ChannelName, ChannelNameParam - -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType, CreateWarehouseRequestWarehouseTypeParam - -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy, SpotInstancePolicyParam +from typing import Literal class WarehousePermissionLevel(Enum): """ Permission level """ + CAN_MANAGE = "CAN_MANAGE" IS_OWNER = "IS_OWNER" CAN_USE = "CAN_USE" CAN_MONITOR = "CAN_MONITOR" CAN_VIEW = "CAN_VIEW" -WarehousePermissionLevelParam = Literal["CAN_MANAGE", "IS_OWNER", "CAN_USE", "CAN_MONITOR", "CAN_VIEW"] | WarehousePermissionLevel + +WarehousePermissionLevelParam = ( + Literal["CAN_MANAGE", "IS_OWNER", "CAN_USE", "CAN_MONITOR", "CAN_VIEW"] + | WarehousePermissionLevel +) diff --git a/python/databricks/bundles/synced_database_tables/__init__.py b/python/databricks/bundles/synced_database_tables/__init__.py index a0729cab266..0b3a240e460 100644 --- a/python/databricks/bundles/synced_database_tables/__init__.py +++ b/python/databricks/bundles/synced_database_tables/__init__.py @@ -1,34 +1,58 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'NewPipelineSpec', - 'NewPipelineSpecDict', - 'NewPipelineSpecParam', - 'SyncedDatabaseTable', - 'SyncedDatabaseTableDict', - 'SyncedDatabaseTableParam', - 'SyncedTableSchedulingPolicy', - 'SyncedTableSchedulingPolicyParam', - 'SyncedTableSpec', - 'SyncedTableSpecDict', - 'SyncedTableSpecParam', - 'SyncedTableSpecPgSpecificType', - 'SyncedTableSpecPgSpecificTypeParam', - 'SyncedTableSpecTypeOverride', - 'SyncedTableSpecTypeOverrideDict', - 'SyncedTableSpecTypeOverrideParam', + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "NewPipelineSpec", + "NewPipelineSpecDict", + "NewPipelineSpecParam", + "SyncedDatabaseTable", + "SyncedDatabaseTableDict", + "SyncedDatabaseTableParam", + "SyncedTableSchedulingPolicy", + "SyncedTableSchedulingPolicyParam", + "SyncedTableSpec", + "SyncedTableSpecDict", + "SyncedTableSpecParam", + "SyncedTableSpecPgSpecificType", + "SyncedTableSpecPgSpecificTypeParam", + "SyncedTableSpecTypeOverride", + "SyncedTableSpecTypeOverrideDict", + "SyncedTableSpecTypeOverrideParam", ] -from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam -from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam -from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam - -from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam - +from databricks.bundles.synced_database_tables._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.synced_database_tables._models.new_pipeline_spec import ( + NewPipelineSpec, + NewPipelineSpecDict, + NewPipelineSpecParam, +) +from databricks.bundles.synced_database_tables._models.synced_database_table import ( + SyncedDatabaseTable, + SyncedDatabaseTableDict, + SyncedDatabaseTableParam, +) +from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import ( + SyncedTableSchedulingPolicy, + SyncedTableSchedulingPolicyParam, +) +from databricks.bundles.synced_database_tables._models.synced_table_spec import ( + SyncedTableSpec, + SyncedTableSpecDict, + SyncedTableSpecParam, +) +from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import ( + SyncedTableSpecPgSpecificType, + SyncedTableSpecPgSpecificTypeParam, +) +from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import ( + SyncedTableSpecTypeOverride, + SyncedTableSpecTypeOverrideDict, + SyncedTableSpecTypeOverrideParam, +) diff --git a/python/databricks/bundles/synced_database_tables/_models/lifecycle.py b/python/databricks/bundles/synced_database_tables/_models/lifecycle.py index cbfe5e2bb92..697776a198e 100644 --- a/python/databricks/bundles/synced_database_tables/_models/lifecycle.py +++ b/python/databricks/bundles/synced_database_tables/_models/lifecycle.py @@ -1,26 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam -from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam -from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam - -from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -31,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/synced_database_tables/_models/new_pipeline_spec.py b/python/databricks/bundles/synced_database_tables/_models/new_pipeline_spec.py index cf5896231ff..9a47da58441 100644 --- a/python/databricks/bundles/synced_database_tables/_models/new_pipeline_spec.py +++ b/python/databricks/bundles/synced_database_tables/_models/new_pipeline_spec.py @@ -1,32 +1,23 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam -from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam -from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam - -from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class NewPipelineSpec: """ Custom fields that user can set for pipeline while creating SyncedDatabaseTable. Note that other fields of pipeline are still inferred by table def internally """ + budget_policy_id: VariableOrOptional[str] = None """ :meta private: [EXPERIMENTAL] @@ -51,12 +42,11 @@ class NewPipelineSpec: """ @classmethod - def from_dict(cls, value: 'NewPipelineSpecDict') -> 'Self': + def from_dict(cls, value: "NewPipelineSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'NewPipelineSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "NewPipelineSpecDict": + return _transform_to_json_value(self) # type:ignore class NewPipelineSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_database_table.py b/python/databricks/bundles/synced_database_tables/_models/synced_database_table.py index 0efeacfaec9..c98b869b76e 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_database_table.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_database_table.py @@ -1,26 +1,25 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam -from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam - -from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.synced_database_tables._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.synced_database_tables._models.synced_table_spec import ( + SyncedTableSpec, + SyncedTableSpecParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SyncedDatabaseTable(Resource): """""" @@ -63,12 +62,11 @@ class SyncedDatabaseTable(Resource): """ @classmethod - def from_dict(cls, value: 'SyncedDatabaseTableDict') -> 'Self': + def from_dict(cls, value: "SyncedDatabaseTableDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SyncedDatabaseTableDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SyncedDatabaseTableDict": + return _transform_to_json_value(self) # type:ignore class SyncedDatabaseTableDict(TypedDict, total=False): diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_table_scheduling_policy.py b/python/databricks/bundles/synced_database_tables/_models/synced_table_scheduling_policy.py index d2f6f64abb2..379d61c2c17 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_table_scheduling_policy.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_table_scheduling_policy.py @@ -1,20 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam -from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam -from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam +from typing import Literal class SyncedTableSchedulingPolicy(Enum): @@ -22,4 +9,7 @@ class SyncedTableSchedulingPolicy(Enum): TRIGGERED = "TRIGGERED" SNAPSHOT = "SNAPSHOT" -SyncedTableSchedulingPolicyParam = Literal["CONTINUOUS", "TRIGGERED", "SNAPSHOT"] | SyncedTableSchedulingPolicy + +SyncedTableSchedulingPolicyParam = ( + Literal["CONTINUOUS", "TRIGGERED", "SNAPSHOT"] | SyncedTableSchedulingPolicy +) diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec.py b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec.py index bd32c91936f..c6be97882ae 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec.py @@ -1,31 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam -from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam -from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam - -from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.synced_database_tables._models.new_pipeline_spec import ( + NewPipelineSpec, + NewPipelineSpecParam, +) +from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import ( + SyncedTableSchedulingPolicy, + SyncedTableSchedulingPolicyParam, +) +from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import ( + SyncedTableSpecTypeOverride, + SyncedTableSpecTypeOverrideParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SyncedTableSpec: """ Specification of a synced database table. """ + accelerated_sync: VariableOrOptional[bool] = None """ :meta private: [EXPERIMENTAL] @@ -80,7 +83,9 @@ class SyncedTableSpec: [Public Preview] Time series key to deduplicate (tie-break) rows with the same primary key. """ - type_overrides: VariableOrList[SyncedTableSpecTypeOverride] = field(default_factory=list) + type_overrides: VariableOrList[SyncedTableSpecTypeOverride] = field( + default_factory=list + ) """ :meta private: [EXPERIMENTAL] @@ -89,12 +94,11 @@ class SyncedTableSpec: """ @classmethod - def from_dict(cls, value: 'SyncedTableSpecDict') -> 'Self': + def from_dict(cls, value: "SyncedTableSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SyncedTableSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SyncedTableSpecDict": + return _transform_to_json_value(self) # type:ignore class SyncedTableSpecDict(TypedDict, total=False): diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_pg_specific_type.py b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_pg_specific_type.py index e77b35923e7..52c3b8dbe9a 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_pg_specific_type.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_pg_specific_type.py @@ -1,30 +1,26 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec_type_override import SyncedTableSpecTypeOverride, SyncedTableSpecTypeOverrideDict, SyncedTableSpecTypeOverrideParam -from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam -from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam +from typing import Literal class SyncedTableSpecPgSpecificType(Enum): """ :meta private: [EXPERIMENTAL] - + PostgreSQL-specific target types that can override the default Delta-to-PG mapping. """ + PG_SPECIFIC_TYPE_VECTOR = "PG_SPECIFIC_TYPE_VECTOR" PG_SPECIFIC_TYPE_HALFVEC = "PG_SPECIFIC_TYPE_HALFVEC" PG_SPECIFIC_TYPE_VARCHAR = "PG_SPECIFIC_TYPE_VARCHAR" -SyncedTableSpecPgSpecificTypeParam = Literal["PG_SPECIFIC_TYPE_VECTOR", "PG_SPECIFIC_TYPE_HALFVEC", "PG_SPECIFIC_TYPE_VARCHAR"] | SyncedTableSpecPgSpecificType + +SyncedTableSpecPgSpecificTypeParam = ( + Literal[ + "PG_SPECIFIC_TYPE_VECTOR", + "PG_SPECIFIC_TYPE_HALFVEC", + "PG_SPECIFIC_TYPE_VARCHAR", + ] + | SyncedTableSpecPgSpecificType +) diff --git a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_type_override.py b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_type_override.py index 63100d41d37..5cc3fa72720 100644 --- a/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_type_override.py +++ b/python/databricks/bundles/synced_database_tables/_models/synced_table_spec_type_override.py @@ -1,33 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.synced_database_tables._models.new_pipeline_spec import NewPipelineSpec, NewPipelineSpecDict, NewPipelineSpecParam -from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec, SyncedTableSpecDict, SyncedTableSpecParam -from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable, SyncedDatabaseTableDict, SyncedDatabaseTableParam -from databricks.bundles.synced_database_tables._models.synced_table_scheduling_policy import SyncedTableSchedulingPolicy, SyncedTableSchedulingPolicyParam - -from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import SyncedTableSpecPgSpecificType, SyncedTableSpecPgSpecificTypeParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.synced_database_tables._models.synced_table_spec_pg_specific_type import ( + SyncedTableSpecPgSpecificType, + SyncedTableSpecPgSpecificTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class SyncedTableSpecTypeOverride: """ :meta private: [EXPERIMENTAL] - + Overrides the default Delta-to-PostgreSQL type mapping for a single column. """ + column_name: VariableOr[str] """ :meta private: [EXPERIMENTAL] @@ -51,12 +46,11 @@ class SyncedTableSpecTypeOverride: """ @classmethod - def from_dict(cls, value: 'SyncedTableSpecTypeOverrideDict') -> 'Self': + def from_dict(cls, value: "SyncedTableSpecTypeOverrideDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'SyncedTableSpecTypeOverrideDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "SyncedTableSpecTypeOverrideDict": + return _transform_to_json_value(self) # type:ignore class SyncedTableSpecTypeOverrideDict(TypedDict, total=False): @@ -85,4 +79,6 @@ class SyncedTableSpecTypeOverrideDict(TypedDict, total=False): """ -SyncedTableSpecTypeOverrideParam = SyncedTableSpecTypeOverrideDict | SyncedTableSpecTypeOverride +SyncedTableSpecTypeOverrideParam = ( + SyncedTableSpecTypeOverrideDict | SyncedTableSpecTypeOverride +) diff --git a/python/databricks/bundles/vector_search_endpoints/__init__.py b/python/databricks/bundles/vector_search_endpoints/__init__.py index 223e3cf67b2..aabc0081f5d 100644 --- a/python/databricks/bundles/vector_search_endpoints/__init__.py +++ b/python/databricks/bundles/vector_search_endpoints/__init__.py @@ -1,26 +1,42 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'EndpointType', - 'EndpointTypeParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'VectorSearchEndpoint', - 'VectorSearchEndpointDict', - 'VectorSearchEndpointParam', - 'VectorSearchEndpointPermission', - 'VectorSearchEndpointPermissionDict', - 'VectorSearchEndpointPermissionLevel', - 'VectorSearchEndpointPermissionLevelParam', - 'VectorSearchEndpointPermissionParam', + "EndpointType", + "EndpointTypeParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "VectorSearchEndpoint", + "VectorSearchEndpointDict", + "VectorSearchEndpointParam", + "VectorSearchEndpointPermission", + "VectorSearchEndpointPermissionDict", + "VectorSearchEndpointPermissionLevel", + "VectorSearchEndpointPermissionLevelParam", + "VectorSearchEndpointPermissionParam", ] -from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam -from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam - -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam - +from databricks.bundles.vector_search_endpoints._models.endpoint_type import ( + EndpointType, + EndpointTypeParam, +) +from databricks.bundles.vector_search_endpoints._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( + VectorSearchEndpoint, + VectorSearchEndpointDict, + VectorSearchEndpointParam, +) +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import ( + VectorSearchEndpointPermission, + VectorSearchEndpointPermissionDict, + VectorSearchEndpointPermissionParam, +) +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import ( + VectorSearchEndpointPermissionLevel, + VectorSearchEndpointPermissionLevelParam, +) diff --git a/python/databricks/bundles/vector_search_endpoints/_models/endpoint_type.py b/python/databricks/bundles/vector_search_endpoints/_models/endpoint_type.py index a26e8dadddf..476295df34b 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/endpoint_type.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/endpoint_type.py @@ -1,25 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam +from typing import Literal class EndpointType(Enum): """ Type of endpoint. """ + STORAGE_OPTIMIZED = "STORAGE_OPTIMIZED" STANDARD = "STANDARD" + EndpointTypeParam = Literal["STORAGE_OPTIMIZED", "STANDARD"] | EndpointType diff --git a/python/databricks/bundles/vector_search_endpoints/_models/lifecycle.py b/python/databricks/bundles/vector_search_endpoints/_models/lifecycle.py index 0d29e52955b..697776a198e 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/lifecycle.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/lifecycle.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam -from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam - -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -29,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint.py b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint.py index d01810ba304..8072329f47b 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint.py @@ -1,24 +1,33 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam -from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam - -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.vector_search_endpoints._models.endpoint_type import ( + EndpointType, + EndpointTypeParam, +) +from databricks.bundles.vector_search_endpoints._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import ( + VectorSearchEndpointPermission, + VectorSearchEndpointPermissionParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class VectorSearchEndpoint(Resource): """""" @@ -43,7 +52,9 @@ class VectorSearchEndpoint(Resource): Settings that control the deployment lifecycle of the resource, such as preventing it from being destroyed. """ - permissions: VariableOrList[VectorSearchEndpointPermission] = field(default_factory=list) + permissions: VariableOrList[VectorSearchEndpointPermission] = field( + default_factory=list + ) """ The permissions to apply to this resource. """ @@ -63,12 +74,11 @@ class VectorSearchEndpoint(Resource): """ @classmethod - def from_dict(cls, value: 'VectorSearchEndpointDict') -> 'Self': + def from_dict(cls, value: "VectorSearchEndpointDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'VectorSearchEndpointDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "VectorSearchEndpointDict": + return _transform_to_json_value(self) # type:ignore class VectorSearchEndpointDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission.py b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission.py index 7d779ac64f0..cfbdc1e31d6 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission.py @@ -1,24 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam -from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam - -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel, VectorSearchEndpointPermissionLevelParam - +from databricks.bundles.core._variable import VariableOr, VariableOrOptional +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import ( + VectorSearchEndpointPermissionLevel, + VectorSearchEndpointPermissionLevelParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class VectorSearchEndpointPermission: """""" @@ -32,12 +28,11 @@ class VectorSearchEndpointPermission: user_name: VariableOrOptional[str] = None @classmethod - def from_dict(cls, value: 'VectorSearchEndpointPermissionDict') -> 'Self': + def from_dict(cls, value: "VectorSearchEndpointPermissionDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'VectorSearchEndpointPermissionDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "VectorSearchEndpointPermissionDict": + return _transform_to_json_value(self) # type:ignore class VectorSearchEndpointPermissionDict(TypedDict, total=False): @@ -52,4 +47,6 @@ class VectorSearchEndpointPermissionDict(TypedDict, total=False): user_name: VariableOrOptional[str] -VectorSearchEndpointPermissionParam = VectorSearchEndpointPermissionDict | VectorSearchEndpointPermission +VectorSearchEndpointPermissionParam = ( + VectorSearchEndpointPermissionDict | VectorSearchEndpointPermission +) diff --git a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission_level.py b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission_level.py index 406d260f4d7..d2618d39126 100644 --- a/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission_level.py +++ b/python/databricks/bundles/vector_search_endpoints/_models/vector_search_endpoint_permission_level.py @@ -1,26 +1,19 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint, VectorSearchEndpointDict, VectorSearchEndpointParam -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission, VectorSearchEndpointPermissionDict, VectorSearchEndpointPermissionParam -from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType, EndpointTypeParam +from typing import Literal class VectorSearchEndpointPermissionLevel(Enum): """ Permission level """ + CAN_CREATE = "CAN_CREATE" CAN_MANAGE = "CAN_MANAGE" CAN_USE = "CAN_USE" -VectorSearchEndpointPermissionLevelParam = Literal["CAN_CREATE", "CAN_MANAGE", "CAN_USE"] | VectorSearchEndpointPermissionLevel + +VectorSearchEndpointPermissionLevelParam = ( + Literal["CAN_CREATE", "CAN_MANAGE", "CAN_USE"] | VectorSearchEndpointPermissionLevel +) diff --git a/python/databricks/bundles/vector_search_indexes/__init__.py b/python/databricks/bundles/vector_search_indexes/__init__.py index 678b3508272..8e3c88eb018 100644 --- a/python/databricks/bundles/vector_search_indexes/__init__.py +++ b/python/databricks/bundles/vector_search_indexes/__init__.py @@ -1,50 +1,86 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'DeltaSyncVectorIndexSpecRequest', - 'DeltaSyncVectorIndexSpecRequestDict', - 'DeltaSyncVectorIndexSpecRequestParam', - 'DirectAccessVectorIndexSpec', - 'DirectAccessVectorIndexSpecDict', - 'DirectAccessVectorIndexSpecParam', - 'EmbeddingSourceColumn', - 'EmbeddingSourceColumnDict', - 'EmbeddingSourceColumnParam', - 'EmbeddingVectorColumn', - 'EmbeddingVectorColumnDict', - 'EmbeddingVectorColumnParam', - 'IndexSubtype', - 'IndexSubtypeParam', - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'PipelineType', - 'PipelineTypeParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'VectorIndexType', - 'VectorIndexTypeParam', - 'VectorSearchIndex', - 'VectorSearchIndexDict', - 'VectorSearchIndexParam', + "DeltaSyncVectorIndexSpecRequest", + "DeltaSyncVectorIndexSpecRequestDict", + "DeltaSyncVectorIndexSpecRequestParam", + "DirectAccessVectorIndexSpec", + "DirectAccessVectorIndexSpecDict", + "DirectAccessVectorIndexSpecParam", + "EmbeddingSourceColumn", + "EmbeddingSourceColumnDict", + "EmbeddingSourceColumnParam", + "EmbeddingVectorColumn", + "EmbeddingVectorColumnDict", + "EmbeddingVectorColumnParam", + "IndexSubtype", + "IndexSubtypeParam", + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "PipelineType", + "PipelineTypeParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", + "VectorIndexType", + "VectorIndexTypeParam", + "VectorSearchIndex", + "VectorSearchIndexDict", + "VectorSearchIndexParam", ] -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam - +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import ( + DeltaSyncVectorIndexSpecRequest, + DeltaSyncVectorIndexSpecRequestDict, + DeltaSyncVectorIndexSpecRequestParam, +) +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import ( + DirectAccessVectorIndexSpec, + DirectAccessVectorIndexSpecDict, + DirectAccessVectorIndexSpecParam, +) +from databricks.bundles.vector_search_indexes._models.embedding_source_column import ( + EmbeddingSourceColumn, + EmbeddingSourceColumnDict, + EmbeddingSourceColumnParam, +) +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import ( + EmbeddingVectorColumn, + EmbeddingVectorColumnDict, + EmbeddingVectorColumnParam, +) +from databricks.bundles.vector_search_indexes._models.index_subtype import ( + IndexSubtype, + IndexSubtypeParam, +) +from databricks.bundles.vector_search_indexes._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) +from databricks.bundles.vector_search_indexes._models.pipeline_type import ( + PipelineType, + PipelineTypeParam, +) +from databricks.bundles.vector_search_indexes._models.privilege import ( + Privilege, + PrivilegeParam, +) +from databricks.bundles.vector_search_indexes._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) +from databricks.bundles.vector_search_indexes._models.vector_index_type import ( + VectorIndexType, + VectorIndexTypeParam, +) +from databricks.bundles.vector_search_indexes._models.vector_search_index import ( + VectorSearchIndex, + VectorSearchIndexDict, + VectorSearchIndexParam, +) diff --git a/python/databricks/bundles/vector_search_indexes/_models/delta_sync_vector_index_spec_request.py b/python/databricks/bundles/vector_search_indexes/_models/delta_sync_vector_index_spec_request.py index ae2c5f6968d..b7af851711b 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/delta_sync_vector_index_spec_request.py +++ b/python/databricks/bundles/vector_search_indexes/_models/delta_sync_vector_index_spec_request.py @@ -1,32 +1,28 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.vector_search_indexes._models.embedding_source_column import ( + EmbeddingSourceColumn, + EmbeddingSourceColumnParam, +) +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import ( + EmbeddingVectorColumn, + EmbeddingVectorColumnParam, +) +from databricks.bundles.vector_search_indexes._models.pipeline_type import ( + PipelineType, + PipelineTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DeltaSyncVectorIndexSpecRequest: """""" @@ -46,12 +42,16 @@ class DeltaSyncVectorIndexSpecRequest: embedding vector column are always synced. """ - embedding_source_columns: VariableOrList[EmbeddingSourceColumn] = field(default_factory=list) + embedding_source_columns: VariableOrList[EmbeddingSourceColumn] = field( + default_factory=list + ) """ The columns that contain the embedding source. """ - embedding_vector_columns: VariableOrList[EmbeddingVectorColumn] = field(default_factory=list) + embedding_vector_columns: VariableOrList[EmbeddingVectorColumn] = field( + default_factory=list + ) """ The columns that contain the embedding vectors. """ @@ -74,12 +74,11 @@ class DeltaSyncVectorIndexSpecRequest: """ @classmethod - def from_dict(cls, value: 'DeltaSyncVectorIndexSpecRequestDict') -> 'Self': + def from_dict(cls, value: "DeltaSyncVectorIndexSpecRequestDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DeltaSyncVectorIndexSpecRequestDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DeltaSyncVectorIndexSpecRequestDict": + return _transform_to_json_value(self) # type:ignore class DeltaSyncVectorIndexSpecRequestDict(TypedDict, total=False): @@ -128,4 +127,6 @@ class DeltaSyncVectorIndexSpecRequestDict(TypedDict, total=False): """ -DeltaSyncVectorIndexSpecRequestParam = DeltaSyncVectorIndexSpecRequestDict | DeltaSyncVectorIndexSpecRequest +DeltaSyncVectorIndexSpecRequestParam = ( + DeltaSyncVectorIndexSpecRequestDict | DeltaSyncVectorIndexSpecRequest +) diff --git a/python/databricks/bundles/vector_search_indexes/_models/direct_access_vector_index_spec.py b/python/databricks/bundles/vector_search_indexes/_models/direct_access_vector_index_spec.py index 426995a3b09..76ef31b4b30 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/direct_access_vector_index_spec.py +++ b/python/databricks/bundles/vector_search_indexes/_models/direct_access_vector_index_spec.py @@ -1,42 +1,38 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.vector_search_indexes._models.embedding_source_column import ( + EmbeddingSourceColumn, + EmbeddingSourceColumnParam, +) +from databricks.bundles.vector_search_indexes._models.embedding_vector_column import ( + EmbeddingVectorColumn, + EmbeddingVectorColumnParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class DirectAccessVectorIndexSpec: """""" - embedding_source_columns: VariableOrList[EmbeddingSourceColumn] = field(default_factory=list) + embedding_source_columns: VariableOrList[EmbeddingSourceColumn] = field( + default_factory=list + ) """ The columns that contain the embedding source. The format should be array[double]. """ - embedding_vector_columns: VariableOrList[EmbeddingVectorColumn] = field(default_factory=list) + embedding_vector_columns: VariableOrList[EmbeddingVectorColumn] = field( + default_factory=list + ) """ The columns that contain the embedding vectors. The format should be array[double]. """ @@ -49,12 +45,11 @@ class DirectAccessVectorIndexSpec: """ @classmethod - def from_dict(cls, value: 'DirectAccessVectorIndexSpecDict') -> 'Self': + def from_dict(cls, value: "DirectAccessVectorIndexSpecDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'DirectAccessVectorIndexSpecDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "DirectAccessVectorIndexSpecDict": + return _transform_to_json_value(self) # type:ignore class DirectAccessVectorIndexSpecDict(TypedDict, total=False): @@ -78,4 +73,6 @@ class DirectAccessVectorIndexSpecDict(TypedDict, total=False): """ -DirectAccessVectorIndexSpecParam = DirectAccessVectorIndexSpecDict | DirectAccessVectorIndexSpec +DirectAccessVectorIndexSpecParam = ( + DirectAccessVectorIndexSpecDict | DirectAccessVectorIndexSpec +) diff --git a/python/databricks/bundles/vector_search_indexes/_models/embedding_source_column.py b/python/databricks/bundles/vector_search_indexes/_models/embedding_source_column.py index 26371674396..7debdc3f104 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/embedding_source_column.py +++ b/python/databricks/bundles/vector_search_indexes/_models/embedding_source_column.py @@ -1,32 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EmbeddingSourceColumn: """""" @@ -47,12 +31,11 @@ class EmbeddingSourceColumn: """ @classmethod - def from_dict(cls, value: 'EmbeddingSourceColumnDict') -> 'Self': + def from_dict(cls, value: "EmbeddingSourceColumnDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EmbeddingSourceColumnDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EmbeddingSourceColumnDict": + return _transform_to_json_value(self) # type:ignore class EmbeddingSourceColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_indexes/_models/embedding_vector_column.py b/python/databricks/bundles/vector_search_indexes/_models/embedding_vector_column.py index fd9076bc3ed..b8efd6b4f0c 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/embedding_vector_column.py +++ b/python/databricks/bundles/vector_search_indexes/_models/embedding_vector_column.py @@ -1,32 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class EmbeddingVectorColumn: """""" @@ -42,12 +26,11 @@ class EmbeddingVectorColumn: """ @classmethod - def from_dict(cls, value: 'EmbeddingVectorColumnDict') -> 'Self': + def from_dict(cls, value: "EmbeddingVectorColumnDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'EmbeddingVectorColumnDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "EmbeddingVectorColumnDict": + return _transform_to_json_value(self) # type:ignore class EmbeddingVectorColumnDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_indexes/_models/index_subtype.py b/python/databricks/bundles/vector_search_indexes/_models/index_subtype.py index 2dcf22c9d22..5485754d1e6 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/index_subtype.py +++ b/python/databricks/bundles/vector_search_indexes/_models/index_subtype.py @@ -1,26 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam +from typing import Literal class IndexSubtype(Enum): @@ -30,8 +11,10 @@ class IndexSubtype(Enum): - `FULL_TEXT`: An index that uses full-text search without vector embeddings. - `HYBRID`: An index that uses vector embeddings for similarity search and hybrid search. """ + VECTOR = "VECTOR" FULL_TEXT = "FULL_TEXT" HYBRID = "HYBRID" + IndexSubtypeParam = Literal["VECTOR", "FULL_TEXT", "HYBRID"] | IndexSubtype diff --git a/python/databricks/bundles/vector_search_indexes/_models/lifecycle.py b/python/databricks/bundles/vector_search_indexes/_models/lifecycle.py index b090b45839d..697776a198e 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/lifecycle.py +++ b/python/databricks/bundles/vector_search_indexes/_models/lifecycle.py @@ -1,32 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -37,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_indexes/_models/pipeline_type.py b/python/databricks/bundles/vector_search_indexes/_models/pipeline_type.py index 20898efdfb7..6821dec812e 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/pipeline_type.py +++ b/python/databricks/bundles/vector_search_indexes/_models/pipeline_type.py @@ -1,26 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam +from typing import Literal class PipelineType(Enum): @@ -29,7 +10,9 @@ class PipelineType(Enum): - `TRIGGERED`: If the pipeline uses the triggered execution mode, the system stops processing after successfully refreshing the source table in the pipeline once, ensuring the table is updated based on the data available when the update started. - `CONTINUOUS`: If the pipeline uses continuous execution, the pipeline processes new data as it arrives in the source table to keep vector index fresh. """ + TRIGGERED = "TRIGGERED" CONTINUOUS = "CONTINUOUS" + PipelineTypeParam = Literal["TRIGGERED", "CONTINUOUS"] | PipelineType diff --git a/python/databricks/bundles/vector_search_indexes/_models/privilege.py b/python/databricks/bundles/vector_search_indexes/_models/privilege.py index d6a59038647..df45211a76a 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/privilege.py +++ b/python/databricks/bundles/vector_search_indexes/_models/privilege.py @@ -1,26 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam +from typing import Literal class Privilege(Enum): @@ -77,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/vector_search_indexes/_models/privilege_assignment.py b/python/databricks/bundles/vector_search_indexes/_models/privilege_assignment.py index 08748413bfd..55100240cc4 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/privilege_assignment.py +++ b/python/databricks/bundles/vector_search_indexes/_models/privilege_assignment.py @@ -1,32 +1,20 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam - +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional +from databricks.bundles.vector_search_indexes._models.privilege import ( + Privilege, + PrivilegeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -43,12 +31,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/vector_search_indexes/_models/vector_index_type.py b/python/databricks/bundles/vector_search_indexes/_models/vector_index_type.py index 32d9790a05e..43b8550c8b2 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/vector_index_type.py +++ b/python/databricks/bundles/vector_search_indexes/_models/vector_index_type.py @@ -1,26 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex, VectorSearchIndexDict, VectorSearchIndexParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam +from typing import Literal class VectorIndexType(Enum): @@ -29,7 +10,9 @@ class VectorIndexType(Enum): - `DELTA_SYNC`: An index that automatically syncs with a source Delta Table, automatically and incrementally updating the index as the underlying data in the Delta Table changes. - `DIRECT_ACCESS`: An index that supports direct read and write of vectors and metadata through our REST and SDK APIs. With this model, the user manages index updates. """ + DELTA_SYNC = "DELTA_SYNC" DIRECT_ACCESS = "DIRECT_ACCESS" + VectorIndexTypeParam = Literal["DELTA_SYNC", "DIRECT_ACCESS"] | VectorIndexType diff --git a/python/databricks/bundles/vector_search_indexes/_models/vector_search_index.py b/python/databricks/bundles/vector_search_indexes/_models/vector_search_index.py index 1297e366dd2..dba21478b1f 100644 --- a/python/databricks/bundles/vector_search_indexes/_models/vector_search_index.py +++ b/python/databricks/bundles/vector_search_indexes/_models/vector_search_index.py @@ -1,32 +1,45 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest, DeltaSyncVectorIndexSpecRequestDict, DeltaSyncVectorIndexSpecRequestParam -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec, DirectAccessVectorIndexSpecDict, DirectAccessVectorIndexSpecParam -from databricks.bundles.vector_search_indexes._models.embedding_source_column import EmbeddingSourceColumn, EmbeddingSourceColumnDict, EmbeddingSourceColumnParam -from databricks.bundles.vector_search_indexes._models.embedding_vector_column import EmbeddingVectorColumn, EmbeddingVectorColumnDict, EmbeddingVectorColumnParam -from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.vector_search_indexes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.vector_search_indexes._models.index_subtype import IndexSubtype, IndexSubtypeParam - -from databricks.bundles.vector_search_indexes._models.pipeline_type import PipelineType, PipelineTypeParam - -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType, VectorIndexTypeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import ( + DeltaSyncVectorIndexSpecRequest, + DeltaSyncVectorIndexSpecRequestParam, +) +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import ( + DirectAccessVectorIndexSpec, + DirectAccessVectorIndexSpecParam, +) +from databricks.bundles.vector_search_indexes._models.index_subtype import ( + IndexSubtype, + IndexSubtypeParam, +) +from databricks.bundles.vector_search_indexes._models.lifecycle import ( + Lifecycle, + LifecycleParam, +) +from databricks.bundles.vector_search_indexes._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) +from databricks.bundles.vector_search_indexes._models.vector_index_type import ( + VectorIndexType, + VectorIndexTypeParam, +) if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class VectorSearchIndex(Resource): """""" @@ -81,12 +94,11 @@ class VectorSearchIndex(Resource): """ @classmethod - def from_dict(cls, value: 'VectorSearchIndexDict') -> 'Self': + def from_dict(cls, value: "VectorSearchIndexDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'VectorSearchIndexDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "VectorSearchIndexDict": + return _transform_to_json_value(self) # type:ignore class VectorSearchIndexDict(TypedDict, total=False): diff --git a/python/databricks/bundles/volumes/__init__.py b/python/databricks/bundles/volumes/__init__.py index 38bf7b6db23..97bc1d54bd7 100644 --- a/python/databricks/bundles/volumes/__init__.py +++ b/python/databricks/bundles/volumes/__init__.py @@ -1,35 +1,41 @@ # Code generated by pydabs-codegen. DO NOT EDIT. __all__ = [ - 'Lifecycle', - 'LifecycleDict', - 'LifecycleParam', - 'Privilege', - 'PrivilegeAssignment', - 'PrivilegeAssignmentDict', - 'PrivilegeAssignmentParam', - 'PrivilegeParam', - 'Volume', - 'VolumeDict', - 'VolumeParam', - 'VolumeType', - 'VolumeTypeParam', - 'VolumeGrant', - 'VolumeGrantDict', - 'VolumeGrantParam', - 'VolumeGrantPrivilege', - 'VolumeGrantPrivilegeParam', + "Lifecycle", + "LifecycleDict", + "LifecycleParam", + "Privilege", + "PrivilegeAssignment", + "PrivilegeAssignmentDict", + "PrivilegeAssignmentParam", + "PrivilegeParam", + "Volume", + "VolumeDict", + "VolumeGrant", + "VolumeGrantDict", + "VolumeGrantParam", + "VolumeGrantPrivilege", + "VolumeGrantPrivilegeParam", + "VolumeParam", + "VolumeType", + "VolumeTypeParam", ] -from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam +from databricks.bundles.volumes._models.lifecycle import ( + Lifecycle, + LifecycleDict, + LifecycleParam, +) from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.volumes._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentDict, + PrivilegeAssignmentParam, +) +from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam - VolumeGrant = PrivilegeAssignment VolumeGrantDict = PrivilegeAssignmentDict VolumeGrantParam = PrivilegeAssignmentParam diff --git a/python/databricks/bundles/volumes/_models/lifecycle.py b/python/databricks/bundles/volumes/_models/lifecycle.py index 1a826fe2134..697776a198e 100644 --- a/python/databricks/bundles/volumes/_models/lifecycle.py +++ b/python/databricks/bundles/volumes/_models/lifecycle.py @@ -1,24 +1,16 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam -from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam - -from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam - +from databricks.bundles.core._variable import VariableOrOptional if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Lifecycle: """""" @@ -29,12 +21,11 @@ class Lifecycle: """ @classmethod - def from_dict(cls, value: 'LifecycleDict') -> 'Self': + def from_dict(cls, value: "LifecycleDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'LifecycleDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "LifecycleDict": + return _transform_to_json_value(self) # type:ignore class LifecycleDict(TypedDict, total=False): diff --git a/python/databricks/bundles/volumes/_models/privilege.py b/python/databricks/bundles/volumes/_models/privilege.py index 88dbf104c93..df45211a76a 100644 --- a/python/databricks/bundles/volumes/_models/privilege.py +++ b/python/databricks/bundles/volumes/_models/privilege.py @@ -1,18 +1,7 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam -from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam +from typing import Literal class Privilege(Enum): @@ -69,4 +58,61 @@ class Privilege(Enum): READ_METADATA = "READ_METADATA" EXTERNAL_USE_LOCATION = "EXTERNAL_USE_LOCATION" -PrivilegeParam = Literal["SELECT", "READ_PRIVATE_FILES", "WRITE_PRIVATE_FILES", "CREATE", "USAGE", "USE_CATALOG", "USE_SCHEMA", "CREATE_SCHEMA", "CREATE_VIEW", "CREATE_EXTERNAL_TABLE", "CREATE_MATERIALIZED_VIEW", "CREATE_FUNCTION", "CREATE_MODEL", "CREATE_CATALOG", "CREATE_MANAGED_STORAGE", "CREATE_EXTERNAL_LOCATION", "CREATE_STORAGE_CREDENTIAL", "CREATE_SERVICE_CREDENTIAL", "ACCESS", "CREATE_SHARE", "CREATE_RECIPIENT", "CREATE_PROVIDER", "USE_SHARE", "USE_RECIPIENT", "USE_PROVIDER", "USE_MARKETPLACE_ASSETS", "SET_SHARE_PERMISSION", "MODIFY", "REFRESH", "EXECUTE", "READ_FILES", "WRITE_FILES", "CREATE_TABLE", "ALL_PRIVILEGES", "CREATE_CONNECTION", "USE_CONNECTION", "APPLY_TAG", "CREATE_FOREIGN_CATALOG", "CREATE_FOREIGN_SECURABLE", "MANAGE_ALLOWLIST", "CREATE_VOLUME", "CREATE_EXTERNAL_VOLUME", "READ_VOLUME", "WRITE_VOLUME", "MANAGE", "BROWSE", "CREATE_CLEAN_ROOM", "MODIFY_CLEAN_ROOM", "EXECUTE_CLEAN_ROOM_TASK", "EXTERNAL_USE_SCHEMA", "READ_METADATA", "EXTERNAL_USE_LOCATION"] | Privilege + +PrivilegeParam = ( + Literal[ + "SELECT", + "READ_PRIVATE_FILES", + "WRITE_PRIVATE_FILES", + "CREATE", + "USAGE", + "USE_CATALOG", + "USE_SCHEMA", + "CREATE_SCHEMA", + "CREATE_VIEW", + "CREATE_EXTERNAL_TABLE", + "CREATE_MATERIALIZED_VIEW", + "CREATE_FUNCTION", + "CREATE_MODEL", + "CREATE_CATALOG", + "CREATE_MANAGED_STORAGE", + "CREATE_EXTERNAL_LOCATION", + "CREATE_STORAGE_CREDENTIAL", + "CREATE_SERVICE_CREDENTIAL", + "ACCESS", + "CREATE_SHARE", + "CREATE_RECIPIENT", + "CREATE_PROVIDER", + "USE_SHARE", + "USE_RECIPIENT", + "USE_PROVIDER", + "USE_MARKETPLACE_ASSETS", + "SET_SHARE_PERMISSION", + "MODIFY", + "REFRESH", + "EXECUTE", + "READ_FILES", + "WRITE_FILES", + "CREATE_TABLE", + "ALL_PRIVILEGES", + "CREATE_CONNECTION", + "USE_CONNECTION", + "APPLY_TAG", + "CREATE_FOREIGN_CATALOG", + "CREATE_FOREIGN_SECURABLE", + "MANAGE_ALLOWLIST", + "CREATE_VOLUME", + "CREATE_EXTERNAL_VOLUME", + "READ_VOLUME", + "WRITE_VOLUME", + "MANAGE", + "BROWSE", + "CREATE_CLEAN_ROOM", + "MODIFY_CLEAN_ROOM", + "EXECUTE_CLEAN_ROOM_TASK", + "EXTERNAL_USE_SCHEMA", + "READ_METADATA", + "EXTERNAL_USE_LOCATION", + ] + | Privilege +) diff --git a/python/databricks/bundles/volumes/_models/privilege_assignment.py b/python/databricks/bundles/volumes/_models/privilege_assignment.py index b09f932ef76..612e9d85eb0 100644 --- a/python/databricks/bundles/volumes/_models/privilege_assignment.py +++ b/python/databricks/bundles/volumes/_models/privilege_assignment.py @@ -1,24 +1,17 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict -from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam +from databricks.bundles.core._variable import VariableOrList, VariableOrOptional from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam -from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam - - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class PrivilegeAssignment: """""" @@ -35,12 +28,11 @@ class PrivilegeAssignment: """ @classmethod - def from_dict(cls, value: 'PrivilegeAssignmentDict') -> 'Self': + def from_dict(cls, value: "PrivilegeAssignmentDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'PrivilegeAssignmentDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "PrivilegeAssignmentDict": + return _transform_to_json_value(self) # type:ignore class PrivilegeAssignmentDict(TypedDict, total=False): diff --git a/python/databricks/bundles/volumes/_models/volume.py b/python/databricks/bundles/volumes/_models/volume.py index 79bed352640..db248fbba94 100644 --- a/python/databricks/bundles/volumes/_models/volume.py +++ b/python/databricks/bundles/volumes/_models/volume.py @@ -1,24 +1,27 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING -from enum import Enum -from dataclasses import dataclass, replace, field +from dataclasses import dataclass, field +from typing import TYPE_CHECKING, TypedDict from databricks.bundles.core._resource import Resource from databricks.bundles.core._transform import _transform from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam - +from databricks.bundles.core._variable import ( + VariableOr, + VariableOrList, + VariableOrOptional, +) +from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleParam +from databricks.bundles.volumes._models.privilege_assignment import ( + PrivilegeAssignment, + PrivilegeAssignmentParam, +) from databricks.bundles.volumes._models.volume_type import VolumeType, VolumeTypeParam - if TYPE_CHECKING: from typing_extensions import Self + @dataclass(kw_only=True) class Volume(Resource): """""" @@ -66,12 +69,11 @@ class Volume(Resource): """ @classmethod - def from_dict(cls, value: 'VolumeDict') -> 'Self': + def from_dict(cls, value: "VolumeDict") -> "Self": return _transform(cls, value) - def as_dict(self) -> 'VolumeDict': - return _transform_to_json_value(self) # type:ignore - + def as_dict(self) -> "VolumeDict": + return _transform_to_json_value(self) # type:ignore class VolumeDict(TypedDict, total=False): diff --git a/python/databricks/bundles/volumes/_models/volume_type.py b/python/databricks/bundles/volumes/_models/volume_type.py index 42574bde7b2..0caf77b1c33 100644 --- a/python/databricks/bundles/volumes/_models/volume_type.py +++ b/python/databricks/bundles/volumes/_models/volume_type.py @@ -1,22 +1,12 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from typing import Any, Literal, Optional, TypedDict, ClassVar, TYPE_CHECKING from enum import Enum -from dataclasses import dataclass, replace, field - -from databricks.bundles.core._resource import Resource -from databricks.bundles.core._transform import _transform -from databricks.bundles.core._transform_to_json import _transform_to_json_value -from databricks.bundles.core._variable import VariableOr, VariableOrOptional, VariableOrList, VariableOrDict - -from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment, PrivilegeAssignmentDict, PrivilegeAssignmentParam -from databricks.bundles.volumes._models.lifecycle import Lifecycle, LifecycleDict, LifecycleParam -from databricks.bundles.volumes._models.volume import Volume, VolumeDict, VolumeParam -from databricks.bundles.volumes._models.privilege import Privilege, PrivilegeParam +from typing import Literal class VolumeType(Enum): MANAGED = "MANAGED" EXTERNAL = "EXTERNAL" + VolumeTypeParam = Literal["MANAGED", "EXTERNAL"] | VolumeType diff --git a/python/databricks_tests/core/_generated/alerts.py b/python/databricks_tests/core/_generated/alerts.py index a0ef68845ad..2d8e117c99c 100644 --- a/python/databricks_tests/core/_generated/alerts.py +++ b/python/databricks_tests/core/_generated/alerts.py @@ -1,25 +1,57 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from databricks.bundles.core import Resources, alert_mutator -from databricks.bundles.core._generated.alerts import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.alerts._models.alert import Alert from databricks.bundles.alerts._models.alert_v2_evaluation import AlertV2Evaluation -from databricks.bundles.alerts._models.alert_v2_operand_column import AlertV2OperandColumn +from databricks.bundles.alerts._models.alert_v2_operand_column import ( + AlertV2OperandColumn, +) from databricks.bundles.alerts._models.alert_v2_run_as import AlertV2RunAs from databricks.bundles.alerts._models.comparison_operator import ComparisonOperator from databricks.bundles.alerts._models.cron_schedule import CronSchedule from databricks.bundles.alerts._models.lifecycle import Lifecycle from databricks.bundles.alerts._models.permission import Permission from databricks.bundles.alerts._models.permission_level import PermissionLevel +from databricks.bundles.core import Resources, alert_mutator +from databricks.bundles.core._generated.alerts import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_alert, - dict_example={"display_name": "display_name", "evaluation": {"comparison_operator": "LESS_THAN", "source": {"name": "name"}}, "lifecycle": {}, "permissions": [{"level": "CAN_MANAGE"}], "query_text": "query_text", "run_as": {}, "schedule": {"quartz_cron_schedule": "quartz_cron_schedule", "timezone_id": "timezone_id"}, "warehouse_id": "warehouse_id"}, - dataclass_example=Alert(display_name="display_name", evaluation=AlertV2Evaluation(comparison_operator=ComparisonOperator.LESS_THAN, source=AlertV2OperandColumn(name="name")), lifecycle=Lifecycle(), permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], query_text="query_text", run_as=AlertV2RunAs(), schedule=CronSchedule(quartz_cron_schedule="quartz_cron_schedule", timezone_id="timezone_id"), warehouse_id="warehouse_id"), + dict_example={ + "display_name": "display_name", + "evaluation": { + "comparison_operator": "LESS_THAN", + "source": {"name": "name"}, + }, + "lifecycle": {}, + "permissions": [{"level": "CAN_MANAGE"}], + "query_text": "query_text", + "run_as": {}, + "schedule": { + "quartz_cron_schedule": "quartz_cron_schedule", + "timezone_id": "timezone_id", + }, + "warehouse_id": "warehouse_id", + }, + dataclass_example=Alert( + display_name="display_name", + evaluation=AlertV2Evaluation( + comparison_operator=ComparisonOperator.LESS_THAN, + source=AlertV2OperandColumn(name="name"), + ), + lifecycle=Lifecycle(), + permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], + query_text="query_text", + run_as=AlertV2RunAs(), + schedule=CronSchedule( + quartz_cron_schedule="quartz_cron_schedule", + timezone_id="timezone_id", + ), + warehouse_id="warehouse_id", + ), mutator=alert_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/apps.py b/python/databricks_tests/core/_generated/apps.py index 79635665ea8..41518c8c2e5 100644 --- a/python/databricks_tests/core/_generated/apps.py +++ b/python/databricks_tests/core/_generated/apps.py @@ -1,8 +1,5 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from databricks.bundles.core import Resources, app_mutator -from databricks.bundles.core._generated.apps import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.apps._models.app import App from databricks.bundles.apps._models.app_config import AppConfig from databricks.bundles.apps._models.app_permission import AppPermission @@ -12,15 +9,42 @@ from databricks.bundles.apps._models.git_repository import GitRepository from databricks.bundles.apps._models.git_source import GitSource from databricks.bundles.apps._models.lifecycle_with_started import LifecycleWithStarted -from databricks.bundles.apps._models.telemetry_export_destination import TelemetryExportDestination +from databricks.bundles.apps._models.telemetry_export_destination import ( + TelemetryExportDestination, +) +from databricks.bundles.core import Resources, app_mutator +from databricks.bundles.core._generated.apps import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_app, - dict_example={"compute_size": "MEDIUM", "config": {}, "git_repository": {"provider": "provider", "url": "url"}, "git_source": {}, "lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_MANAGE"}], "resources": [{"name": "name"}], "telemetry_export_destinations": [{}], "user_api_scopes": ["user_api_scopes"]}, - dataclass_example=App(compute_size=ComputeSize.MEDIUM, config=AppConfig(), git_repository=GitRepository(provider="provider", url="url"), git_source=GitSource(), lifecycle=LifecycleWithStarted(), name="name", permissions=[AppPermission(level=AppPermissionLevel.CAN_MANAGE)], resources=[AppResource(name="name")], telemetry_export_destinations=[TelemetryExportDestination()], user_api_scopes=["user_api_scopes"]), + dict_example={ + "compute_size": "MEDIUM", + "config": {}, + "git_repository": {"provider": "provider", "url": "url"}, + "git_source": {}, + "lifecycle": {}, + "name": "name", + "permissions": [{"level": "CAN_MANAGE"}], + "resources": [{"name": "name"}], + "telemetry_export_destinations": [{}], + "user_api_scopes": ["user_api_scopes"], + }, + dataclass_example=App( + compute_size=ComputeSize.MEDIUM, + config=AppConfig(), + git_repository=GitRepository(provider="provider", url="url"), + git_source=GitSource(), + lifecycle=LifecycleWithStarted(), + name="name", + permissions=[AppPermission(level=AppPermissionLevel.CAN_MANAGE)], + resources=[AppResource(name="name")], + telemetry_export_destinations=[TelemetryExportDestination()], + user_api_scopes=["user_api_scopes"], + ), mutator=app_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/catalogs.py b/python/databricks_tests/core/_generated/catalogs.py index 60e60d6790d..117d964071c 100644 --- a/python/databricks_tests/core/_generated/catalogs.py +++ b/python/databricks_tests/core/_generated/catalogs.py @@ -1,20 +1,34 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from databricks.bundles.core import Resources, catalog_mutator -from databricks.bundles.core._generated.catalogs import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.catalogs._models.catalog import Catalog from databricks.bundles.catalogs._models.encryption_settings import EncryptionSettings from databricks.bundles.catalogs._models.lifecycle import Lifecycle from databricks.bundles.catalogs._models.privilege_assignment import PrivilegeAssignment +from databricks.bundles.core import Resources, catalog_mutator +from databricks.bundles.core._generated.catalogs import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_catalog, - dict_example={"grants": [{}], "lifecycle": {}, "managed_encryption_settings": {}, "name": "name", "options": {"key": "value"}, "properties": {"key": "value"}}, - dataclass_example=Catalog(grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), managed_encryption_settings=EncryptionSettings(), name="name", options={"key": "value"}, properties={"key": "value"}), + dict_example={ + "grants": [{}], + "lifecycle": {}, + "managed_encryption_settings": {}, + "name": "name", + "options": {"key": "value"}, + "properties": {"key": "value"}, + }, + dataclass_example=Catalog( + grants=[PrivilegeAssignment()], + lifecycle=Lifecycle(), + managed_encryption_settings=EncryptionSettings(), + name="name", + options={"key": "value"}, + properties={"key": "value"}, + ), mutator=catalog_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/cluster_policies.py b/python/databricks_tests/core/_generated/cluster_policies.py index 9892dbdcd3a..72fd868be8a 100644 --- a/python/databricks_tests/core/_generated/cluster_policies.py +++ b/python/databricks_tests/core/_generated/cluster_policies.py @@ -1,21 +1,35 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from databricks.bundles.core import Resources, cluster_policy_mutator -from databricks.bundles.core._generated.cluster_policies import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.cluster_policies._models.cluster_policy import ClusterPolicy -from databricks.bundles.cluster_policies._models.cluster_policy_permission import ClusterPolicyPermission -from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ClusterPolicyPermissionLevel +from databricks.bundles.cluster_policies._models.cluster_policy_permission import ( + ClusterPolicyPermission, +) +from databricks.bundles.cluster_policies._models.cluster_policy_permission_level import ( + ClusterPolicyPermissionLevel, +) from databricks.bundles.cluster_policies._models.library import Library from databricks.bundles.cluster_policies._models.lifecycle import Lifecycle +from databricks.bundles.core import Resources, cluster_policy_mutator +from databricks.bundles.core._generated.cluster_policies import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_cluster_policy, - dict_example={"libraries": [{}], "lifecycle": {}, "permissions": [{"level": "CAN_USE"}]}, - dataclass_example=ClusterPolicy(libraries=[Library()], lifecycle=Lifecycle(), permissions=[ClusterPolicyPermission(level=ClusterPolicyPermissionLevel.CAN_USE)]), + dict_example={ + "libraries": [{}], + "lifecycle": {}, + "permissions": [{"level": "CAN_USE"}], + }, + dataclass_example=ClusterPolicy( + libraries=[Library()], + lifecycle=Lifecycle(), + permissions=[ + ClusterPolicyPermission(level=ClusterPolicyPermissionLevel.CAN_USE) + ], + ), mutator=cluster_policy_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/clusters.py b/python/databricks_tests/core/_generated/clusters.py index a3247c7e6b4..2de29367b86 100644 --- a/python/databricks_tests/core/_generated/clusters.py +++ b/python/databricks_tests/core/_generated/clusters.py @@ -1,8 +1,5 @@ # Code generated by pydabs-codegen. DO NOT EDIT. -from databricks.bundles.core import Resources, cluster_mutator -from databricks.bundles.core._generated.clusters import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.clusters._models.auto_scale import AutoScale from databricks.bundles.clusters._models.aws_attributes import AwsAttributes from databricks.bundles.clusters._models.azure_attributes import AzureAttributes @@ -10,25 +7,78 @@ from databricks.bundles.clusters._models.cluster import Cluster from databricks.bundles.clusters._models.cluster_log_conf import ClusterLogConf from databricks.bundles.clusters._models.cluster_permission import ClusterPermission -from databricks.bundles.clusters._models.cluster_permission_level import ClusterPermissionLevel +from databricks.bundles.clusters._models.cluster_permission_level import ( + ClusterPermissionLevel, +) from databricks.bundles.clusters._models.data_security_mode import DataSecurityMode from databricks.bundles.clusters._models.docker_image import DockerImage from databricks.bundles.clusters._models.gcp_attributes import GcpAttributes from databricks.bundles.clusters._models.init_script_info import InitScriptInfo from databricks.bundles.clusters._models.kind import Kind from databricks.bundles.clusters._models.library import Library -from databricks.bundles.clusters._models.lifecycle_with_started import LifecycleWithStarted -from databricks.bundles.clusters._models.node_type_flexibility import NodeTypeFlexibility +from databricks.bundles.clusters._models.lifecycle_with_started import ( + LifecycleWithStarted, +) +from databricks.bundles.clusters._models.node_type_flexibility import ( + NodeTypeFlexibility, +) from databricks.bundles.clusters._models.runtime_engine import RuntimeEngine from databricks.bundles.clusters._models.workload_type import WorkloadType +from databricks.bundles.core import Resources, cluster_mutator +from databricks.bundles.core._generated.clusters import _resource_type +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_cluster, - dict_example={"autoscale": {}, "aws_attributes": {}, "azure_attributes": {}, "cluster_log_conf": {}, "custom_tags": {"key": "value"}, "data_security_mode": "NONE", "docker_image": {}, "driver_node_type_flexibility": {}, "gcp_attributes": {}, "init_scripts": [{}], "kind": "CLASSIC_PREVIEW", "libraries": [{}], "lifecycle": {}, "permissions": [{"level": "CAN_MANAGE"}], "runtime_engine": "NULL", "spark_conf": {"key": "value"}, "spark_env_vars": {"key": "value"}, "ssh_public_keys": ["ssh_public_keys"], "worker_node_type_flexibility": {}, "workload_type": {"clients": {}}}, - dataclass_example=Cluster(autoscale=AutoScale(), aws_attributes=AwsAttributes(), azure_attributes=AzureAttributes(), cluster_log_conf=ClusterLogConf(), custom_tags={"key": "value"}, data_security_mode=DataSecurityMode.NONE, docker_image=DockerImage(), driver_node_type_flexibility=NodeTypeFlexibility(), gcp_attributes=GcpAttributes(), init_scripts=[InitScriptInfo()], kind=Kind.CLASSIC_PREVIEW, libraries=[Library()], lifecycle=LifecycleWithStarted(), permissions=[ClusterPermission(level=ClusterPermissionLevel.CAN_MANAGE)], runtime_engine=RuntimeEngine.NULL, spark_conf={"key": "value"}, spark_env_vars={"key": "value"}, ssh_public_keys=["ssh_public_keys"], worker_node_type_flexibility=NodeTypeFlexibility(), workload_type=WorkloadType(clients=ClientsTypes())), + dict_example={ + "autoscale": {}, + "aws_attributes": {}, + "azure_attributes": {}, + "cluster_log_conf": {}, + "custom_tags": {"key": "value"}, + "data_security_mode": "NONE", + "docker_image": {}, + "driver_node_type_flexibility": {}, + "gcp_attributes": {}, + "init_scripts": [{}], + "kind": "CLASSIC_PREVIEW", + "libraries": [{}], + "lifecycle": {}, + "permissions": [{"level": "CAN_MANAGE"}], + "runtime_engine": "NULL", + "spark_conf": {"key": "value"}, + "spark_env_vars": {"key": "value"}, + "ssh_public_keys": ["ssh_public_keys"], + "worker_node_type_flexibility": {}, + "workload_type": {"clients": {}}, + }, + dataclass_example=Cluster( + autoscale=AutoScale(), + aws_attributes=AwsAttributes(), + azure_attributes=AzureAttributes(), + cluster_log_conf=ClusterLogConf(), + custom_tags={"key": "value"}, + data_security_mode=DataSecurityMode.NONE, + docker_image=DockerImage(), + driver_node_type_flexibility=NodeTypeFlexibility(), + gcp_attributes=GcpAttributes(), + init_scripts=[InitScriptInfo()], + kind=Kind.CLASSIC_PREVIEW, + libraries=[Library()], + lifecycle=LifecycleWithStarted(), + permissions=[ + ClusterPermission(level=ClusterPermissionLevel.CAN_MANAGE) + ], + runtime_engine=RuntimeEngine.NULL, + spark_conf={"key": "value"}, + spark_env_vars={"key": "value"}, + ssh_public_keys=["ssh_public_keys"], + worker_node_type_flexibility=NodeTypeFlexibility(), + workload_type=WorkloadType(clients=ClientsTypes()), + ), mutator=cluster_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/dashboards.py b/python/databricks_tests/core/_generated/dashboards.py index 49a1458dc93..3ec459fb4ba 100644 --- a/python/databricks_tests/core/_generated/dashboards.py +++ b/python/databricks_tests/core/_generated/dashboards.py @@ -2,20 +2,28 @@ from databricks.bundles.core import Resources, dashboard_mutator from databricks.bundles.core._generated.dashboards import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.dashboards._models.dashboard import Dashboard from databricks.bundles.dashboards._models.lifecycle import Lifecycle from databricks.bundles.dashboards._models.lifecycle_state import LifecycleState from databricks.bundles.dashboards._models.permission import Permission from databricks.bundles.dashboards._models.permission_level import PermissionLevel +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_dashboard, - dict_example={"lifecycle": {}, "lifecycle_state": "ACTIVE", "permissions": [{"level": "CAN_MANAGE"}]}, - dataclass_example=Dashboard(lifecycle=Lifecycle(), lifecycle_state=LifecycleState.ACTIVE, permissions=[Permission(level=PermissionLevel.CAN_MANAGE)]), + dict_example={ + "lifecycle": {}, + "lifecycle_state": "ACTIVE", + "permissions": [{"level": "CAN_MANAGE"}], + }, + dataclass_example=Dashboard( + lifecycle=Lifecycle(), + lifecycle_state=LifecycleState.ACTIVE, + permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], + ), mutator=dashboard_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/database_catalogs.py b/python/databricks_tests/core/_generated/database_catalogs.py index c8ae49535fc..43919e9c7d9 100644 --- a/python/databricks_tests/core/_generated/database_catalogs.py +++ b/python/databricks_tests/core/_generated/database_catalogs.py @@ -2,17 +2,29 @@ from databricks.bundles.core import Resources, database_catalog_mutator from databricks.bundles.core._generated.database_catalogs import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase -from databricks.bundles.database_catalogs._models.database_catalog import DatabaseCatalog +from databricks.bundles.database_catalogs._models.database_catalog import ( + DatabaseCatalog, +) from databricks.bundles.database_catalogs._models.lifecycle import Lifecycle +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_database_catalog, - dict_example={"database_instance_name": "database_instance_name", "database_name": "database_name", "lifecycle": {}, "name": "name"}, - dataclass_example=DatabaseCatalog(database_instance_name="database_instance_name", database_name="database_name", lifecycle=Lifecycle(), name="name"), + dict_example={ + "database_instance_name": "database_instance_name", + "database_name": "database_name", + "lifecycle": {}, + "name": "name", + }, + dataclass_example=DatabaseCatalog( + database_instance_name="database_instance_name", + database_name="database_name", + lifecycle=Lifecycle(), + name="name", + ), mutator=database_catalog_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/database_instances.py b/python/databricks_tests/core/_generated/database_instances.py index bfbdd2fc10c..9fb8c2d6962 100644 --- a/python/databricks_tests/core/_generated/database_instances.py +++ b/python/databricks_tests/core/_generated/database_instances.py @@ -2,20 +2,36 @@ from databricks.bundles.core import Resources, database_instance_mutator from databricks.bundles.core._generated.database_instances import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase -from databricks.bundles.database_instances._models.database_instance import DatabaseInstance -from databricks.bundles.database_instances._models.database_instance_ref import DatabaseInstanceRef +from databricks.bundles.database_instances._models.database_instance import ( + DatabaseInstance, +) +from databricks.bundles.database_instances._models.database_instance_ref import ( + DatabaseInstanceRef, +) from databricks.bundles.database_instances._models.lifecycle import Lifecycle from databricks.bundles.database_instances._models.permission import Permission -from databricks.bundles.database_instances._models.permission_level import PermissionLevel +from databricks.bundles.database_instances._models.permission_level import ( + PermissionLevel, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_database_instance, - dict_example={"lifecycle": {}, "name": "name", "parent_instance_ref": {}, "permissions": [{"level": "CAN_MANAGE"}]}, - dataclass_example=DatabaseInstance(lifecycle=Lifecycle(), name="name", parent_instance_ref=DatabaseInstanceRef(), permissions=[Permission(level=PermissionLevel.CAN_MANAGE)]), + dict_example={ + "lifecycle": {}, + "name": "name", + "parent_instance_ref": {}, + "permissions": [{"level": "CAN_MANAGE"}], + }, + dataclass_example=DatabaseInstance( + lifecycle=Lifecycle(), + name="name", + parent_instance_ref=DatabaseInstanceRef(), + permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], + ), mutator=database_instance_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/experiments.py b/python/databricks_tests/core/_generated/experiments.py index b3fa64f5ba7..229dec55646 100644 --- a/python/databricks_tests/core/_generated/experiments.py +++ b/python/databricks_tests/core/_generated/experiments.py @@ -2,20 +2,38 @@ from databricks.bundles.core import Resources, mlflow_experiment_mutator from databricks.bundles.core._generated.experiments import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase -from databricks.bundles.experiments._models.experiment_permission_level import ExperimentPermissionLevel +from databricks.bundles.experiments._models.experiment_permission_level import ( + ExperimentPermissionLevel, +) from databricks.bundles.experiments._models.experiment_tag import ExperimentTag from databricks.bundles.experiments._models.lifecycle import Lifecycle from databricks.bundles.experiments._models.mlflow_experiment import MlflowExperiment -from databricks.bundles.experiments._models.mlflow_experiment_permission import MlflowExperimentPermission +from databricks.bundles.experiments._models.mlflow_experiment_permission import ( + MlflowExperimentPermission, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_mlflow_experiment, - dict_example={"lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_MANAGE"}], "tags": [{}]}, - dataclass_example=MlflowExperiment(lifecycle=Lifecycle(), name="name", permissions=[MlflowExperimentPermission(level=ExperimentPermissionLevel.CAN_MANAGE)], tags=[ExperimentTag()]), + dict_example={ + "lifecycle": {}, + "name": "name", + "permissions": [{"level": "CAN_MANAGE"}], + "tags": [{}], + }, + dataclass_example=MlflowExperiment( + lifecycle=Lifecycle(), + name="name", + permissions=[ + MlflowExperimentPermission( + level=ExperimentPermissionLevel.CAN_MANAGE + ) + ], + tags=[ExperimentTag()], + ), mutator=mlflow_experiment_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/external_locations.py b/python/databricks_tests/core/_generated/external_locations.py index 37b7c8ba88c..d2bb07e0644 100644 --- a/python/databricks_tests/core/_generated/external_locations.py +++ b/python/databricks_tests/core/_generated/external_locations.py @@ -2,20 +2,44 @@ from databricks.bundles.core import Resources, external_location_mutator from databricks.bundles.core._generated.external_locations import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase -from databricks.bundles.external_locations._models.encryption_details import EncryptionDetails -from databricks.bundles.external_locations._models.external_location import ExternalLocation -from databricks.bundles.external_locations._models.file_event_queue import FileEventQueue +from databricks.bundles.external_locations._models.encryption_details import ( + EncryptionDetails, +) +from databricks.bundles.external_locations._models.external_location import ( + ExternalLocation, +) +from databricks.bundles.external_locations._models.file_event_queue import ( + FileEventQueue, +) from databricks.bundles.external_locations._models.lifecycle import Lifecycle -from databricks.bundles.external_locations._models.privilege_assignment import PrivilegeAssignment +from databricks.bundles.external_locations._models.privilege_assignment import ( + PrivilegeAssignment, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_external_location, - dict_example={"credential_name": "credential_name", "encryption_details": {}, "file_event_queue": {}, "grants": [{}], "lifecycle": {}, "name": "name", "url": "url"}, - dataclass_example=ExternalLocation(credential_name="credential_name", encryption_details=EncryptionDetails(), file_event_queue=FileEventQueue(), grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), name="name", url="url"), + dict_example={ + "credential_name": "credential_name", + "encryption_details": {}, + "file_event_queue": {}, + "grants": [{}], + "lifecycle": {}, + "name": "name", + "url": "url", + }, + dataclass_example=ExternalLocation( + credential_name="credential_name", + encryption_details=EncryptionDetails(), + file_event_queue=FileEventQueue(), + grants=[PrivilegeAssignment()], + lifecycle=Lifecycle(), + name="name", + url="url", + ), mutator=external_location_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/features.py b/python/databricks_tests/core/_generated/features.py index 60e2b777025..19344ab0e56 100644 --- a/python/databricks_tests/core/_generated/features.py +++ b/python/databricks_tests/core/_generated/features.py @@ -2,21 +2,35 @@ from databricks.bundles.core import Resources, feature_mutator from databricks.bundles.core._generated.features import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.features._models.data_source import DataSource from databricks.bundles.features._models.entity_column import EntityColumn from databricks.bundles.features._models.feature import Feature from databricks.bundles.features._models.function import Function from databricks.bundles.features._models.lifecycle import Lifecycle from databricks.bundles.features._models.timeseries_column import TimeseriesColumn +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_feature, - dict_example={"entities": [{"name": "name"}], "full_name": "full_name", "function": {}, "lifecycle": {}, "source": {}, "timeseries_column": {"name": "name"}}, - dataclass_example=Feature(entities=[EntityColumn(name="name")], full_name="full_name", function=Function(), lifecycle=Lifecycle(), source=DataSource(), timeseries_column=TimeseriesColumn(name="name")), + dict_example={ + "entities": [{"name": "name"}], + "full_name": "full_name", + "function": {}, + "lifecycle": {}, + "source": {}, + "timeseries_column": {"name": "name"}, + }, + dataclass_example=Feature( + entities=[EntityColumn(name="name")], + full_name="full_name", + function=Function(), + lifecycle=Lifecycle(), + source=DataSource(), + timeseries_column=TimeseriesColumn(name="name"), + ), mutator=feature_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/genie_spaces.py b/python/databricks_tests/core/_generated/genie_spaces.py index c2c35def27a..c219a22cda4 100644 --- a/python/databricks_tests/core/_generated/genie_spaces.py +++ b/python/databricks_tests/core/_generated/genie_spaces.py @@ -2,11 +2,11 @@ from databricks.bundles.core import Resources, genie_space_mutator from databricks.bundles.core._generated.genie_spaces import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.genie_spaces._models.genie_space import GenieSpace from databricks.bundles.genie_spaces._models.lifecycle import Lifecycle from databricks.bundles.genie_spaces._models.permission import Permission from databricks.bundles.genie_spaces._models.permission_level import PermissionLevel +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): @@ -14,7 +14,10 @@ def _test_case(): ResourceTestCase( add_resource=Resources.add_genie_space, dict_example={"lifecycle": {}, "permissions": [{"level": "CAN_MANAGE"}]}, - dataclass_example=GenieSpace(lifecycle=Lifecycle(), permissions=[Permission(level=PermissionLevel.CAN_MANAGE)]), + dataclass_example=GenieSpace( + lifecycle=Lifecycle(), + permissions=[Permission(level=PermissionLevel.CAN_MANAGE)], + ), mutator=genie_space_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/instance_pools.py b/python/databricks_tests/core/_generated/instance_pools.py index dd866e81fdb..585cdfaec8e 100644 --- a/python/databricks_tests/core/_generated/instance_pools.py +++ b/python/databricks_tests/core/_generated/instance_pools.py @@ -2,25 +2,65 @@ from databricks.bundles.core import Resources, instance_pool_mutator from databricks.bundles.core._generated.instance_pools import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.instance_pools._models.disk_spec import DiskSpec from databricks.bundles.instance_pools._models.docker_image import DockerImage from databricks.bundles.instance_pools._models.instance_pool import InstancePool -from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import InstancePoolAwsAttributes -from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import InstancePoolAzureAttributes -from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import InstancePoolGcpAttributes -from databricks.bundles.instance_pools._models.instance_pool_permission import InstancePoolPermission -from databricks.bundles.instance_pools._models.instance_pool_permission_level import InstancePoolPermissionLevel +from databricks.bundles.instance_pools._models.instance_pool_aws_attributes import ( + InstancePoolAwsAttributes, +) +from databricks.bundles.instance_pools._models.instance_pool_azure_attributes import ( + InstancePoolAzureAttributes, +) +from databricks.bundles.instance_pools._models.instance_pool_gcp_attributes import ( + InstancePoolGcpAttributes, +) +from databricks.bundles.instance_pools._models.instance_pool_permission import ( + InstancePoolPermission, +) +from databricks.bundles.instance_pools._models.instance_pool_permission_level import ( + InstancePoolPermissionLevel, +) from databricks.bundles.instance_pools._models.lifecycle import Lifecycle -from databricks.bundles.instance_pools._models.node_type_flexibility import NodeTypeFlexibility +from databricks.bundles.instance_pools._models.node_type_flexibility import ( + NodeTypeFlexibility, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_instance_pool, - dict_example={"aws_attributes": {}, "azure_attributes": {}, "custom_tags": {"key": "value"}, "disk_spec": {}, "gcp_attributes": {}, "instance_pool_name": "instance_pool_name", "lifecycle": {}, "node_type_flexibility": {}, "node_type_id": "node_type_id", "permissions": [{"level": "CAN_MANAGE"}], "preloaded_docker_images": [{}], "preloaded_spark_versions": ["preloaded_spark_versions"]}, - dataclass_example=InstancePool(aws_attributes=InstancePoolAwsAttributes(), azure_attributes=InstancePoolAzureAttributes(), custom_tags={"key": "value"}, disk_spec=DiskSpec(), gcp_attributes=InstancePoolGcpAttributes(), instance_pool_name="instance_pool_name", lifecycle=Lifecycle(), node_type_flexibility=NodeTypeFlexibility(), node_type_id="node_type_id", permissions=[InstancePoolPermission(level=InstancePoolPermissionLevel.CAN_MANAGE)], preloaded_docker_images=[DockerImage()], preloaded_spark_versions=["preloaded_spark_versions"]), + dict_example={ + "aws_attributes": {}, + "azure_attributes": {}, + "custom_tags": {"key": "value"}, + "disk_spec": {}, + "gcp_attributes": {}, + "instance_pool_name": "instance_pool_name", + "lifecycle": {}, + "node_type_flexibility": {}, + "node_type_id": "node_type_id", + "permissions": [{"level": "CAN_MANAGE"}], + "preloaded_docker_images": [{}], + "preloaded_spark_versions": ["preloaded_spark_versions"], + }, + dataclass_example=InstancePool( + aws_attributes=InstancePoolAwsAttributes(), + azure_attributes=InstancePoolAzureAttributes(), + custom_tags={"key": "value"}, + disk_spec=DiskSpec(), + gcp_attributes=InstancePoolGcpAttributes(), + instance_pool_name="instance_pool_name", + lifecycle=Lifecycle(), + node_type_flexibility=NodeTypeFlexibility(), + node_type_id="node_type_id", + permissions=[ + InstancePoolPermission(level=InstancePoolPermissionLevel.CAN_MANAGE) + ], + preloaded_docker_images=[DockerImage()], + preloaded_spark_versions=["preloaded_spark_versions"], + ), mutator=instance_pool_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/job_runs.py b/python/databricks_tests/core/_generated/job_runs.py index d0aeea28131..09b1c10f6eb 100644 --- a/python/databricks_tests/core/_generated/job_runs.py +++ b/python/databricks_tests/core/_generated/job_runs.py @@ -2,20 +2,36 @@ from databricks.bundles.core import Resources, job_run_mutator from databricks.bundles.core._generated.job_runs import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.job_runs._models.job_run import JobRun from databricks.bundles.job_runs._models.job_run_lifecycle import JobRunLifecycle from databricks.bundles.job_runs._models.performance_target import PerformanceTarget from databricks.bundles.job_runs._models.pipeline_params import PipelineParams from databricks.bundles.job_runs._models.queue_settings import QueueSettings +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_job_run, - dict_example={"job_id": 0, "job_parameters": {"key": "value"}, "lifecycle": {}, "only": ["only"], "performance_target": "PERFORMANCE_OPTIMIZED", "pipeline_params": {}, "queue": {"enabled": True}}, - dataclass_example=JobRun(job_id=0, job_parameters={"key": "value"}, lifecycle=JobRunLifecycle(), only=["only"], performance_target=PerformanceTarget.PERFORMANCE_OPTIMIZED, pipeline_params=PipelineParams(), queue=QueueSettings(enabled=True)), + dict_example={ + "job_id": 0, + "job_parameters": {"key": "value"}, + "lifecycle": {}, + "only": ["only"], + "performance_target": "PERFORMANCE_OPTIMIZED", + "pipeline_params": {}, + "queue": {"enabled": True}, + }, + dataclass_example=JobRun( + job_id=0, + job_parameters={"key": "value"}, + lifecycle=JobRunLifecycle(), + only=["only"], + performance_target=PerformanceTarget.PERFORMANCE_OPTIMIZED, + pipeline_params=PipelineParams(), + queue=QueueSettings(enabled=True), + ), mutator=job_run_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/jobs.py b/python/databricks_tests/core/_generated/jobs.py index 4c41941111a..3ed344b3811 100644 --- a/python/databricks_tests/core/_generated/jobs.py +++ b/python/databricks_tests/core/_generated/jobs.py @@ -2,17 +2,22 @@ from databricks.bundles.core import Resources, job_mutator from databricks.bundles.core._generated.jobs import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.jobs._models.continuous import Continuous from databricks.bundles.jobs._models.cron_schedule import CronSchedule from databricks.bundles.jobs._models.git_provider import GitProvider from databricks.bundles.jobs._models.git_source import GitSource from databricks.bundles.jobs._models.job import Job from databricks.bundles.jobs._models.job_cluster import JobCluster -from databricks.bundles.jobs._models.job_email_notifications import JobEmailNotifications +from databricks.bundles.jobs._models.job_email_notifications import ( + JobEmailNotifications, +) from databricks.bundles.jobs._models.job_environment import JobEnvironment -from databricks.bundles.jobs._models.job_notification_settings import JobNotificationSettings -from databricks.bundles.jobs._models.job_parameter_definition import JobParameterDefinition +from databricks.bundles.jobs._models.job_notification_settings import ( + JobNotificationSettings, +) +from databricks.bundles.jobs._models.job_parameter_definition import ( + JobParameterDefinition, +) from databricks.bundles.jobs._models.job_permission import JobPermission from databricks.bundles.jobs._models.job_permission_level import JobPermissionLevel from databricks.bundles.jobs._models.job_run_as import JobRunAs @@ -23,14 +28,61 @@ from databricks.bundles.jobs._models.task import Task from databricks.bundles.jobs._models.trigger_settings import TriggerSettings from databricks.bundles.jobs._models.webhook_notifications import WebhookNotifications +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_job, - dict_example={"continuous": {}, "email_notifications": {}, "environments": [{"environment_key": "environment_key"}], "git_source": {"git_provider": "gitHub", "git_url": "git_url"}, "health": {}, "job_clusters": [{"job_cluster_key": "job_cluster_key"}], "lifecycle": {}, "notification_settings": {}, "parameters": [{"default": "default", "name": "name"}], "performance_target": "PERFORMANCE_OPTIMIZED", "permissions": [{"level": "CAN_MANAGE"}], "queue": {"enabled": True}, "run_as": {}, "schedule": {"quartz_cron_expression": "quartz_cron_expression", "timezone_id": "timezone_id"}, "tags": {"key": "value"}, "tasks": [{"task_key": "task_key"}], "trigger": {}, "webhook_notifications": {}}, - dataclass_example=Job(continuous=Continuous(), email_notifications=JobEmailNotifications(), environments=[JobEnvironment(environment_key="environment_key")], git_source=GitSource(git_provider=GitProvider.GIT_HUB, git_url="git_url"), health=JobsHealthRules(), job_clusters=[JobCluster(job_cluster_key="job_cluster_key")], lifecycle=Lifecycle(), notification_settings=JobNotificationSettings(), parameters=[JobParameterDefinition(default="default", name="name")], performance_target=PerformanceTarget.PERFORMANCE_OPTIMIZED, permissions=[JobPermission(level=JobPermissionLevel.CAN_MANAGE)], queue=QueueSettings(enabled=True), run_as=JobRunAs(), schedule=CronSchedule(quartz_cron_expression="quartz_cron_expression", timezone_id="timezone_id"), tags={"key": "value"}, tasks=[Task(task_key="task_key")], trigger=TriggerSettings(), webhook_notifications=WebhookNotifications()), + dict_example={ + "continuous": {}, + "email_notifications": {}, + "environments": [{"environment_key": "environment_key"}], + "git_source": {"git_provider": "gitHub", "git_url": "git_url"}, + "health": {}, + "job_clusters": [{"job_cluster_key": "job_cluster_key"}], + "lifecycle": {}, + "notification_settings": {}, + "parameters": [{"default": "default", "name": "name"}], + "performance_target": "PERFORMANCE_OPTIMIZED", + "permissions": [{"level": "CAN_MANAGE"}], + "queue": {"enabled": True}, + "run_as": {}, + "schedule": { + "quartz_cron_expression": "quartz_cron_expression", + "timezone_id": "timezone_id", + }, + "tags": {"key": "value"}, + "tasks": [{"task_key": "task_key"}], + "trigger": {}, + "webhook_notifications": {}, + }, + dataclass_example=Job( + continuous=Continuous(), + email_notifications=JobEmailNotifications(), + environments=[JobEnvironment(environment_key="environment_key")], + git_source=GitSource( + git_provider=GitProvider.GIT_HUB, git_url="git_url" + ), + health=JobsHealthRules(), + job_clusters=[JobCluster(job_cluster_key="job_cluster_key")], + lifecycle=Lifecycle(), + notification_settings=JobNotificationSettings(), + parameters=[JobParameterDefinition(default="default", name="name")], + performance_target=PerformanceTarget.PERFORMANCE_OPTIMIZED, + permissions=[JobPermission(level=JobPermissionLevel.CAN_MANAGE)], + queue=QueueSettings(enabled=True), + run_as=JobRunAs(), + schedule=CronSchedule( + quartz_cron_expression="quartz_cron_expression", + timezone_id="timezone_id", + ), + tags={"key": "value"}, + tasks=[Task(task_key="task_key")], + trigger=TriggerSettings(), + webhook_notifications=WebhookNotifications(), + ), mutator=job_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/mcp_services.py b/python/databricks_tests/core/_generated/mcp_services.py index 9e5ab45845e..ca1247a300d 100644 --- a/python/databricks_tests/core/_generated/mcp_services.py +++ b/python/databricks_tests/core/_generated/mcp_services.py @@ -2,19 +2,33 @@ from databricks.bundles.core import Resources, mcp_service_mutator from databricks.bundles.core._generated.mcp_services import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.mcp_services._models.lifecycle import Lifecycle from databricks.bundles.mcp_services._models.mcp_service import McpService from databricks.bundles.mcp_services._models.mcp_service_config import McpServiceConfig -from databricks.bundles.mcp_services._models.privilege_assignment import PrivilegeAssignment +from databricks.bundles.mcp_services._models.privilege_assignment import ( + PrivilegeAssignment, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_mcp_service, - dict_example={"config": {}, "grants": [{}], "lifecycle": {}, "mcp_service_id": "mcp_service_id", "parent": "parent"}, - dataclass_example=McpService(config=McpServiceConfig(), grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), mcp_service_id="mcp_service_id", parent="parent"), + dict_example={ + "config": {}, + "grants": [{}], + "lifecycle": {}, + "mcp_service_id": "mcp_service_id", + "parent": "parent", + }, + dataclass_example=McpService( + config=McpServiceConfig(), + grants=[PrivilegeAssignment()], + lifecycle=Lifecycle(), + mcp_service_id="mcp_service_id", + parent="parent", + ), mutator=mcp_service_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/model_provider_services.py b/python/databricks_tests/core/_generated/model_provider_services.py index bc933b54a92..994624aeee8 100644 --- a/python/databricks_tests/core/_generated/model_provider_services.py +++ b/python/databricks_tests/core/_generated/model_provider_services.py @@ -2,19 +2,37 @@ from databricks.bundles.core import Resources, model_provider_service_mutator from databricks.bundles.core._generated.model_provider_services import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.model_provider_services._models.lifecycle import Lifecycle -from databricks.bundles.model_provider_services._models.model_provider_service import ModelProviderService -from databricks.bundles.model_provider_services._models.model_provider_service_config import ModelProviderServiceConfig -from databricks.bundles.model_provider_services._models.privilege_assignment import PrivilegeAssignment +from databricks.bundles.model_provider_services._models.model_provider_service import ( + ModelProviderService, +) +from databricks.bundles.model_provider_services._models.model_provider_service_config import ( + ModelProviderServiceConfig, +) +from databricks.bundles.model_provider_services._models.privilege_assignment import ( + PrivilegeAssignment, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_model_provider_service, - dict_example={"config": {}, "grants": [{}], "lifecycle": {}, "model_provider_service_id": "model_provider_service_id", "parent": "parent"}, - dataclass_example=ModelProviderService(config=ModelProviderServiceConfig(), grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), model_provider_service_id="model_provider_service_id", parent="parent"), + dict_example={ + "config": {}, + "grants": [{}], + "lifecycle": {}, + "model_provider_service_id": "model_provider_service_id", + "parent": "parent", + }, + dataclass_example=ModelProviderService( + config=ModelProviderServiceConfig(), + grants=[PrivilegeAssignment()], + lifecycle=Lifecycle(), + model_provider_service_id="model_provider_service_id", + parent="parent", + ), mutator=model_provider_service_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/model_services.py b/python/databricks_tests/core/_generated/model_services.py index f188ac17721..5d301ae0203 100644 --- a/python/databricks_tests/core/_generated/model_services.py +++ b/python/databricks_tests/core/_generated/model_services.py @@ -2,19 +2,35 @@ from databricks.bundles.core import Resources, model_service_mutator from databricks.bundles.core._generated.model_services import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.model_services._models.lifecycle import Lifecycle from databricks.bundles.model_services._models.model_service import ModelService -from databricks.bundles.model_services._models.model_service_config import ModelServiceConfig -from databricks.bundles.model_services._models.privilege_assignment import PrivilegeAssignment +from databricks.bundles.model_services._models.model_service_config import ( + ModelServiceConfig, +) +from databricks.bundles.model_services._models.privilege_assignment import ( + PrivilegeAssignment, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_model_service, - dict_example={"config": {}, "grants": [{}], "lifecycle": {}, "model_service_id": "model_service_id", "parent": "parent"}, - dataclass_example=ModelService(config=ModelServiceConfig(), grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), model_service_id="model_service_id", parent="parent"), + dict_example={ + "config": {}, + "grants": [{}], + "lifecycle": {}, + "model_service_id": "model_service_id", + "parent": "parent", + }, + dataclass_example=ModelService( + config=ModelServiceConfig(), + grants=[PrivilegeAssignment()], + lifecycle=Lifecycle(), + model_service_id="model_service_id", + parent="parent", + ), mutator=model_service_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/model_serving_endpoints.py b/python/databricks_tests/core/_generated/model_serving_endpoints.py index af99b59b3cc..f658d74e899 100644 --- a/python/databricks_tests/core/_generated/model_serving_endpoints.py +++ b/python/databricks_tests/core/_generated/model_serving_endpoints.py @@ -2,24 +2,60 @@ from databricks.bundles.core import Resources, model_serving_endpoint_mutator from databricks.bundles.core._generated.model_serving_endpoints import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase -from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import AiGatewayConfig -from databricks.bundles.model_serving_endpoints._models.email_notifications import EmailNotifications -from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import EndpointCoreConfigInput +from databricks.bundles.model_serving_endpoints._models.ai_gateway_config import ( + AiGatewayConfig, +) +from databricks.bundles.model_serving_endpoints._models.email_notifications import ( + EmailNotifications, +) +from databricks.bundles.model_serving_endpoints._models.endpoint_core_config_input import ( + EndpointCoreConfigInput, +) from databricks.bundles.model_serving_endpoints._models.endpoint_tag import EndpointTag from databricks.bundles.model_serving_endpoints._models.lifecycle import Lifecycle -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ModelServingEndpoint -from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ModelServingEndpointPermission -from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ServingEndpointPermissionLevel -from databricks.bundles.model_serving_endpoints._models.telemetry_config import TelemetryConfig +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint import ( + ModelServingEndpoint, +) +from databricks.bundles.model_serving_endpoints._models.model_serving_endpoint_permission import ( + ModelServingEndpointPermission, +) +from databricks.bundles.model_serving_endpoints._models.serving_endpoint_permission_level import ( + ServingEndpointPermissionLevel, +) +from databricks.bundles.model_serving_endpoints._models.telemetry_config import ( + TelemetryConfig, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_model_serving_endpoint, - dict_example={"ai_gateway": {}, "config": {}, "email_notifications": {}, "lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_MANAGE"}], "tags": [{"key": "key"}], "telemetry_config": {}}, - dataclass_example=ModelServingEndpoint(ai_gateway=AiGatewayConfig(), config=EndpointCoreConfigInput(), email_notifications=EmailNotifications(), lifecycle=Lifecycle(), name="name", permissions=[ModelServingEndpointPermission(level=ServingEndpointPermissionLevel.CAN_MANAGE)], tags=[EndpointTag(key="key")], telemetry_config=TelemetryConfig()), + dict_example={ + "ai_gateway": {}, + "config": {}, + "email_notifications": {}, + "lifecycle": {}, + "name": "name", + "permissions": [{"level": "CAN_MANAGE"}], + "tags": [{"key": "key"}], + "telemetry_config": {}, + }, + dataclass_example=ModelServingEndpoint( + ai_gateway=AiGatewayConfig(), + config=EndpointCoreConfigInput(), + email_notifications=EmailNotifications(), + lifecycle=Lifecycle(), + name="name", + permissions=[ + ModelServingEndpointPermission( + level=ServingEndpointPermissionLevel.CAN_MANAGE + ) + ], + tags=[EndpointTag(key="key")], + telemetry_config=TelemetryConfig(), + ), mutator=model_serving_endpoint_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/models.py b/python/databricks_tests/core/_generated/models.py index 6facdc94b7d..e1c3c16deea 100644 --- a/python/databricks_tests/core/_generated/models.py +++ b/python/databricks_tests/core/_generated/models.py @@ -2,20 +2,38 @@ from databricks.bundles.core import Resources, mlflow_model_mutator from databricks.bundles.core._generated.models import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.models._models.lifecycle import Lifecycle from databricks.bundles.models._models.mlflow_model import MlflowModel -from databricks.bundles.models._models.mlflow_model_permission import MlflowModelPermission +from databricks.bundles.models._models.mlflow_model_permission import ( + MlflowModelPermission, +) from databricks.bundles.models._models.model_tag import ModelTag -from databricks.bundles.models._models.registered_model_permission_level import RegisteredModelPermissionLevel +from databricks.bundles.models._models.registered_model_permission_level import ( + RegisteredModelPermissionLevel, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_mlflow_model, - dict_example={"lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_MANAGE"}], "tags": [{}]}, - dataclass_example=MlflowModel(lifecycle=Lifecycle(), name="name", permissions=[MlflowModelPermission(level=RegisteredModelPermissionLevel.CAN_MANAGE)], tags=[ModelTag()]), + dict_example={ + "lifecycle": {}, + "name": "name", + "permissions": [{"level": "CAN_MANAGE"}], + "tags": [{}], + }, + dataclass_example=MlflowModel( + lifecycle=Lifecycle(), + name="name", + permissions=[ + MlflowModelPermission( + level=RegisteredModelPermissionLevel.CAN_MANAGE + ) + ], + tags=[ModelTag()], + ), mutator=mlflow_model_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/pipelines.py b/python/databricks_tests/core/_generated/pipelines.py index 99ac31bf846..46ecc044b1f 100644 --- a/python/databricks_tests/core/_generated/pipelines.py +++ b/python/databricks_tests/core/_generated/pipelines.py @@ -2,27 +2,61 @@ from databricks.bundles.core import Resources, pipeline_mutator from databricks.bundles.core._generated.pipelines import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.pipelines._models.event_log_spec import EventLogSpec from databricks.bundles.pipelines._models.filters import Filters -from databricks.bundles.pipelines._models.ingestion_pipeline_definition import IngestionPipelineDefinition +from databricks.bundles.pipelines._models.ingestion_pipeline_definition import ( + IngestionPipelineDefinition, +) from databricks.bundles.pipelines._models.lifecycle import Lifecycle from databricks.bundles.pipelines._models.notifications import Notifications from databricks.bundles.pipelines._models.pipeline import Pipeline from databricks.bundles.pipelines._models.pipeline_cluster import PipelineCluster from databricks.bundles.pipelines._models.pipeline_library import PipelineLibrary from databricks.bundles.pipelines._models.pipeline_permission import PipelinePermission -from databricks.bundles.pipelines._models.pipeline_permission_level import PipelinePermissionLevel -from databricks.bundles.pipelines._models.pipelines_environment import PipelinesEnvironment +from databricks.bundles.pipelines._models.pipeline_permission_level import ( + PipelinePermissionLevel, +) +from databricks.bundles.pipelines._models.pipelines_environment import ( + PipelinesEnvironment, +) from databricks.bundles.pipelines._models.run_as import RunAs +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_pipeline, - dict_example={"clusters": [{}], "configuration": {"key": "value"}, "environment": {}, "event_log": {}, "filters": {}, "ingestion_definition": {}, "libraries": [{}], "lifecycle": {}, "notifications": [{}], "permissions": [{"level": "CAN_MANAGE"}], "run_as": {}, "tags": {"key": "value"}}, - dataclass_example=Pipeline(clusters=[PipelineCluster()], configuration={"key": "value"}, environment=PipelinesEnvironment(), event_log=EventLogSpec(), filters=Filters(), ingestion_definition=IngestionPipelineDefinition(), libraries=[PipelineLibrary()], lifecycle=Lifecycle(), notifications=[Notifications()], permissions=[PipelinePermission(level=PipelinePermissionLevel.CAN_MANAGE)], run_as=RunAs(), tags={"key": "value"}), + dict_example={ + "clusters": [{}], + "configuration": {"key": "value"}, + "environment": {}, + "event_log": {}, + "filters": {}, + "ingestion_definition": {}, + "libraries": [{}], + "lifecycle": {}, + "notifications": [{}], + "permissions": [{"level": "CAN_MANAGE"}], + "run_as": {}, + "tags": {"key": "value"}, + }, + dataclass_example=Pipeline( + clusters=[PipelineCluster()], + configuration={"key": "value"}, + environment=PipelinesEnvironment(), + event_log=EventLogSpec(), + filters=Filters(), + ingestion_definition=IngestionPipelineDefinition(), + libraries=[PipelineLibrary()], + lifecycle=Lifecycle(), + notifications=[Notifications()], + permissions=[ + PipelinePermission(level=PipelinePermissionLevel.CAN_MANAGE) + ], + run_as=RunAs(), + tags={"key": "value"}, + ), mutator=pipeline_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/quality_monitors.py b/python/databricks_tests/core/_generated/quality_monitors.py index f3b0040d937..4e244d7cdd5 100644 --- a/python/databricks_tests/core/_generated/quality_monitors.py +++ b/python/databricks_tests/core/_generated/quality_monitors.py @@ -2,25 +2,100 @@ from databricks.bundles.core import Resources, quality_monitor_mutator from databricks.bundles.core._generated.quality_monitors import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.quality_monitors._models.lifecycle import Lifecycle -from databricks.bundles.quality_monitors._models.monitor_cron_schedule import MonitorCronSchedule -from databricks.bundles.quality_monitors._models.monitor_inference_log import MonitorInferenceLog -from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import MonitorInferenceLogProblemType +from databricks.bundles.quality_monitors._models.monitor_cron_schedule import ( + MonitorCronSchedule, +) +from databricks.bundles.quality_monitors._models.monitor_inference_log import ( + MonitorInferenceLog, +) +from databricks.bundles.quality_monitors._models.monitor_inference_log_problem_type import ( + MonitorInferenceLogProblemType, +) from databricks.bundles.quality_monitors._models.monitor_metric import MonitorMetric -from databricks.bundles.quality_monitors._models.monitor_metric_type import MonitorMetricType -from databricks.bundles.quality_monitors._models.monitor_notifications import MonitorNotifications +from databricks.bundles.quality_monitors._models.monitor_metric_type import ( + MonitorMetricType, +) +from databricks.bundles.quality_monitors._models.monitor_notifications import ( + MonitorNotifications, +) from databricks.bundles.quality_monitors._models.monitor_snapshot import MonitorSnapshot -from databricks.bundles.quality_monitors._models.monitor_time_series import MonitorTimeSeries +from databricks.bundles.quality_monitors._models.monitor_time_series import ( + MonitorTimeSeries, +) from databricks.bundles.quality_monitors._models.quality_monitor import QualityMonitor +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_quality_monitor, - dict_example={"assets_dir": "assets_dir", "custom_metrics": [{"definition": "definition", "input_columns": ["input_columns"], "name": "name", "output_data_type": "output_data_type", "type": "CUSTOM_METRIC_TYPE_AGGREGATE"}], "inference_log": {"granularities": ["granularities"], "model_id_col": "model_id_col", "prediction_col": "prediction_col", "problem_type": "PROBLEM_TYPE_CLASSIFICATION", "timestamp_col": "timestamp_col"}, "lifecycle": {}, "notifications": {}, "output_schema_name": "output_schema_name", "schedule": {"quartz_cron_expression": "quartz_cron_expression", "timezone_id": "timezone_id"}, "slicing_exprs": ["slicing_exprs"], "snapshot": {}, "table_name": "table_name", "time_series": {"granularities": ["granularities"], "timestamp_col": "timestamp_col"}}, - dataclass_example=QualityMonitor(assets_dir="assets_dir", custom_metrics=[MonitorMetric(definition="definition", input_columns=["input_columns"], name="name", output_data_type="output_data_type", type=MonitorMetricType.CUSTOM_METRIC_TYPE_AGGREGATE)], inference_log=MonitorInferenceLog(granularities=["granularities"], model_id_col="model_id_col", prediction_col="prediction_col", problem_type=MonitorInferenceLogProblemType.PROBLEM_TYPE_CLASSIFICATION, timestamp_col="timestamp_col"), lifecycle=Lifecycle(), notifications=MonitorNotifications(), output_schema_name="output_schema_name", schedule=MonitorCronSchedule(quartz_cron_expression="quartz_cron_expression", timezone_id="timezone_id"), slicing_exprs=["slicing_exprs"], snapshot=MonitorSnapshot(), table_name="table_name", time_series=MonitorTimeSeries(granularities=["granularities"], timestamp_col="timestamp_col")), + dict_example={ + "assets_dir": "assets_dir", + "custom_metrics": [ + { + "definition": "definition", + "input_columns": ["input_columns"], + "name": "name", + "output_data_type": "output_data_type", + "type": "CUSTOM_METRIC_TYPE_AGGREGATE", + } + ], + "inference_log": { + "granularities": ["granularities"], + "model_id_col": "model_id_col", + "prediction_col": "prediction_col", + "problem_type": "PROBLEM_TYPE_CLASSIFICATION", + "timestamp_col": "timestamp_col", + }, + "lifecycle": {}, + "notifications": {}, + "output_schema_name": "output_schema_name", + "schedule": { + "quartz_cron_expression": "quartz_cron_expression", + "timezone_id": "timezone_id", + }, + "slicing_exprs": ["slicing_exprs"], + "snapshot": {}, + "table_name": "table_name", + "time_series": { + "granularities": ["granularities"], + "timestamp_col": "timestamp_col", + }, + }, + dataclass_example=QualityMonitor( + assets_dir="assets_dir", + custom_metrics=[ + MonitorMetric( + definition="definition", + input_columns=["input_columns"], + name="name", + output_data_type="output_data_type", + type=MonitorMetricType.CUSTOM_METRIC_TYPE_AGGREGATE, + ) + ], + inference_log=MonitorInferenceLog( + granularities=["granularities"], + model_id_col="model_id_col", + prediction_col="prediction_col", + problem_type=MonitorInferenceLogProblemType.PROBLEM_TYPE_CLASSIFICATION, + timestamp_col="timestamp_col", + ), + lifecycle=Lifecycle(), + notifications=MonitorNotifications(), + output_schema_name="output_schema_name", + schedule=MonitorCronSchedule( + quartz_cron_expression="quartz_cron_expression", + timezone_id="timezone_id", + ), + slicing_exprs=["slicing_exprs"], + snapshot=MonitorSnapshot(), + table_name="table_name", + time_series=MonitorTimeSeries( + granularities=["granularities"], timestamp_col="timestamp_col" + ), + ), mutator=quality_monitor_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/registered_models.py b/python/databricks_tests/core/_generated/registered_models.py index b3689337a5f..81d5f2d7ed4 100644 --- a/python/databricks_tests/core/_generated/registered_models.py +++ b/python/databricks_tests/core/_generated/registered_models.py @@ -2,11 +2,17 @@ from databricks.bundles.core import Resources, registered_model_mutator from databricks.bundles.core._generated.registered_models import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.registered_models._models.lifecycle import Lifecycle -from databricks.bundles.registered_models._models.privilege_assignment import PrivilegeAssignment -from databricks.bundles.registered_models._models.registered_model import RegisteredModel -from databricks.bundles.registered_models._models.registered_model_alias import RegisteredModelAlias +from databricks.bundles.registered_models._models.privilege_assignment import ( + PrivilegeAssignment, +) +from databricks.bundles.registered_models._models.registered_model import ( + RegisteredModel, +) +from databricks.bundles.registered_models._models.registered_model_alias import ( + RegisteredModelAlias, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): @@ -14,7 +20,11 @@ def _test_case(): ResourceTestCase( add_resource=Resources.add_registered_model, dict_example={"aliases": [{}], "grants": [{}], "lifecycle": {}}, - dataclass_example=RegisteredModel(aliases=[RegisteredModelAlias()], grants=[PrivilegeAssignment()], lifecycle=Lifecycle()), + dataclass_example=RegisteredModel( + aliases=[RegisteredModelAlias()], + grants=[PrivilegeAssignment()], + lifecycle=Lifecycle(), + ), mutator=registered_model_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/schemas.py b/python/databricks_tests/core/_generated/schemas.py index a68dcedc8ed..576b0c7d3cd 100644 --- a/python/databricks_tests/core/_generated/schemas.py +++ b/python/databricks_tests/core/_generated/schemas.py @@ -2,18 +2,30 @@ from databricks.bundles.core import Resources, schema_mutator from databricks.bundles.core._generated.schemas import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.schemas._models.lifecycle import Lifecycle from databricks.bundles.schemas._models.privilege_assignment import PrivilegeAssignment from databricks.bundles.schemas._models.schema import Schema +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_schema, - dict_example={"catalog_name": "catalog_name", "grants": [{}], "lifecycle": {}, "name": "name", "properties": {"key": "value"}}, - dataclass_example=Schema(catalog_name="catalog_name", grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), name="name", properties={"key": "value"}), + dict_example={ + "catalog_name": "catalog_name", + "grants": [{}], + "lifecycle": {}, + "name": "name", + "properties": {"key": "value"}, + }, + dataclass_example=Schema( + catalog_name="catalog_name", + grants=[PrivilegeAssignment()], + lifecycle=Lifecycle(), + name="name", + properties={"key": "value"}, + ), mutator=schema_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/secret_scopes.py b/python/databricks_tests/core/_generated/secret_scopes.py index 601c0194360..8edc6ce6e1b 100644 --- a/python/databricks_tests/core/_generated/secret_scopes.py +++ b/python/databricks_tests/core/_generated/secret_scopes.py @@ -2,21 +2,46 @@ from databricks.bundles.core import Resources, secret_scope_mutator from databricks.bundles.core._generated.secret_scopes import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase -from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import AzureKeyVaultSecretScopeMetadata +from databricks.bundles.secret_scopes._models.azure_key_vault_secret_scope_metadata import ( + AzureKeyVaultSecretScopeMetadata, +) from databricks.bundles.secret_scopes._models.lifecycle import Lifecycle from databricks.bundles.secret_scopes._models.scope_backend_type import ScopeBackendType from databricks.bundles.secret_scopes._models.secret_scope import SecretScope -from databricks.bundles.secret_scopes._models.secret_scope_permission import SecretScopePermission -from databricks.bundles.secret_scopes._models.secret_scope_permission_level import SecretScopePermissionLevel +from databricks.bundles.secret_scopes._models.secret_scope_permission import ( + SecretScopePermission, +) +from databricks.bundles.secret_scopes._models.secret_scope_permission_level import ( + SecretScopePermissionLevel, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_secret_scope, - dict_example={"backend_type": "DATABRICKS", "keyvault_metadata": {"dns_name": "dns_name", "resource_id": "resource_id"}, "lifecycle": {}, "name": "name", "permissions": [{"level": "READ"}]}, - dataclass_example=SecretScope(backend_type=ScopeBackendType.DATABRICKS, keyvault_metadata=AzureKeyVaultSecretScopeMetadata(dns_name="dns_name", resource_id="resource_id"), lifecycle=Lifecycle(), name="name", permissions=[SecretScopePermission(level=SecretScopePermissionLevel.READ)]), + dict_example={ + "backend_type": "DATABRICKS", + "keyvault_metadata": { + "dns_name": "dns_name", + "resource_id": "resource_id", + }, + "lifecycle": {}, + "name": "name", + "permissions": [{"level": "READ"}], + }, + dataclass_example=SecretScope( + backend_type=ScopeBackendType.DATABRICKS, + keyvault_metadata=AzureKeyVaultSecretScopeMetadata( + dns_name="dns_name", resource_id="resource_id" + ), + lifecycle=Lifecycle(), + name="name", + permissions=[ + SecretScopePermission(level=SecretScopePermissionLevel.READ) + ], + ), mutator=secret_scope_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/secrets.py b/python/databricks_tests/core/_generated/secrets.py index 65cf7791b63..65e782459d3 100644 --- a/python/databricks_tests/core/_generated/secrets.py +++ b/python/databricks_tests/core/_generated/secrets.py @@ -2,18 +2,32 @@ from databricks.bundles.core import Resources, secret_mutator from databricks.bundles.core._generated.secrets import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.secrets._models.lifecycle import Lifecycle from databricks.bundles.secrets._models.privilege_assignment import PrivilegeAssignment from databricks.bundles.secrets._models.secret import Secret +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_secret, - dict_example={"catalog_name": "catalog_name", "grants": [{}], "lifecycle": {}, "name": "name", "schema_name": "schema_name", "value": "value"}, - dataclass_example=Secret(catalog_name="catalog_name", grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), name="name", schema_name="schema_name", value="value"), + dict_example={ + "catalog_name": "catalog_name", + "grants": [{}], + "lifecycle": {}, + "name": "name", + "schema_name": "schema_name", + "value": "value", + }, + dataclass_example=Secret( + catalog_name="catalog_name", + grants=[PrivilegeAssignment()], + lifecycle=Lifecycle(), + name="name", + schema_name="schema_name", + value="value", + ), mutator=secret_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/sql_warehouses.py b/python/databricks_tests/core/_generated/sql_warehouses.py index 2f3754f9063..e6810ecd57c 100644 --- a/python/databricks_tests/core/_generated/sql_warehouses.py +++ b/python/databricks_tests/core/_generated/sql_warehouses.py @@ -2,23 +2,49 @@ from databricks.bundles.core import Resources, sql_warehouse_mutator from databricks.bundles.core._generated.sql_warehouses import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.sql_warehouses._models.channel import Channel -from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import CreateWarehouseRequestWarehouseType +from databricks.bundles.sql_warehouses._models.create_warehouse_request_warehouse_type import ( + CreateWarehouseRequestWarehouseType, +) from databricks.bundles.sql_warehouses._models.endpoint_tags import EndpointTags -from databricks.bundles.sql_warehouses._models.lifecycle_with_started import LifecycleWithStarted -from databricks.bundles.sql_warehouses._models.spot_instance_policy import SpotInstancePolicy +from databricks.bundles.sql_warehouses._models.lifecycle_with_started import ( + LifecycleWithStarted, +) +from databricks.bundles.sql_warehouses._models.spot_instance_policy import ( + SpotInstancePolicy, +) from databricks.bundles.sql_warehouses._models.sql_warehouse import SqlWarehouse -from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import SqlWarehousePermission -from databricks.bundles.sql_warehouses._models.warehouse_permission_level import WarehousePermissionLevel +from databricks.bundles.sql_warehouses._models.sql_warehouse_permission import ( + SqlWarehousePermission, +) +from databricks.bundles.sql_warehouses._models.warehouse_permission_level import ( + WarehousePermissionLevel, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_sql_warehouse, - dict_example={"channel": {}, "lifecycle": {}, "permissions": [{"level": "CAN_MANAGE"}], "spot_instance_policy": "POLICY_UNSPECIFIED", "tags": {}, "warehouse_type": "TYPE_UNSPECIFIED"}, - dataclass_example=SqlWarehouse(channel=Channel(), lifecycle=LifecycleWithStarted(), permissions=[SqlWarehousePermission(level=WarehousePermissionLevel.CAN_MANAGE)], spot_instance_policy=SpotInstancePolicy.POLICY_UNSPECIFIED, tags=EndpointTags(), warehouse_type=CreateWarehouseRequestWarehouseType.TYPE_UNSPECIFIED), + dict_example={ + "channel": {}, + "lifecycle": {}, + "permissions": [{"level": "CAN_MANAGE"}], + "spot_instance_policy": "POLICY_UNSPECIFIED", + "tags": {}, + "warehouse_type": "TYPE_UNSPECIFIED", + }, + dataclass_example=SqlWarehouse( + channel=Channel(), + lifecycle=LifecycleWithStarted(), + permissions=[ + SqlWarehousePermission(level=WarehousePermissionLevel.CAN_MANAGE) + ], + spot_instance_policy=SpotInstancePolicy.POLICY_UNSPECIFIED, + tags=EndpointTags(), + warehouse_type=CreateWarehouseRequestWarehouseType.TYPE_UNSPECIFIED, + ), mutator=sql_warehouse_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/synced_database_tables.py b/python/databricks_tests/core/_generated/synced_database_tables.py index 2e20187b252..dea7172089a 100644 --- a/python/databricks_tests/core/_generated/synced_database_tables.py +++ b/python/databricks_tests/core/_generated/synced_database_tables.py @@ -2,10 +2,14 @@ from databricks.bundles.core import Resources, synced_database_table_mutator from databricks.bundles.core._generated.synced_database_tables import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.synced_database_tables._models.lifecycle import Lifecycle -from databricks.bundles.synced_database_tables._models.synced_database_table import SyncedDatabaseTable -from databricks.bundles.synced_database_tables._models.synced_table_spec import SyncedTableSpec +from databricks.bundles.synced_database_tables._models.synced_database_table import ( + SyncedDatabaseTable, +) +from databricks.bundles.synced_database_tables._models.synced_table_spec import ( + SyncedTableSpec, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): @@ -13,7 +17,9 @@ def _test_case(): ResourceTestCase( add_resource=Resources.add_synced_database_table, dict_example={"lifecycle": {}, "name": "name", "spec": {}}, - dataclass_example=SyncedDatabaseTable(lifecycle=Lifecycle(), name="name", spec=SyncedTableSpec()), + dataclass_example=SyncedDatabaseTable( + lifecycle=Lifecycle(), name="name", spec=SyncedTableSpec() + ), mutator=synced_database_table_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/vector_search_endpoints.py b/python/databricks_tests/core/_generated/vector_search_endpoints.py index fe2a4f5f844..30bdb70e91e 100644 --- a/python/databricks_tests/core/_generated/vector_search_endpoints.py +++ b/python/databricks_tests/core/_generated/vector_search_endpoints.py @@ -2,20 +2,42 @@ from databricks.bundles.core import Resources, vector_search_endpoint_mutator from databricks.bundles.core._generated.vector_search_endpoints import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase -from databricks.bundles.vector_search_endpoints._models.endpoint_type import EndpointType +from databricks.bundles.vector_search_endpoints._models.endpoint_type import ( + EndpointType, +) from databricks.bundles.vector_search_endpoints._models.lifecycle import Lifecycle -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import VectorSearchEndpoint -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import VectorSearchEndpointPermission -from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import VectorSearchEndpointPermissionLevel +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint import ( + VectorSearchEndpoint, +) +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission import ( + VectorSearchEndpointPermission, +) +from databricks.bundles.vector_search_endpoints._models.vector_search_endpoint_permission_level import ( + VectorSearchEndpointPermissionLevel, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_vector_search_endpoint, - dict_example={"endpoint_type": "STORAGE_OPTIMIZED", "lifecycle": {}, "name": "name", "permissions": [{"level": "CAN_CREATE"}]}, - dataclass_example=VectorSearchEndpoint(endpoint_type=EndpointType.STORAGE_OPTIMIZED, lifecycle=Lifecycle(), name="name", permissions=[VectorSearchEndpointPermission(level=VectorSearchEndpointPermissionLevel.CAN_CREATE)]), + dict_example={ + "endpoint_type": "STORAGE_OPTIMIZED", + "lifecycle": {}, + "name": "name", + "permissions": [{"level": "CAN_CREATE"}], + }, + dataclass_example=VectorSearchEndpoint( + endpoint_type=EndpointType.STORAGE_OPTIMIZED, + lifecycle=Lifecycle(), + name="name", + permissions=[ + VectorSearchEndpointPermission( + level=VectorSearchEndpointPermissionLevel.CAN_CREATE + ) + ], + ), mutator=vector_search_endpoint_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/vector_search_indexes.py b/python/databricks_tests/core/_generated/vector_search_indexes.py index ba7205f6fe1..d72ed337f55 100644 --- a/python/databricks_tests/core/_generated/vector_search_indexes.py +++ b/python/databricks_tests/core/_generated/vector_search_indexes.py @@ -2,21 +2,49 @@ from databricks.bundles.core import Resources, vector_search_index_mutator from databricks.bundles.core._generated.vector_search_indexes import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase -from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import DeltaSyncVectorIndexSpecRequest -from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import DirectAccessVectorIndexSpec +from databricks.bundles.vector_search_indexes._models.delta_sync_vector_index_spec_request import ( + DeltaSyncVectorIndexSpecRequest, +) +from databricks.bundles.vector_search_indexes._models.direct_access_vector_index_spec import ( + DirectAccessVectorIndexSpec, +) from databricks.bundles.vector_search_indexes._models.lifecycle import Lifecycle -from databricks.bundles.vector_search_indexes._models.privilege_assignment import PrivilegeAssignment -from databricks.bundles.vector_search_indexes._models.vector_index_type import VectorIndexType -from databricks.bundles.vector_search_indexes._models.vector_search_index import VectorSearchIndex +from databricks.bundles.vector_search_indexes._models.privilege_assignment import ( + PrivilegeAssignment, +) +from databricks.bundles.vector_search_indexes._models.vector_index_type import ( + VectorIndexType, +) +from databricks.bundles.vector_search_indexes._models.vector_search_index import ( + VectorSearchIndex, +) +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_vector_search_index, - dict_example={"delta_sync_index_spec": {}, "direct_access_index_spec": {}, "endpoint_name": "endpoint_name", "grants": [{}], "index_type": "DELTA_SYNC", "lifecycle": {}, "name": "name", "primary_key": "primary_key"}, - dataclass_example=VectorSearchIndex(delta_sync_index_spec=DeltaSyncVectorIndexSpecRequest(), direct_access_index_spec=DirectAccessVectorIndexSpec(), endpoint_name="endpoint_name", grants=[PrivilegeAssignment()], index_type=VectorIndexType.DELTA_SYNC, lifecycle=Lifecycle(), name="name", primary_key="primary_key"), + dict_example={ + "delta_sync_index_spec": {}, + "direct_access_index_spec": {}, + "endpoint_name": "endpoint_name", + "grants": [{}], + "index_type": "DELTA_SYNC", + "lifecycle": {}, + "name": "name", + "primary_key": "primary_key", + }, + dataclass_example=VectorSearchIndex( + delta_sync_index_spec=DeltaSyncVectorIndexSpecRequest(), + direct_access_index_spec=DirectAccessVectorIndexSpec(), + endpoint_name="endpoint_name", + grants=[PrivilegeAssignment()], + index_type=VectorIndexType.DELTA_SYNC, + lifecycle=Lifecycle(), + name="name", + primary_key="primary_key", + ), mutator=vector_search_index_mutator, ), _resource_type(), diff --git a/python/databricks_tests/core/_generated/volumes.py b/python/databricks_tests/core/_generated/volumes.py index eb4fbb4b732..58d1b9a4459 100644 --- a/python/databricks_tests/core/_generated/volumes.py +++ b/python/databricks_tests/core/_generated/volumes.py @@ -2,19 +2,33 @@ from databricks.bundles.core import Resources, volume_mutator from databricks.bundles.core._generated.volumes import _resource_type -from databricks_tests.core._resource_test_case import ResourceTestCase from databricks.bundles.volumes._models.lifecycle import Lifecycle from databricks.bundles.volumes._models.privilege_assignment import PrivilegeAssignment from databricks.bundles.volumes._models.volume import Volume from databricks.bundles.volumes._models.volume_type import VolumeType +from databricks_tests.core._resource_test_case import ResourceTestCase def _test_case(): return ( ResourceTestCase( add_resource=Resources.add_volume, - dict_example={"catalog_name": "catalog_name", "grants": [{}], "lifecycle": {}, "name": "name", "schema_name": "schema_name", "volume_type": "MANAGED"}, - dataclass_example=Volume(catalog_name="catalog_name", grants=[PrivilegeAssignment()], lifecycle=Lifecycle(), name="name", schema_name="schema_name", volume_type=VolumeType.MANAGED), + dict_example={ + "catalog_name": "catalog_name", + "grants": [{}], + "lifecycle": {}, + "name": "name", + "schema_name": "schema_name", + "volume_type": "MANAGED", + }, + dataclass_example=Volume( + catalog_name="catalog_name", + grants=[PrivilegeAssignment()], + lifecycle=Lifecycle(), + name="name", + schema_name="schema_name", + volume_type=VolumeType.MANAGED, + ), mutator=volume_mutator, ), _resource_type(), From e9db3bc2eb08102506b6eeef8c76cb5a4c8a24aa Mon Sep 17 00:00:00 2001 From: Jan Rose Date: Thu, 22 Oct 2026 06:03:47 +0000 Subject: [PATCH 3/4] Wire new SQL warehouse fields through the direct engine v0.190.0 adds CustomerVersionLabel, MaxDbu and ReydenClusterSizeMultiplier to the warehouse create, edit and get types. Copy them in RemapState and DoUpdate's EditWarehouseRequest like the other fields (exhaustruct). customer_version_label is output-only; the generated sql_warehouses config already ignores its remote changes. Co-authored-by: Isaac --- bundle/direct/dresources/sql_warehouse.go | 70 ++++++++++++----------- 1 file changed, 38 insertions(+), 32 deletions(-) diff --git a/bundle/direct/dresources/sql_warehouse.go b/bundle/direct/dresources/sql_warehouse.go index 7d4578dd6f2..f7b9ca09fbe 100644 --- a/bundle/direct/dresources/sql_warehouse.go +++ b/bundle/direct/dresources/sql_warehouse.go @@ -72,22 +72,25 @@ func (*ResourceSqlWarehouse) PrepareState(input *resources.SqlWarehouse) *SqlWar // Started is derived from warehouse state so the planner can detect start/stop changes. func (*ResourceSqlWarehouse) RemapState(warehouse *SqlWarehouseRemote) *SqlWarehouseState { return &SqlWarehouseState{ - AutoStopMins: warehouse.AutoStopMins, - Channel: warehouse.Channel, - ClusterSize: warehouse.ClusterSize, - CreatorName: warehouse.CreatorName, - EnablePhoton: warehouse.EnablePhoton, - EnableServerlessCompute: warehouse.EnableServerlessCompute, - InstanceProfileArn: warehouse.InstanceProfileArn, - MaxNumClusters: warehouse.MaxNumClusters, - MinNumClusters: warehouse.MinNumClusters, - Name: warehouse.Name, - SpotInstancePolicy: warehouse.SpotInstancePolicy, - StatementTimeout: warehouse.StatementTimeout, - Tags: warehouse.Tags, - WarehouseType: sql.CreateWarehouseRequestWarehouseType(warehouse.WarehouseType), - ForceSendFields: utils.FilterFields[sql.CreateWarehouseRequest](warehouse.ForceSendFields), - Lifecycle: warehouse.Lifecycle, + AutoStopMins: warehouse.AutoStopMins, + Channel: warehouse.Channel, + ClusterSize: warehouse.ClusterSize, + CreatorName: warehouse.CreatorName, + CustomerVersionLabel: warehouse.CustomerVersionLabel, + EnablePhoton: warehouse.EnablePhoton, + EnableServerlessCompute: warehouse.EnableServerlessCompute, + InstanceProfileArn: warehouse.InstanceProfileArn, + MaxDbu: warehouse.MaxDbu, + MaxNumClusters: warehouse.MaxNumClusters, + MinNumClusters: warehouse.MinNumClusters, + Name: warehouse.Name, + ReydenClusterSizeMultiplier: warehouse.ReydenClusterSizeMultiplier, + SpotInstancePolicy: warehouse.SpotInstancePolicy, + StatementTimeout: warehouse.StatementTimeout, + Tags: warehouse.Tags, + WarehouseType: sql.CreateWarehouseRequestWarehouseType(warehouse.WarehouseType), + ForceSendFields: utils.FilterFields[sql.CreateWarehouseRequest](warehouse.ForceSendFields), + Lifecycle: warehouse.Lifecycle, } } @@ -135,22 +138,25 @@ func (r *ResourceSqlWarehouse) DoUpdate(ctx context.Context, id string, config * edited := hasWarehouseChanges(entry) if edited { request := sql.EditWarehouseRequest{ - AutoStopMins: config.AutoStopMins, - Channel: config.Channel, - ClusterSize: config.ClusterSize, - CreatorName: config.CreatorName, - EnablePhoton: config.EnablePhoton, - EnableServerlessCompute: config.EnableServerlessCompute, - Id: id, - InstanceProfileArn: config.InstanceProfileArn, - MaxNumClusters: config.MaxNumClusters, - MinNumClusters: config.MinNumClusters, - Name: config.Name, - SpotInstancePolicy: config.SpotInstancePolicy, - StatementTimeout: config.StatementTimeout, - Tags: config.Tags, - WarehouseType: sql.EditWarehouseRequestWarehouseType(config.WarehouseType), - ForceSendFields: utils.FilterFields[sql.EditWarehouseRequest](config.ForceSendFields), + AutoStopMins: config.AutoStopMins, + Channel: config.Channel, + ClusterSize: config.ClusterSize, + CreatorName: config.CreatorName, + CustomerVersionLabel: config.CustomerVersionLabel, + EnablePhoton: config.EnablePhoton, + EnableServerlessCompute: config.EnableServerlessCompute, + Id: id, + InstanceProfileArn: config.InstanceProfileArn, + MaxDbu: config.MaxDbu, + MaxNumClusters: config.MaxNumClusters, + MinNumClusters: config.MinNumClusters, + Name: config.Name, + ReydenClusterSizeMultiplier: config.ReydenClusterSizeMultiplier, + SpotInstancePolicy: config.SpotInstancePolicy, + StatementTimeout: config.StatementTimeout, + Tags: config.Tags, + WarehouseType: sql.EditWarehouseRequestWarehouseType(config.WarehouseType), + ForceSendFields: utils.FilterFields[sql.EditWarehouseRequest](config.ForceSendFields), } waiter, err := r.client.Warehouses.Edit(ctx, request) From c9d1ba4bd53835067f244f1960c7b673dd015f9b Mon Sep 17 00:00:00 2001 From: Jan Rose Date: Thu, 22 Oct 2026 06:04:08 +0000 Subject: [PATCH 4/4] Add PR link to the Go SDK v0.190.0 changelog entry Co-authored-by: Isaac --- .nextchanges/dependency-updates/go-sdk-v0.190.0.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.nextchanges/dependency-updates/go-sdk-v0.190.0.md b/.nextchanges/dependency-updates/go-sdk-v0.190.0.md index e36c8f8d41f..fc629ea54ef 100644 --- a/.nextchanges/dependency-updates/go-sdk-v0.190.0.md +++ b/.nextchanges/dependency-updates/go-sdk-v0.190.0.md @@ -1 +1 @@ -* Update the Go SDK to [v0.190.0](https://github.com/databricks/databricks-sdk-go/releases/tag/v0.190.0). +* Update the Go SDK to [v0.190.0](https://github.com/databricks/databricks-sdk-go/releases/tag/v0.190.0). ([#7155](https://github.com/databricks/cli/pull/7155))